azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Analysis API] add tests for KtSubstitutor

Browse Source
This commit is contained in:
Ilya Kirillov
2022-06-10 15:58:48 +02:00
parent f4d47cfe59
commit b9907963fe
24 changed files with 347 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/types/FirIdeDependentAnalysisSourceModuleAnalysisApiSubstitutorsTestGenerated.java
Generated
+72
View File
@@ -0,0 +1,72 @@
/*
* Copyright 2010-2021 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.fir.test.cases.generated.cases.types;
import com.intellij.testFramework.TestDataPath;
import org.jetbrains.kotlin.test.util.KtTestUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.kotlin.analysis.api.fir.test.configurators.AnalysisApiFirTestConfiguratorFactory;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfiguratorFactoryData;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfigurator;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.TestModuleKind;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.FrontendKind;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisSessionMode;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiMode;
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.types.AbstractAnalysisApiSubstitutorsTest;
import org.jetbrains.kotlin.test.TestMetadata;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import java.io.File;
import java.util.regex.Pattern;
/** This class is generated by {@link GenerateNewCompilerTests.kt}. DO NOT MODIFY MANUALLY */
@SuppressWarnings("all")
@TestMetadata("analysis/analysis-api/testData/substitutors/typeSubstitution")
@TestDataPath("$PROJECT_ROOT")
public class FirIdeDependentAnalysisSourceModuleAnalysisApiSubstitutorsTestGenerated extends AbstractAnalysisApiSubstitutorsTest {
@NotNull
@Override
public AnalysisApiTestConfigurator getConfigurator() {
return AnalysisApiFirTestConfiguratorFactory.INSTANCE.createConfigurator(
new AnalysisApiTestConfiguratorFactoryData(
FrontendKind.Fir,
TestModuleKind.Source,
AnalysisSessionMode.Dependent,
AnalysisApiMode.Ide
)
);
}
@Test
public void testAllFilesPresentInTypeSubstitution() throws Exception {
KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/analysis-api/testData/substitutors/typeSubstitution"), Pattern.compile("^(.+)\\.kt$"), null, true);
}
@Test
@TestMetadata("directSubstitution.kt")
public void testDirectSubstitution() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/directSubstitution.kt");
}
@Test
@TestMetadata("emptySubstitution.kt")
public void testEmptySubstitution() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/emptySubstitution.kt");
}
@Test
@TestMetadata("notApplicableSubstitutor.kt")
public void testNotApplicableSubstitutor() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/notApplicableSubstitutor.kt");
}
@Test
@TestMetadata("substitutionWithTypeParams.kt")
public void testSubstitutionWithTypeParams() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/substitutionWithTypeParams.kt");
}
}
analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/types/FirIdeNormalAnalysisSourceModuleAnalysisApiSubstitutorsTestGenerated.java
Generated
+72
View File
@@ -0,0 +1,72 @@
/*
* Copyright 2010-2021 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.fir.test.cases.generated.cases.types;
import com.intellij.testFramework.TestDataPath;
import org.jetbrains.kotlin.test.util.KtTestUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.kotlin.analysis.api.fir.test.configurators.AnalysisApiFirTestConfiguratorFactory;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfiguratorFactoryData;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfigurator;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.TestModuleKind;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.FrontendKind;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisSessionMode;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiMode;
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.types.AbstractAnalysisApiSubstitutorsTest;
import org.jetbrains.kotlin.test.TestMetadata;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import java.io.File;
import java.util.regex.Pattern;
/** This class is generated by {@link GenerateNewCompilerTests.kt}. DO NOT MODIFY MANUALLY */
@SuppressWarnings("all")
@TestMetadata("analysis/analysis-api/testData/substitutors/typeSubstitution")
@TestDataPath("$PROJECT_ROOT")
public class FirIdeNormalAnalysisSourceModuleAnalysisApiSubstitutorsTestGenerated extends AbstractAnalysisApiSubstitutorsTest {
@NotNull
@Override
public AnalysisApiTestConfigurator getConfigurator() {
return AnalysisApiFirTestConfiguratorFactory.INSTANCE.createConfigurator(
new AnalysisApiTestConfiguratorFactoryData(
FrontendKind.Fir,
TestModuleKind.Source,
AnalysisSessionMode.Normal,
AnalysisApiMode.Ide
)
);
}
@Test
public void testAllFilesPresentInTypeSubstitution() throws Exception {
KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/analysis-api/testData/substitutors/typeSubstitution"), Pattern.compile("^(.+)\\.kt$"), null, true);
}
@Test
@TestMetadata("directSubstitution.kt")
public void testDirectSubstitution() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/directSubstitution.kt");
}
@Test
@TestMetadata("emptySubstitution.kt")
public void testEmptySubstitution() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/emptySubstitution.kt");
}
@Test
@TestMetadata("notApplicableSubstitutor.kt")
public void testNotApplicableSubstitutor() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/notApplicableSubstitutor.kt");
}
@Test
@TestMetadata("substitutionWithTypeParams.kt")
public void testSubstitutionWithTypeParams() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/substitutionWithTypeParams.kt");
}
}
analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/components/signatureSubstitution/AbstractAnalysisApiSignatureSubstitutionTest.kt
+3 -2
View File
@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiSing
import org.jetbrains.kotlin.analysis.test.framework.services.SubstitutionParser
import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
import org.jetbrains.kotlin.psi.KtCallableDeclaration
import org.jetbrains.kotlin.psi.KtDeclaration
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.test.model.TestModule
@@ -20,11 +21,11 @@ import org.jetbrains.kotlin.test.services.assertions
abstract class AbstractAnalysisApiSignatureSubstitutionTest : AbstractAnalysisApiSingleFileTest() {
override fun doTestByFileStructure(ktFile: KtFile, module: TestModule, testServices: TestServices) {
val declaration = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtDeclaration>(ktFile)
val declaration = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtCallableDeclaration>(ktFile)
val actual = analyseForTest(declaration) {
val symbol = declaration.getSymbolOfType<KtCallableSymbol>()
val substitutor = with(SubstitutionParser) { parseSubstitutors(module, ktFile) }.single()
val substitutor = SubstitutionParser.parseSubstitutor(ktFile, declaration)
val signatureBeforeSubstitution = symbol.asSignature()
val signatureAfterSubstitution = signatureBeforeSubstitution.substitute(substitutor)
analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/components/signatureSubstitution/AbstractAnalysisApiSymbolSubstitutionTest.kt
+3 -2
View File
@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiSing
import org.jetbrains.kotlin.analysis.test.framework.services.SubstitutionParser
import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
import org.jetbrains.kotlin.psi.KtCallableDeclaration
import org.jetbrains.kotlin.psi.KtDeclaration
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.test.model.TestModule
@@ -20,11 +21,11 @@ import org.jetbrains.kotlin.test.services.assertions
abstract class AbstractAnalysisApiSymbolSubstitutionTest : AbstractAnalysisApiSingleFileTest() {
override fun doTestByFileStructure(ktFile: KtFile, module: TestModule, testServices: TestServices) {
val declaration = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtDeclaration>(ktFile)
val declaration = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtCallableDeclaration>(ktFile)
val actual = analyseForTest(declaration) {
val symbol = declaration.getSymbolOfType<KtCallableSymbol>()
val substitutor = with(SubstitutionParser) { parseSubstitutors(module, ktFile) }.single()
val substitutor = SubstitutionParser.parseSubstitutor(ktFile, declaration)
val signature = symbol.substitute(substitutor)
prettyPrint {
analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/components/testUtils.kt
+2
View File
@@ -6,6 +6,7 @@
package org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components
import com.intellij.psi.PsiElement
import org.jetbrains.kotlin.analysis.api.KtAnalysisApiInternals
import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
import org.jetbrains.kotlin.analysis.api.calls.KtCall
import org.jetbrains.kotlin.analysis.api.calls.KtCallableMemberCall
@@ -20,6 +21,7 @@ import kotlin.reflect.KProperty1
import kotlin.reflect.KVisibility
import kotlin.reflect.full.memberProperties
@OptIn(KtAnalysisApiInternals::class)
internal fun KtAnalysisSession.stringRepresentation(any: Any): String = with(any) {
fun KtType.render() = asStringForDebugging().replace('/', '.')
fun String.indented() = replace("\n", "\n ")
analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/types/AbstractAnalysisApiSubstitutorsTest.kt
+39
View File
@@ -0,0 +1,39 @@
/*
* Copyright 2010-2022 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.types
import org.jetbrains.kotlin.analysis.api.symbols.KtCallableSymbol
import org.jetbrains.kotlin.analysis.api.symbols.getSymbolOfType
import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiSingleFileTest
import org.jetbrains.kotlin.analysis.test.framework.services.SubstitutionParser
import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
import org.jetbrains.kotlin.psi.KtCallableDeclaration
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.test.model.TestModule
import org.jetbrains.kotlin.test.services.TestServices
import org.jetbrains.kotlin.test.services.assertions
abstract class AbstractAnalysisApiSubstitutorsTest : AbstractAnalysisApiSingleFileTest() {
override fun doTestByFileStructure(ktFile: KtFile, module: TestModule, testServices: TestServices) {
val declaration = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtCallableDeclaration>(ktFile)
val actual = analyseForTest(declaration) {
val substitutor = SubstitutionParser.parseSubstitutor(ktFile, declaration)
val symbol = declaration.getSymbolOfType<KtCallableSymbol>()
val type = symbol.returnType
val substituted = substitutor.substitute(type)
val substitutedOrNull = substitutor.substituteOrNull(type)
prettyPrint {
appendLine("PSI type: ${declaration.typeReference?.text}")
appendLine("KtType: ${type.render()}")
appendLine("substitutor.substitute: ${substituted.render()}")
appendLine("substitutor.substituteOrNull: ${substitutedOrNull?.render()}")
}
}
testServices.assertions.assertEqualsToTestDataFileSibling(actual)
}
}
analysis/analysis-api-standalone/tests-gen/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/generated/cases/types/FirStandaloneNormalAnalysisSourceModuleAnalysisApiSubstitutorsTestGenerated.java
Generated
+72
View File
@@ -0,0 +1,72 @@
/*
* Copyright 2010-2021 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.standalone.fir.test.cases.generated.cases.types;
import com.intellij.testFramework.TestDataPath;
import org.jetbrains.kotlin.test.util.KtTestUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.kotlin.analysis.api.standalone.fir.test.AnalysisApiFirStandaloneModeTestConfiguratorFactory;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfiguratorFactoryData;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfigurator;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.TestModuleKind;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.FrontendKind;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisSessionMode;
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiMode;
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.types.AbstractAnalysisApiSubstitutorsTest;
import org.jetbrains.kotlin.test.TestMetadata;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import java.io.File;
import java.util.regex.Pattern;
/** This class is generated by {@link GenerateNewCompilerTests.kt}. DO NOT MODIFY MANUALLY */
@SuppressWarnings("all")
@TestMetadata("analysis/analysis-api/testData/substitutors/typeSubstitution")
@TestDataPath("$PROJECT_ROOT")
public class FirStandaloneNormalAnalysisSourceModuleAnalysisApiSubstitutorsTestGenerated extends AbstractAnalysisApiSubstitutorsTest {
@NotNull
@Override
public AnalysisApiTestConfigurator getConfigurator() {
return AnalysisApiFirStandaloneModeTestConfiguratorFactory.INSTANCE.createConfigurator(
new AnalysisApiTestConfiguratorFactoryData(
FrontendKind.Fir,
TestModuleKind.Source,
AnalysisSessionMode.Normal,
AnalysisApiMode.Standalone
)
);
}
@Test
public void testAllFilesPresentInTypeSubstitution() throws Exception {
KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/analysis-api/testData/substitutors/typeSubstitution"), Pattern.compile("^(.+)\\.kt$"), null, true);
}
@Test
@TestMetadata("directSubstitution.kt")
public void testDirectSubstitution() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/directSubstitution.kt");
}
@Test
@TestMetadata("emptySubstitution.kt")
public void testEmptySubstitution() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/emptySubstitution.kt");
}
@Test
@TestMetadata("notApplicableSubstitutor.kt")
public void testNotApplicableSubstitutor() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/notApplicableSubstitutor.kt");
}
@Test
@TestMetadata("substitutionWithTypeParams.kt")
public void testSubstitutionWithTypeParams() throws Exception {
runTest("analysis/analysis-api/testData/substitutors/typeSubstitution/substitutionWithTypeParams.kt");
}
}
analysis/analysis-api/testData/components/signatureSubstitution/signatureSubstitution/function.kt
Vendored
+1 -1
View File
@@ -1,3 +1,3 @@
// SUBSTITUTOR: T->kotlin.collections.List<S>;S->kotlin.Long
// SUBSTITUTOR: T -> kotlin.collections.List<S>, S -> kotlin.Long
fun <T, S> f<caret>oo(x: List<T>, y: Map<T, List<S>>, k: String): T
analysis/analysis-api/testData/components/signatureSubstitution/signatureSubstitution/property.kt
Vendored
+1 -1
View File
@@ -1,3 +1,3 @@
// SUBSTITUTOR: T->kotlin.collections.List<S>;S->kotlin.Long
// SUBSTITUTOR: T-> kotlin.collections.List<S>, S -> kotlin.Long
val <T, S> Map<T, S>.val<caret>ue: List<S>
analysis/analysis-api/testData/components/signatureSubstitution/symbolSubstitution/function.kt
Vendored
+1 -1
View File
@@ -1,3 +1,3 @@
// SUBSTITUTOR: T->kotlin.collections.List<S>;S->kotlin.Long
// SUBSTITUTOR: T -> kotlin.collections.List<S>, S -> kotlin.Long
fun <T, S> f<caret>oo(x: List<T>, y: Map<T, List<S>>, k: String): T
analysis/analysis-api/testData/components/signatureSubstitution/symbolSubstitution/property.kt
Vendored
+1 -1
View File
@@ -1,3 +1,3 @@
// SUBSTITUTOR: T->kotlin.collections.List<S>;S->kotlin.Long
// SUBSTITUTOR: T -> kotlin.collections.List<S>, S -> kotlin.Long
val <T, S> Map<T, S>.val<caret>ue: List<S>
analysis/analysis-api/testData/substitutors/typeSubstitution/directSubstitution.kt
Vendored
+3
View File
@@ -0,0 +1,3 @@
// SUBSTITUTOR: A -> kotlin.Int, B -> kotlin.String
fun <A, B> fo<caret>o(): Map<List<A>, B>
analysis/analysis-api/testData/substitutors/typeSubstitution/directSubstitution.txt
Vendored
+4
View File
@@ -0,0 +1,4 @@
PSI type: Map<List<A>, B>
KtType: kotlin.collections.Map<kotlin.collections.List<A>, B>
substitutor.substitute: kotlin.collections.Map<kotlin.collections.List<kotlin.Int>, kotlin.String>
substitutor.substituteOrNull: kotlin.collections.Map<kotlin.collections.List<kotlin.Int>, kotlin.String>
analysis/analysis-api/testData/substitutors/typeSubstitution/emptySubstitution.kt
Vendored
+3
View File
@@ -0,0 +1,3 @@
// SUBSTITUTOR: A -> kotlin.Int
fun <A> fo<caret>o(): List<String>
analysis/analysis-api/testData/substitutors/typeSubstitution/emptySubstitution.txt
Vendored
+4
View File
@@ -0,0 +1,4 @@
PSI type: List<String>
KtType: kotlin.collections.List<kotlin.String>
substitutor.substitute: kotlin.collections.List<kotlin.String>
substitutor.substituteOrNull: null
analysis/analysis-api/testData/substitutors/typeSubstitution/notApplicableSubstitutor.kt
Vendored
+3
View File
@@ -0,0 +1,3 @@
// SUBSTITUTOR:
fun <A> fo<caret>o(): List<String>
analysis/analysis-api/testData/substitutors/typeSubstitution/notApplicableSubstitutor.txt
Vendored
+4
View File
@@ -0,0 +1,4 @@
PSI type: List<String>
KtType: kotlin.collections.List<kotlin.String>
substitutor.substitute: kotlin.collections.List<kotlin.String>
substitutor.substituteOrNull: null
analysis/analysis-api/testData/substitutors/typeSubstitution/substitutionWithTypeParams.kt
Vendored
+3
View File
@@ -0,0 +1,3 @@
// SUBSTITUTOR: A -> B, B -> kotlin.Int
fun <A, B> fo<caret>o(): Map<List<A>, B>
analysis/analysis-api/testData/substitutors/typeSubstitution/substitutionWithTypeParams.txt
Vendored
+4
View File
@@ -0,0 +1,4 @@
PSI type: Map<List<A>, B>
KtType: kotlin.collections.Map<kotlin.collections.List<A>, B>
substitutor.substitute: kotlin.collections.Map<kotlin.collections.List<B>, kotlin.Int>
substitutor.substituteOrNull: kotlin.collections.Map<kotlin.collections.List<B>, kotlin.Int>
analysis/analysis-test-framework/build.gradle.kts
+4
View File
@@ -33,3 +33,7 @@ projectTest(jUnitMode = JUnitMode.JUnit5) {
testsJar()
tasks.withType<org.jetbrains.kotlin.gradle.tasks.KotlinCompile>().configureEach {
kotlinOptions.freeCompilerArgs += "-Xcontext-receivers"
}
analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/base/AbstractAnalysisApiBasedTest.kt
-1
View File
@@ -104,7 +104,6 @@ abstract class AbstractAnalysisApiBasedTest : TestWithDisposable() {
useDirectives(*AbstractKotlinCompilerTest.defaultDirectiveContainers.toTypedArray())
useDirectives(JvmEnvironmentConfigurationDirectives)
useDirectives(SubstitutionParser.Directives)
useSourcePreprocessor(::ExpressionMarkersSourceFilePreprocessor)
useAdditionalService { ExpressionMarkerProvider() }
analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/services/SubstitutionParser.kt
+35 -18
View File
@@ -5,45 +5,62 @@
package org.jetbrains.kotlin.analysis.test.framework.services
import com.intellij.psi.PsiComment
import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
import org.jetbrains.kotlin.analysis.api.components.buildSubstitutor
import org.jetbrains.kotlin.analysis.api.types.KtSubstitutor
import org.jetbrains.kotlin.psi.KtCallableDeclaration
import org.jetbrains.kotlin.psi.KtElement
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.psi.KtTypeParameter
import org.jetbrains.kotlin.psi.psiUtil.collectDescendantsOfType
import org.jetbrains.kotlin.test.directives.model.SimpleDirectivesContainer
import org.jetbrains.kotlin.test.model.TestModule
object SubstitutionParser {
fun KtAnalysisSession.parseSubstitutors(module: TestModule, file: KtFile): List<KtSubstitutor> {
val directives = module.directives[Directives.SUBSTITUTOR]
if (directives.isEmpty()) return emptyList()
val allTypeParameterSymbols = file
context(KtAnalysisSession)
fun parseSubstitutor(declaration: KtCallableDeclaration): KtSubstitutor {
val comment = declaration.firstChild as PsiComment
return parseSubstitutor(comment, declaration)
}
context(KtAnalysisSession)
fun parseSubstitutor(ktFile: KtFile, declaration: KtCallableDeclaration): KtSubstitutor {
val comment = ktFile.children.filterIsInstance<PsiComment>().single { it.text.startsWith(SUBSTITUTOR_PREFIX) }
return parseSubstitutor(comment, declaration)
}
context(KtAnalysisSession)
fun parseSubstitutor(comment: PsiComment, scopeForTypeParameters: KtElement): KtSubstitutor {
val directivesAsString = comment.text.trim()
check(directivesAsString.startsWith(SUBSTITUTOR_PREFIX))
val substitutorAsMap = parseSubstitutions(directivesAsString.removePrefix(SUBSTITUTOR_PREFIX))
val allTypeParameterSymbols = scopeForTypeParameters
.collectDescendantsOfType<KtTypeParameter>()
.map { it.getTypeParameterSymbol() }
.groupBy { it.name.asString() }
.mapValues { it.value.single() }
return directives.map { subsitutor ->
buildSubstitutor {
subsitutor.forEach { (typeParameterName, typeString) ->
val typeParameterSymbol = allTypeParameterSymbols.getValue(typeParameterName)
val type = with(TypeParser) { parseTypeFromString(typeString, file, allTypeParameterSymbols) }
substitution(typeParameterSymbol, type)
}
return buildSubstitutor {
substitutorAsMap.forEach { (typeParameterName, typeString) ->
val typeParameterSymbol = allTypeParameterSymbols.getValue(typeParameterName)
val type = TypeParser.parseTypeFromString(typeString, scopeForTypeParameters, allTypeParameterSymbols)
substitution(typeParameterSymbol, type)
}
}
}
object Directives : SimpleDirectivesContainer() {
val SUBSTITUTOR by valueDirective("") { stringValue ->
stringValue.trim().split(";").map { it.trim() }.map { substitution ->
private fun parseSubstitutions(substitutionsAsString: String): List<Pair<String, String>> =
substitutionsAsString.trim().split(",")
.map { it.trim() }
.filter { it.isNotEmpty() }
.map { substitution ->
val asList = substitution.split("->").map { it.trim() }
check(asList.size == 2) {
"Substitution should look like `x -> y` but was `$substitution`"
}
asList[0] to asList[1]
}
}
}
const val SUBSTITUTOR_PREFIX = "// SUBSTITUTOR:"
}
analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/services/TypeParser.kt
+6 -4
View File
@@ -15,16 +15,18 @@ import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.psi.*
object TypeParser {
fun KtAnalysisSession.parseTypeFromString(
context (KtAnalysisSession)
fun parseTypeFromString(
stringType: String,
ktFile: KtFile,
contextElement: KtElement,
typeParameterByName: Map<String, KtTypeParameterSymbol>
): KtType {
val type = KtPsiFactory(ktFile).createType(stringType)
val type = KtPsiFactory(contextElement).createType(stringType)
return convertType(type.typeElement ?: incorrectType(type), typeParameterByName)
}
private fun KtAnalysisSession.convertType(type: KtTypeElement, typeParameterByName: Map<String, KtTypeParameterSymbol>): KtType =
context (KtAnalysisSession)
private fun convertType(type: KtTypeElement, typeParameterByName: Map<String, KtTypeParameterSymbol>): KtType =
when (type) {
is KtUserType -> {
val qualifier = fullQualifier(type)
generators/analysis-api-generator/tests/org/jetbrains/kotlin/generators/tests/analysis/api/analysisApi.kt
+7
View File
@@ -42,6 +42,7 @@ import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.scopes.AbstractSub
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.symbols.AbstractSymbolByFqNameTest
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.symbols.AbstractSymbolByPsiTest
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.symbols.AbstractSymbolByReferenceTest
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.types.AbstractAnalysisApiSubstitutorsTest
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiMode
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisSessionMode
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.FrontendKind
@@ -133,6 +134,12 @@ private fun AnalysisApiTestGroup.generateAnalysisApiNonComponentsTests() {
model("annotationsOnFiles")
}
}
group("substitutors", filter = frontendIs(FrontendKind.Fir)) {
test(AbstractAnalysisApiSubstitutorsTest::class) {
model("typeSubstitution")
}
}
}