Fix logic for searching inner classes in LazyResolveBasedCache.findInPackageFragments()

#KT-14058 Fixed

compiler/frontend.java/src/org/jetbrains/kotlin/load/java/components/LazyResolveBasedCache.kt

@@ -16,17 +16,25 @@
 
 package org.jetbrains.kotlin.load.java.components
 
-import org.jetbrains.kotlin.descriptors.*
-import org.jetbrains.kotlin.load.java.structure.*
-import org.jetbrains.kotlin.load.java.structure.impl.*
-import org.jetbrains.kotlin.resolve.lazy.ResolveSession
-import org.jetbrains.kotlin.resolve.lazy.ResolveSessionUtils
+import org.jetbrains.kotlin.descriptors.ClassDescriptor
+import org.jetbrains.kotlin.descriptors.ConstructorDescriptor
+import org.jetbrains.kotlin.descriptors.PropertyDescriptor
+import org.jetbrains.kotlin.descriptors.SimpleFunctionDescriptor
+import org.jetbrains.kotlin.load.java.structure.JavaClass
+import org.jetbrains.kotlin.load.java.structure.JavaElement
+import org.jetbrains.kotlin.load.java.structure.JavaField
+import org.jetbrains.kotlin.load.java.structure.JavaMethod
+import org.jetbrains.kotlin.load.java.structure.impl.JavaClassImpl
+import org.jetbrains.kotlin.load.java.structure.impl.JavaElementImpl
+import org.jetbrains.kotlin.load.java.structure.impl.JavaFieldImpl
+import org.jetbrains.kotlin.load.java.structure.impl.JavaMethodImpl
 import org.jetbrains.kotlin.name.FqName
-import kotlin.properties.Delegates
 import org.jetbrains.kotlin.name.tail
 import org.jetbrains.kotlin.resolve.BindingContext.*
-import org.jetbrains.kotlin.resolve.BindingTrace
 import org.jetbrains.kotlin.resolve.BindingContextUtils
+import org.jetbrains.kotlin.resolve.BindingTrace
+import org.jetbrains.kotlin.resolve.lazy.ResolveSession
+import org.jetbrains.kotlin.resolve.lazy.ResolveSessionUtils
 
 class LazyResolveBasedCache(private val resolveSession: ResolveSession) : JavaResolverCache {
 
@@ -57,10 +65,10 @@ class LazyResolveBasedCache(private val resolveSession: ResolveSession) : JavaRe
 
         while (true) {
             val packageDescriptor = resolveSession.getPackageFragment(fqName)
-            if (packageDescriptor == null) break
-
-            val result = ResolveSessionUtils.findClassByRelativePath(packageDescriptor.getMemberScope(), fullFqName.tail(fqName))
-            if (result != null) return result
+            if (packageDescriptor != null) {
+                val result = ResolveSessionUtils.findClassByRelativePath(packageDescriptor.getMemberScope(), fullFqName.tail(fqName))
+                if (result != null) return result
+            }
 
             if (fqName.isRoot) break
             fqName = fqName.parent()

idea/testData/multiFileHighlighting/dependencies/A.java

@@ -0,0 +1,14 @@
+package dependencies;
+
+import sealedClass.Foo;
+
+public class A {
+    public static class B<T> {
+    }
+
+    public static B<Foo.Bar> myList = null;
+
+    public static <T> T bar(B<T> tb) {
+        return null;
+    }
+}

idea/testData/multiFileHighlighting/dependencies/sealedClass.kt

@@ -0,0 +1,5 @@
+package sealedClass
+
+sealed class Foo {
+    class Bar : Foo()
+}

idea/testData/multiFileHighlighting/missingDependencyClass.kt

@@ -0,0 +1,5 @@
+package dependencies
+
+fun test() {
+    val <warning>bar</warning> = A.bar(A.myList) // MISSING_DEPENDENCY_CLASS foo.Foo.Bar
+}

idea/tests/org/jetbrains/kotlin/idea/stubs/MultiFileHighlightingTestGenerated.java

@@ -60,6 +60,12 @@ public class MultiFileHighlightingTestGenerated extends AbstractMultiFileHighlig
         doTest(fileName);
     }
 
+    @TestMetadata("missingDependencyClass.kt")
+    public void testMissingDependencyClass() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("idea/testData/multiFileHighlighting/missingDependencyClass.kt");
+        doTest(fileName);
+    }
+
     @TestMetadata("referencesFunWithUnspecifiedType.kt")
     public void testReferencesFunWithUnspecifiedType() throws Exception {
         String fileName = KotlinTestUtils.navigationMetadata("idea/testData/multiFileHighlighting/referencesFunWithUnspecifiedType.kt");