Find Usages: Fix reference matching for declarations located in libraries

#KT-5762 Fixed

idea/idea-analysis/src/org/jetbrains/jet/plugin/references/referenceUtil.kt

@@ -40,7 +40,7 @@ import org.jetbrains.kotlin.psi.psiUtil.getNonStrictParentOfType
 public val PsiReference.unwrappedTargets: Set<PsiElement>
     get() {
         fun PsiElement.adjust(): PsiElement? {
-            val target = unwrapped
+            val target = unwrapped?.getOriginalElement()
             return when {
                 target is JetPropertyAccessor -> target.getNonStrictParentOfType<JetProperty>()
                 target is JetObjectDeclaration && target.isClassObject() -> target.getNonStrictParentOfType<JetClass>()
@@ -55,7 +55,7 @@ public val PsiReference.unwrappedTargets: Set<PsiElement>
     }
 
 public fun PsiReference.matchesTarget(target: PsiElement): Boolean {
-    val unwrapped = target.unwrapped
+    val unwrapped = target.unwrapped?.getOriginalElement()
     return when {
         unwrapped in unwrappedTargets ->
             true

idea/testData/findUsages/kotlin/library/LibraryClassUsages.0.kt

@@ -0,0 +1,7 @@
+// PSI_ELEMENT: com.intellij.psi.PsiClass
+// OPTIONS: usages
+// FIND_BY_REF
+// FIND_BY_MIRROR_ELEMENT
+fun test() {
+    <caret>System.exit(0)
+}

idea/testData/findUsages/kotlin/library/LibraryClassUsages.results.txt

@@ -0,0 +1 @@
+Nested class/object (6: 5) System.exit(0)

idea/testData/findUsages/kotlin/library/LibraryFieldUsages.0.kt

@@ -0,0 +1,9 @@
+// PSI_ELEMENT: com.intellij.psi.PsiField
+// OPTIONS: usages
+// FIND_BY_REF
+// FIND_BY_MIRROR_ELEMENT
+import java.awt.Dimension
+
+fun test() {
+    Dimension().<caret>width = 1
+}

idea/testData/findUsages/kotlin/library/LibraryFieldUsages.results.txt

@@ -0,0 +1 @@
+Value write (8: 17) Dimension().width = 1

idea/testData/findUsages/kotlin/library/LibraryMethodUsages.0.kt

@@ -0,0 +1,9 @@
+// PSI_ELEMENT: com.intellij.psi.PsiMethod
+// OPTIONS: usages
+// FIND_BY_REF
+// FIND_BY_MIRROR_ELEMENT
+import java.awt.Dimension
+
+fun test() {
+    Dimension().<caret>setSize(1.0, 2.0)
+}

idea/testData/findUsages/kotlin/library/LibraryMethodUsages.results.txt

@@ -0,0 +1 @@
+Function call (8: 17) Dimension().setSize(1.0, 2.0)

idea/testData/findUsages/kotlin/library/LibraryStaticFieldUsages.0.kt

@@ -0,0 +1,7 @@
+// PSI_ELEMENT: com.intellij.psi.PsiField
+// OPTIONS: usages
+// FIND_BY_REF
+// FIND_BY_MIRROR_ELEMENT
+fun test() {
+    System.<caret>out.println()
+}

idea/testData/findUsages/kotlin/library/LibraryStaticFieldUsages.results.txt

@@ -0,0 +1 @@
+Receiver (6: 12) System.out.println()

idea/testData/findUsages/kotlin/library/LibraryStaticMethodUsages.0.kt

@@ -0,0 +1,7 @@
+// PSI_ELEMENT: com.intellij.psi.PsiMethod
+// OPTIONS: usages
+// FIND_BY_REF
+// FIND_BY_MIRROR_ELEMENT
+fun test() {
+    System.<caret>exit(0)
+}

idea/testData/findUsages/kotlin/library/LibraryStaticMethodUsages.results.txt

@@ -0,0 +1 @@
+Function call (6: 12) System.exit(0)

idea/tests/org/jetbrains/kotlin/findUsages/AbstractJetFindUsagesTest.java

@@ -328,6 +328,10 @@ public abstract class AbstractJetFindUsagesTest extends JetLightCodeInsightFixtu
                 InTextDirectivesUtils.isDirectiveDefined(mainFileText, "// FIND_BY_REF")
                 ? TargetElementUtilBase.findTargetElement(myFixture.getEditor(), TargetElementUtilBase.REFERENCED_ELEMENT_ACCEPTED)
                 : myFixture.getElementAtCaret();
+        if (InTextDirectivesUtils.isDirectiveDefined(mainFileText, "// FIND_BY_MIRROR_ELEMENT")) {
+            assert originalElement instanceof PsiCompiledElement : "PsiCompiledElement is expected: " + originalElement;
+            originalElement = ((PsiCompiledElement)originalElement).getMirror();
+        }
         T caretElement = PsiTreeUtil.getParentOfType(originalElement, caretElementClass, false);
         assertNotNull(String.format("Element with type '%s' wasn't found at caret position", caretElementClass), caretElement);
 

idea/tests/org/jetbrains/kotlin/findUsages/JetFindUsagesTestGenerated.java

@@ -33,7 +33,7 @@ import java.util.regex.Pattern;
 public class JetFindUsagesTestGenerated extends AbstractJetFindUsagesTest {
     @TestMetadata("idea/testData/findUsages/kotlin")
     @TestDataPath("$PROJECT_ROOT")
-    @InnerTestClasses({Kotlin.Conventions.class, Kotlin.FindClassUsages.class, Kotlin.FindFunctionUsages.class, Kotlin.FindObjectUsages.class, Kotlin.FindPackageUsages.class, Kotlin.FindParameterUsages.class, Kotlin.FindPropertyUsages.class, Kotlin.FindTypeParameterUsages.class, Kotlin.FindWithFilteringImports.class, Kotlin.FindWithStructuralGrouping.class, Kotlin.UnresolvedAnnotation.class})
+    @InnerTestClasses({Kotlin.Conventions.class, Kotlin.FindClassUsages.class, Kotlin.FindFunctionUsages.class, Kotlin.FindObjectUsages.class, Kotlin.FindPackageUsages.class, Kotlin.FindParameterUsages.class, Kotlin.FindPropertyUsages.class, Kotlin.FindTypeParameterUsages.class, Kotlin.FindWithFilteringImports.class, Kotlin.FindWithStructuralGrouping.class, Kotlin.Library.class, Kotlin.UnresolvedAnnotation.class})
     @RunWith(JUnit3RunnerWithInners.class)
     public static class Kotlin extends AbstractJetFindUsagesTest {
         public void testAllFilesPresentInKotlin() throws Exception {
@@ -844,6 +844,45 @@ public class JetFindUsagesTestGenerated extends AbstractJetFindUsagesTest {
             }
         }
 
+        @TestMetadata("idea/testData/findUsages/kotlin/library")
+        @TestDataPath("$PROJECT_ROOT")
+        @RunWith(JUnit3RunnerWithInners.class)
+        public static class Library extends AbstractJetFindUsagesTest {
+            public void testAllFilesPresentInLibrary() throws Exception {
+                JetTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/findUsages/kotlin/library"), Pattern.compile("^(.+)\\.0\\.kt$"), true);
+            }
+
+            @TestMetadata("LibraryClassUsages.0.kt")
+            public void testLibraryClassUsages() throws Exception {
+                String fileName = JetTestUtils.navigationMetadata("idea/testData/findUsages/kotlin/library/LibraryClassUsages.0.kt");
+                doTest(fileName);
+            }
+
+            @TestMetadata("LibraryFieldUsages.0.kt")
+            public void testLibraryFieldUsages() throws Exception {
+                String fileName = JetTestUtils.navigationMetadata("idea/testData/findUsages/kotlin/library/LibraryFieldUsages.0.kt");
+                doTest(fileName);
+            }
+
+            @TestMetadata("LibraryMethodUsages.0.kt")
+            public void testLibraryMethodUsages() throws Exception {
+                String fileName = JetTestUtils.navigationMetadata("idea/testData/findUsages/kotlin/library/LibraryMethodUsages.0.kt");
+                doTest(fileName);
+            }
+
+            @TestMetadata("LibraryStaticFieldUsages.0.kt")
+            public void testLibraryStaticFieldUsages() throws Exception {
+                String fileName = JetTestUtils.navigationMetadata("idea/testData/findUsages/kotlin/library/LibraryStaticFieldUsages.0.kt");
+                doTest(fileName);
+            }
+
+            @TestMetadata("LibraryStaticMethodUsages.0.kt")
+            public void testLibraryStaticMethodUsages() throws Exception {
+                String fileName = JetTestUtils.navigationMetadata("idea/testData/findUsages/kotlin/library/LibraryStaticMethodUsages.0.kt");
+                doTest(fileName);
+            }
+        }
+
         @TestMetadata("idea/testData/findUsages/kotlin/unresolvedAnnotation")
         @TestDataPath("$PROJECT_ROOT")
         @RunWith(JUnit3RunnerWithInners.class)