Analysis API: add isSubclassOf tests

2021-12-31 19:36:53 +03:00
parent e6fc434489
commit 11c50a48ba
7 changed files with 183 additions and 14 deletions
@@ -0,0 +1,37 @@
+/*
+ * 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.impl.base.test.components.symbolDeclarationOverridesProvider
+
+import org.jetbrains.kotlin.analysis.api.impl.barebone.test.expressionMarkerProvider
+import org.jetbrains.kotlin.analysis.api.impl.base.test.test.framework.AbstractHLApiSingleModuleTest
+import org.jetbrains.kotlin.psi.KtClassOrObject
+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 AbstractIsSubclassOfTest : AbstractHLApiSingleModuleTest() {
+    override fun doTestByFileStructure(ktFiles: List<KtFile>, module: TestModule, testServices: TestServices) {
+        val subClass = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtClassOrObject>(ktFiles.first(), "sub")
+        val superClass = testServices.expressionMarkerProvider.getElementOfTypAtCaret<KtClassOrObject>(ktFiles.first(), "super")
+
+        val actual = executeOnPooledThreadInReadAction {
+            analyseForTest(subClass) {
+                val subClassSymbol = subClass.getClassOrObjectSymbol()
+                val superClassSymbol = superClass.getClassOrObjectSymbol()
+
+                val isSubClass = subClassSymbol.isSubClassOf(superClassSymbol)
+                val isDirectSubClass = subClassSymbol.isDirectSubClassOf(superClassSymbol)
+
+                buildString {
+                    appendLine("IS_SUBCLASS: $isSubClass")
+                    appendLine("IS_DIRECT_SUBCLASS: $isDirectSubClass")
+                }
+            }
+        }
+        testServices.assertions.assertEqualsToTestDataFileSibling(actual)
+    }
+}