[Analysis API] implement an API to get a substitution based on an inheritance relationship between classes

^KT-62090 fixed
2023-09-20 22:34:44 +02:00
parent 49cfbb0576
commit 15e4cbc10d
33 changed files with 716 additions and 2 deletions
@@ -0,0 +1,62 @@
+/*
+ * 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.substitutorProvider
+
+import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.stringRepresentation
+import org.jetbrains.kotlin.analysis.api.renderer.types.impl.KtTypeRendererForSource
+import org.jetbrains.kotlin.analysis.api.symbols.KtConstructorSymbol
+import org.jetbrains.kotlin.analysis.api.symbols.KtFunctionSymbol
+import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiBasedTest
+import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
+import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
+import org.jetbrains.kotlin.psi.KtClassOrObject
+import org.jetbrains.kotlin.test.services.TestModuleStructure
+import org.jetbrains.kotlin.test.services.TestServices
+import org.jetbrains.kotlin.test.services.assertions
+import org.jetbrains.kotlin.types.Variance
+import kotlin.collections.single
+
+abstract class AbstractCreateInheritanceTypeSubstitutorTest : AbstractAnalysisApiBasedTest() {
+    override fun doTestByModuleStructure(moduleStructure: TestModuleStructure, testServices: TestServices) {
+        val baseClass = testServices.expressionMarkerProvider
+            .getElementsOfTypeAtCarets<KtClassOrObject>(moduleStructure, testServices, "base")
+            .single().first
+        val superClass = testServices.expressionMarkerProvider
+            .getElementsOfTypeAtCarets<KtClassOrObject>(moduleStructure, testServices, "super")
+            .single().first
+
+        val substitutorRendered = analyseForTest(baseClass) {
+            val superClassSymbol = superClass.getClassOrObjectSymbol()!!
+            val substitutor = createInheritanceTypeSubstitutor(baseClass.getClassOrObjectSymbol()!!, superClassSymbol)
+            prettyPrint {
+                appendLine("Substitutor: ${stringRepresentation(substitutor)}")
+                if (substitutor != null) {
+                    val functions = superClassSymbol.getDeclaredMemberScope().getAllSymbols()
+                        .filterIsInstance<KtFunctionSymbol>()
+                        .toList()
+                    if (functions.isNotEmpty()) {
+                        appendLine("Substituted callables:")
+                        withIndent {
+                            for (function in functions) {
+                                val signature = function.substitute(substitutor)
+                                append(signature.callableIdIfNonLocal!!.callableName.asString())
+                                printCollection(signature.valueParameters, prefix = "(", postfix = ")") {
+                                    append(it.returnType.render(typeRenderer, position = Variance.IN_VARIANCE))
+                                }
+                                append(": ${signature.returnType.render(typeRenderer, position = Variance.OUT_VARIANCE)}")
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        testServices.assertions.assertEqualsToTestDataFileSibling(substitutorRendered, extension = ".result.txt")
+    }
+
+    companion object {
+        private val typeRenderer = KtTypeRendererForSource.WITH_SHORT_NAMES
+    }
+}