Fix for EA-38626 - can't find class delegate for JetLightClass constructed for kotlin enum entry.

compiler/jet.as.java.psi/src/org/jetbrains/jet/asJava/JavaElementFinder.java

@@ -136,11 +136,15 @@ public class JavaElementFinder extends PsiElementFinder implements JavaPsiFacade
             if (localName != null) {
                 FqName fqn = QualifiedNamesUtil.combine(containerFqn, Name.identifier(localName));
                 if (qualifiedName.equals(fqn)) {
-                    answer.add(new JetLightClass(psiManager, file, qualifiedName));
+                    if (!(declaration instanceof JetEnumEntry)) {
+                        answer.add(new JetLightClass(psiManager, file, qualifiedName));
+                    }
                 }
-                else {
-                    for (JetDeclaration child : ((JetClassOrObject) declaration).getDeclarations()) {
-                        scanClasses(answer, child, qualifiedName, fqn, file);
+                else if (QualifiedNamesUtil.isSubpackageOf(qualifiedName, fqn)) {
+                    if (!(declaration instanceof JetEnumEntry)) {
+                        for (JetDeclaration child : ((JetClassOrObject) declaration).getDeclarations()) {
+                            scanClasses(answer, child, qualifiedName, fqn, file);
+                        }
                     }
                 }
             }

compiler/testData/asJava/findClasses/FromEnumEntry.kt

@@ -0,0 +1,18 @@
+enum class Direction {
+    NORTH
+    SOUTH(val x : Int) {
+        fun again() : String {
+            return "Hello"
+        }
+
+        class Hello
+    }
+    WEST {
+        class Some {
+            fun test() : Int {
+                return 12 + 14
+            }
+        }
+    }
+    EAST
+}

compiler/tests/compiler-tests.iml

@@ -12,6 +12,7 @@
     <orderEntry type="module" module-name="frontend.java" />
     <orderEntry type="module" module-name="runtime" />
     <orderEntry type="module" module-name="cli" />
+    <orderEntry type="module" module-name="jet.as.java.psi" />
     <orderEntry type="library" name="idea-full" level="project" />
     <orderEntry type="library" name="asm-addons" level="project" />
     <orderEntry type="library" name="dx-android" level="project" />

compiler/tests/org/jetbrains/jet/asJava/JavaElementFinderTest.java

@@ -0,0 +1,70 @@
+/*
+ * Copyright 2010-2012 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.jet.asJava;
+
+import com.intellij.psi.PsiClass;
+import com.intellij.psi.search.GlobalSearchScope;
+import org.jetbrains.jet.JetLiteFixture;
+import org.jetbrains.jet.cli.jvm.compiler.JetCoreEnvironment;
+import org.jetbrains.jet.config.CommonConfigurationKeys;
+import org.jetbrains.jet.config.CompilerConfiguration;
+
+/**
+ * @author Nikolay Krasko
+ */
+public class JavaElementFinderTest extends JetLiteFixture {
+    private JavaElementFinder finder;
+
+    public JavaElementFinderTest() {
+        super("asJava/findClasses");
+    }
+
+    @Override
+    protected JetCoreEnvironment createEnvironment() {
+        CompilerConfiguration configuration = new CompilerConfiguration();
+
+        configuration.add(CommonConfigurationKeys.SOURCE_ROOTS_KEY, getTestDataPath() + "/asJava/findClasses/" + getTestName(false) + ".kt");
+
+        return new JetCoreEnvironment(getTestRootDisposable(), configuration);
+    }
+
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+        finder = new JavaElementFinder(getProject());
+    }
+
+    public void testFromEnumEntry() {
+        assertClass("Direction");
+        assertNoClass("Direction.NORTH");
+        assertNoClass("Direction.SOUTH");
+        assertNoClass("Direction.WEST");
+        // TODO: assertClass("Direction.SOUTH.Hello");
+        // TODO: assertClass("Direction.WEST.Some");
+    }
+
+    private void assertClass(String qualifiedName) {
+        PsiClass psiClass = finder.findClass(qualifiedName, GlobalSearchScope.allScope(getProject()));
+        assertNotNull(String.format("Class with fqn='%s' wasn't found.", qualifiedName), psiClass);
+        assertTrue(String.format("Class with fqn='%s' is not valid.", qualifiedName), psiClass.isValid());
+    }
+
+    private void assertNoClass(String qualifiedName) {
+        assertNull(String.format("Class with fqn='%s' isn't expected to be found.", qualifiedName),
+                   finder.findClass(qualifiedName, GlobalSearchScope.allScope(getProject())));
+    }
+}