[LL FIR] tests: extract scope rendering utilities to a separate object
For further reuse ^KT-57966
This commit is contained in:
committed by
Space Team
parent
e9f75b1350
commit
67041851fc
+2
-75
@@ -7,16 +7,7 @@ package org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.scopeP
|
|||||||
|
|
||||||
import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
|
import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
|
||||||
import org.jetbrains.kotlin.analysis.api.components.KtScopeContext
|
import org.jetbrains.kotlin.analysis.api.components.KtScopeContext
|
||||||
import org.jetbrains.kotlin.analysis.api.components.KtScopeKind
|
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.scopeProvider.TestScopeRenderer.renderForTests
|
||||||
import org.jetbrains.kotlin.analysis.api.renderer.base.annotations.KtRendererAnnotationsFilter
|
|
||||||
import org.jetbrains.kotlin.analysis.api.renderer.declarations.impl.KtDeclarationRendererForSource
|
|
||||||
import org.jetbrains.kotlin.analysis.api.renderer.declarations.modifiers.renderers.KtRendererModifierFilter
|
|
||||||
import org.jetbrains.kotlin.analysis.api.renderer.types.impl.KtTypeRendererForSource
|
|
||||||
import org.jetbrains.kotlin.analysis.api.renderer.types.renderers.KtTypeErrorTypeRenderer
|
|
||||||
import org.jetbrains.kotlin.analysis.api.scopes.KtScope
|
|
||||||
import org.jetbrains.kotlin.analysis.api.symbols.DebugSymbolRenderer
|
|
||||||
import org.jetbrains.kotlin.analysis.api.symbols.KtDeclarationSymbol
|
|
||||||
import org.jetbrains.kotlin.analysis.api.types.KtType
|
|
||||||
import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiBasedSingleModuleTest
|
import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiBasedSingleModuleTest
|
||||||
import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
|
import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
|
||||||
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
|
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
|
||||||
@@ -48,70 +39,6 @@ abstract class AbstractScopeContextForPositionTest : AbstractAnalysisApiBasedSin
|
|||||||
printPretty: Boolean = false
|
printPretty: Boolean = false
|
||||||
): String = prettyPrint {
|
): String = prettyPrint {
|
||||||
appendLine("element: ${element.text}")
|
appendLine("element: ${element.text}")
|
||||||
appendLine("implicit receivers:")
|
renderForTests(scopeContext, printPretty)
|
||||||
|
|
||||||
withIndent {
|
|
||||||
for (implicitReceiver in scopeContext.implicitReceivers) {
|
|
||||||
val type = implicitReceiver.type
|
|
||||||
appendLine("type: ${renderType(type, printPretty)}")
|
|
||||||
appendLine("owner symbol: ${implicitReceiver.ownerSymbol::class.simpleName}")
|
|
||||||
appendLine()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
appendLine("scopes:")
|
|
||||||
withIndent {
|
|
||||||
for (scopeWithKind in scopeContext.scopes) {
|
|
||||||
appendLine(renderForTests(scopeWithKind.scope, scopeWithKind.kind, printPretty))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private fun KtAnalysisSession.renderForTests(scope: KtScope, scopeKind: KtScopeKind, printPretty: Boolean): String = prettyPrint {
|
|
||||||
append("${scopeKind::class.simpleName}, index = ${scopeKind.indexInTower}")
|
|
||||||
|
|
||||||
if (scopeKind is KtScopeKind.DefaultSimpleImportingScope || scopeKind is KtScopeKind.DefaultStarImportingScope) {
|
|
||||||
appendLine()
|
|
||||||
return@prettyPrint
|
|
||||||
}
|
|
||||||
|
|
||||||
val callables = scope.getCallableSymbols().toList()
|
|
||||||
val classifiers = scope.getClassifierSymbols().toList()
|
|
||||||
val isEmpty = callables.isEmpty() && classifiers.isEmpty()
|
|
||||||
if (isEmpty) {
|
|
||||||
appendLine(", empty")
|
|
||||||
} else {
|
|
||||||
appendLine()
|
|
||||||
withIndent {
|
|
||||||
appendLine("classifiers: ${classifiers.size}")
|
|
||||||
withIndent { classifiers.forEach { appendLine(renderSymbol(it, printPretty)) } }
|
|
||||||
appendLine("callables: ${callables.size}")
|
|
||||||
withIndent { callables.forEach { appendLine(renderSymbol(it, printPretty)) } }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private fun KtAnalysisSession.renderType(
|
|
||||||
type: KtType,
|
|
||||||
printPretty: Boolean
|
|
||||||
): String = prettyPrint {
|
|
||||||
if (printPretty) prettyPrintTypeRenderer.renderType(type, this) else append(debugRenderer.renderType(type))
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
private fun KtAnalysisSession.renderSymbol(
|
|
||||||
symbol: KtDeclarationSymbol,
|
|
||||||
printPretty: Boolean
|
|
||||||
): String = if (printPretty) symbol.render(prettyPrintSymbolRenderer) else debugRenderer.render(symbol)
|
|
||||||
|
|
||||||
companion object {
|
|
||||||
private val debugRenderer = DebugSymbolRenderer()
|
|
||||||
|
|
||||||
private val prettyPrintTypeRenderer = KtTypeRendererForSource.WITH_QUALIFIED_NAMES.with {
|
|
||||||
typeErrorTypeRenderer = KtTypeErrorTypeRenderer.WITH_ERROR_MESSAGE
|
|
||||||
}
|
|
||||||
private val prettyPrintSymbolRenderer = KtDeclarationRendererForSource.WITH_QUALIFIED_NAMES.with {
|
|
||||||
annotationRenderer = annotationRenderer.with { annotationFilter = KtRendererAnnotationsFilter.NONE }
|
|
||||||
modifiersRenderer = modifiersRenderer.with { modifierFilter = KtRendererModifierFilter.NONE }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+114
@@ -0,0 +1,114 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||||
|
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.scopeProvider
|
||||||
|
|
||||||
|
import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
|
||||||
|
import org.jetbrains.kotlin.analysis.api.components.KtScopeContext
|
||||||
|
import org.jetbrains.kotlin.analysis.api.components.KtScopeKind
|
||||||
|
import org.jetbrains.kotlin.analysis.api.renderer.base.annotations.KtRendererAnnotationsFilter
|
||||||
|
import org.jetbrains.kotlin.analysis.api.renderer.declarations.impl.KtDeclarationRendererForSource
|
||||||
|
import org.jetbrains.kotlin.analysis.api.renderer.declarations.modifiers.renderers.KtRendererModifierFilter
|
||||||
|
import org.jetbrains.kotlin.analysis.api.renderer.types.impl.KtTypeRendererForSource
|
||||||
|
import org.jetbrains.kotlin.analysis.api.renderer.types.renderers.KtTypeErrorTypeRenderer
|
||||||
|
import org.jetbrains.kotlin.analysis.api.scopes.KtScope
|
||||||
|
import org.jetbrains.kotlin.analysis.api.symbols.DebugSymbolRenderer
|
||||||
|
import org.jetbrains.kotlin.analysis.api.symbols.KtDeclarationSymbol
|
||||||
|
import org.jetbrains.kotlin.analysis.api.types.KtType
|
||||||
|
import org.jetbrains.kotlin.analysis.utils.printer.PrettyPrinter
|
||||||
|
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
|
||||||
|
|
||||||
|
internal object TestScopeRenderer {
|
||||||
|
|
||||||
|
context (KtAnalysisSession)
|
||||||
|
fun PrettyPrinter.renderForTests(
|
||||||
|
scopeContext: KtScopeContext,
|
||||||
|
printPretty: Boolean = false
|
||||||
|
) {
|
||||||
|
appendLine("implicit receivers:")
|
||||||
|
|
||||||
|
withIndent {
|
||||||
|
for (implicitReceiver in scopeContext.implicitReceivers) {
|
||||||
|
val type = implicitReceiver.type
|
||||||
|
appendLine("type: ${renderType(type, printPretty)}")
|
||||||
|
appendLine("owner symbol: ${implicitReceiver.ownerSymbol::class.simpleName}")
|
||||||
|
appendLine()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
appendLine("scopes:")
|
||||||
|
withIndent {
|
||||||
|
renderScopeContext(scopeContext, printPretty)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
context (KtAnalysisSession)
|
||||||
|
private fun renderType(
|
||||||
|
type: KtType,
|
||||||
|
printPretty: Boolean
|
||||||
|
): String = prettyPrint {
|
||||||
|
if (printPretty) {
|
||||||
|
prettyPrintTypeRenderer.renderType(type, this)
|
||||||
|
} else {
|
||||||
|
append(debugRenderer.renderType(type))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
context(KtAnalysisSession)
|
||||||
|
private fun PrettyPrinter.renderScopeContext(
|
||||||
|
scopeContext: KtScopeContext,
|
||||||
|
printPretty: Boolean
|
||||||
|
) {
|
||||||
|
for (scopeWithKind in scopeContext.scopes) {
|
||||||
|
appendLine(renderForTests(scopeWithKind.scope, scopeWithKind.kind, printPretty))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
context (KtAnalysisSession)
|
||||||
|
private fun renderForTests(scope: KtScope, scopeKind: KtScopeKind, printPretty: Boolean): String = prettyPrint {
|
||||||
|
append("${scopeKind::class.simpleName}, index = ${scopeKind.indexInTower}")
|
||||||
|
|
||||||
|
if (scopeKind is KtScopeKind.DefaultSimpleImportingScope || scopeKind is KtScopeKind.DefaultStarImportingScope) {
|
||||||
|
appendLine()
|
||||||
|
return@prettyPrint
|
||||||
|
}
|
||||||
|
|
||||||
|
renderScopeMembers(scope, printPretty)
|
||||||
|
}
|
||||||
|
|
||||||
|
context (KtAnalysisSession)
|
||||||
|
private fun PrettyPrinter.renderScopeMembers(scope: KtScope, printPretty: Boolean) {
|
||||||
|
val callables = scope.getCallableSymbols().toList()
|
||||||
|
val classifiers = scope.getClassifierSymbols().toList()
|
||||||
|
val isEmpty = callables.isEmpty() && classifiers.isEmpty()
|
||||||
|
if (isEmpty) {
|
||||||
|
appendLine(", empty")
|
||||||
|
} else {
|
||||||
|
appendLine()
|
||||||
|
withIndent {
|
||||||
|
appendLine("classifiers: ${classifiers.size}")
|
||||||
|
withIndent { classifiers.forEach { appendLine(TestScopeRenderer.renderSymbol(it, printPretty)) } }
|
||||||
|
appendLine("callables: ${callables.size}")
|
||||||
|
withIndent { callables.forEach { appendLine(TestScopeRenderer.renderSymbol(it, printPretty)) } }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
context(KtAnalysisSession)
|
||||||
|
private fun renderSymbol(
|
||||||
|
symbol: KtDeclarationSymbol,
|
||||||
|
printPretty: Boolean
|
||||||
|
): String = if (printPretty) symbol.render(prettyPrintSymbolRenderer) else debugRenderer.render(symbol)
|
||||||
|
|
||||||
|
private val debugRenderer = DebugSymbolRenderer()
|
||||||
|
|
||||||
|
private val prettyPrintSymbolRenderer = KtDeclarationRendererForSource.WITH_QUALIFIED_NAMES.with {
|
||||||
|
annotationRenderer = annotationRenderer.with { annotationFilter = KtRendererAnnotationsFilter.NONE }
|
||||||
|
modifiersRenderer = modifiersRenderer.with { modifierFilter = KtRendererModifierFilter.NONE }
|
||||||
|
}
|
||||||
|
|
||||||
|
private val prettyPrintTypeRenderer = KtTypeRendererForSource.WITH_QUALIFIED_NAMES.with {
|
||||||
|
typeErrorTypeRenderer = KtTypeErrorTypeRenderer.WITH_ERROR_MESSAGE
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user