Fixed KT-2488 Optimizing imports removes references to enum constants when they are referred via short name in code

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java

@@ -133,7 +133,11 @@ public class JetPsiUtil {
     @Nullable
     public static FqName getFQName(JetNamedDeclaration namedDeclaration) {
         if (namedDeclaration instanceof JetObjectDeclarationName) {
-            JetObjectDeclaration objectDeclaration = PsiTreeUtil.getParentOfType(namedDeclaration, JetObjectDeclaration.class);
+            JetNamedDeclaration objectDeclaration = PsiTreeUtil.getParentOfType(namedDeclaration, JetObjectDeclaration.class);
+            if (objectDeclaration == null) {
+                objectDeclaration = PsiTreeUtil.getParentOfType(namedDeclaration, JetEnumEntry.class);
+            }
+            
             if (objectDeclaration == null) {
                 return null;
             }

idea/testData/editor/optimizeImports/kt2488.kt

@@ -0,0 +1,12 @@
+package test
+
+import test.MyEnum.ONE
+
+fun main(args: Array<String>) {
+    println(ONE)
+}
+
+public enum class MyEnum {
+    ONE
+    TWO
+}

idea/testData/editor/optimizeImports/kt2488_after.kt

@@ -0,0 +1,12 @@
+package test
+
+import test.MyEnum.ONE
+
+fun main(args: Array<String>) {
+    println(ONE)
+}
+
+public enum class MyEnum {
+    ONE
+    TWO
+}

idea/tests/org/jetbrains/jet/plugin/importOptimizer/OptimizeImportsTest.java

@@ -52,6 +52,10 @@ public class OptimizeImportsTest extends LightCodeInsightTestCase {
         doTest();
     }
 
+    public void testKt2488() throws Exception {
+        doTest();
+    }
+
     public void doTest() throws Exception {
         configureByFile(fileName());
         invokeFormatFile();
@@ -100,4 +104,4 @@ public class OptimizeImportsTest extends LightCodeInsightTestCase {
             getProject(), new JetImportOptimizer().processFile(getFile()),
             "Optimize Imports", null, UndoConfirmationPolicy.DO_NOT_REQUEST_CONFIRMATION);
     }
-}
+}