Navigation: add expect declarations as supers #KT-16892 Fixed

generators/tests/org/jetbrains/kotlin/generators/tests/GenerateTests.kt

@@ -214,7 +214,7 @@ fun main(args: Array<String>) {
         }
 
         testClass<AbstractGotoSuperTest> {
-            model("navigation/gotoSuper", extension = "test")
+            model("navigation/gotoSuper", extension = "test", recursive = false)
         }
 
         testClass<AbstractGotoTypeDeclarationTest> {

idea/src/org/jetbrains/kotlin/idea/codeInsight/GotoSuperActionHandler.kt

@@ -22,14 +22,14 @@ import org.jetbrains.kotlin.descriptors.*
 import org.jetbrains.kotlin.idea.KotlinBundle
 import org.jetbrains.kotlin.idea.caches.resolve.unsafeResolveToDescriptor
 import org.jetbrains.kotlin.idea.core.getDirectlyOverriddenDeclarations
+import org.jetbrains.kotlin.idea.highlighter.markers.expectedDeclarationIfAny
 import org.jetbrains.kotlin.psi.*
+import org.jetbrains.kotlin.psi.psiUtil.hasActualModifier
 import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
 
 class GotoSuperActionHandler : CodeInsightActionHandler {
-    override fun invoke(project: Project, editor: Editor, file: PsiFile) {
-        FeatureUsageTracker.getInstance().triggerFeatureUsed(GotoSuperAction.FEATURE_ID)
-
-        val element = file.findElementAt(editor.caretModel.offset) ?: return
+    internal fun allSuperDeclarationsAndDescriptor(editor: Editor, file: PsiFile): Pair<List<PsiElement>, DeclarationDescriptor?> {
+        val element = file.findElementAt(editor.caretModel.offset) ?: return emptyList<PsiElement>() to null
         val declaration =
             PsiTreeUtil.getParentOfType<KtDeclaration>(
                 element,
@@ -37,21 +37,28 @@ class GotoSuperActionHandler : CodeInsightActionHandler {
                 KtClass::class.java,
                 KtProperty::class.java,
                 KtObjectDeclaration::class.java
-            ) ?: return
+            ) ?: return emptyList<PsiElement>() to null
+
+        val expectDeclaration = if (declaration.hasActualModifier()) declaration.expectedDeclarationIfAny() else null
 
         val descriptor = declaration.unsafeResolveToDescriptor(BodyResolveMode.PARTIAL)
+        val superDeclarations = findSuperDeclarations(file.project, descriptor)
+        return ((superDeclarations ?: emptyList()) + listOfNotNull(expectDeclaration)) to descriptor
+    }
 
-        val superDeclarations = findSuperDeclarations(project, descriptor)
+    override fun invoke(project: Project, editor: Editor, file: PsiFile) {
+        FeatureUsageTracker.getInstance().triggerFeatureUsed(GotoSuperAction.FEATURE_ID)
 
-        if (superDeclarations == null || superDeclarations.isEmpty()) return
-        if (superDeclarations.size == 1) {
-            val navigatable = EditSourceUtil.getDescriptor(superDeclarations[0])
+        val (allDeclarations, descriptor) = allSuperDeclarationsAndDescriptor(editor, file)
+        if (allDeclarations.isEmpty()) return
+        if (allDeclarations.size == 1) {
+            val navigatable = EditSourceUtil.getDescriptor(allDeclarations[0])
             if (navigatable != null && navigatable.canNavigate()) {
                 navigatable.navigate(true)
             }
         } else {
-            val message = getTitle(descriptor)
-            val superDeclarationsArray = PsiUtilCore.toPsiElementArray(superDeclarations)
+            val message = getTitle(descriptor!!)
+            val superDeclarationsArray = PsiUtilCore.toPsiElementArray(allDeclarations)
             val popup = if (descriptor is ClassDescriptor)
                 NavigationUtil.getPsiElementPopup(superDeclarationsArray, message)
             else
@@ -64,7 +71,7 @@ class GotoSuperActionHandler : CodeInsightActionHandler {
     }
 
     private fun getTitle(descriptor: DeclarationDescriptor): String? =
-        when(descriptor) {
+        when (descriptor) {
             is ClassDescriptor -> KotlinBundle.message("goto.super.class.chooser.title")
             is PropertyDescriptor -> KotlinBundle.message("goto.super.property.chooser.title")
             is SimpleFunctionDescriptor -> KotlinBundle.message("goto.super.function.chooser.title")

idea/testData/navigation/gotoSuper/multiModule/actualClass/common/common.kt

@@ -0,0 +1,3 @@
+package test
+
+expect class Expected

idea/testData/navigation/gotoSuper/multiModule/actualClass/jvm/jvm.kt

@@ -0,0 +1,5 @@
+package test
+
+actual class <caret>Expected
+
+// REF: [common] (test).Expected

idea/testData/navigation/gotoSuper/multiModule/actualFunction/common/common.kt

@@ -0,0 +1,7 @@
+package test
+
+expect class Expected {
+    fun foo(): Int
+
+    val bar: String
+}

idea/testData/navigation/gotoSuper/multiModule/actualFunction/jvm/jvm.kt

@@ -0,0 +1,10 @@
+package test
+
+actual class Expected {
+    actual fun <caret>foo() = 42
+
+    actual val bar = "Hello"
+}
+
+// REF: [common] (in test.Expected).foo()
+

idea/testData/navigation/gotoSuper/multiModule/actualProperty/common/common.kt

@@ -0,0 +1,7 @@
+package test
+
+expect class Expected {
+    fun foo(): Int
+
+    val bar: String
+}

idea/testData/navigation/gotoSuper/multiModule/actualProperty/jvm/jvm.kt

@@ -0,0 +1,10 @@
+package test
+
+actual class Expected {
+    actual fun foo() = 42
+
+    actual val <caret>bar = "Hello"
+}
+
+// REF: [common] (in test.Expected).bar
+

idea/tests/org/jetbrains/kotlin/idea/navigation/GotoSuperTestGenerated.java

@@ -22,7 +22,7 @@ import java.util.regex.Pattern;
 @RunWith(JUnit3RunnerWithInners.class)
 public class GotoSuperTestGenerated extends AbstractGotoSuperTest {
     public void testAllFilesPresentInGotoSuper() throws Exception {
-        KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/navigation/gotoSuper"), Pattern.compile("^(.+)\\.test$"), TargetBackend.ANY, true);
+        KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/navigation/gotoSuper"), Pattern.compile("^(.+)\\.test$"), TargetBackend.ANY, false);
     }
 
     @TestMetadata("BadPositionLambdaParameter.test")

idea/tests/org/jetbrains/kotlin/idea/navigation/KotlinGotoSuperMultiModuleTest.kt

@@ -0,0 +1,36 @@
+/*
+ * Copyright 2000-2018 JetBrains s.r.o. 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.idea.navigation
+
+import com.intellij.codeInsight.navigation.GotoTargetHandler
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiFile
+import org.jetbrains.kotlin.idea.codeInsight.GotoSuperActionHandler
+import org.jetbrains.kotlin.idea.test.PluginTestCaseBase
+import java.io.File
+
+class KotlinGotoSuperMultiModuleTest : AbstractKotlinNavigationMultiModuleTest() {
+    override fun doNavigate(editor: Editor, file: PsiFile): GotoTargetHandler.GotoData {
+        val gotoAction = GotoSuperActionHandler()
+        val (superDeclarations, _) = gotoAction.allSuperDeclarationsAndDescriptor(editor, file)
+        return GotoTargetHandler.GotoData(file.findElementAt(editor.caretModel.offset)!!, superDeclarations.toTypedArray(), emptyList())
+    }
+
+    override fun getTestDataPath() =
+        File(PluginTestCaseBase.getTestDataPathBase(), "/navigation/gotoSuper/multiModule").path + File.separator
+
+    fun testActualClass() {
+        doMultiPlatformTest("jvm.kt")
+    }
+
+    fun testActualFunction() {
+        doMultiPlatformTest("jvm.kt")
+    }
+
+    fun testActualProperty() {
+        doMultiPlatformTest("jvm.kt")
+    }
+}