KT-949 AssertionFailure in quick fix

idea/src/org/jetbrains/jet/plugin/annotations/JetPsiChecker.java

@@ -123,6 +123,11 @@ public class JetPsiChecker implements Annotator {
             catch (ProcessCanceledException e) {
                 throw e;
             }
+            catch (AssertionError e) {
+                // For failing tests and to notify about idea internal error in -ea mode
+                holder.createErrorAnnotation(element, e.getClass().getCanonicalName() + ": " + e.getMessage());
+                throw e;
+            }
             catch (Throwable e) {
                 // TODO
                 holder.createErrorAnnotation(element, e.getClass().getCanonicalName() + ": " + e.getMessage());

idea/src/org/jetbrains/jet/plugin/quickfix/ImportClassFix.java

@@ -177,8 +177,7 @@ public class ImportClassFix extends JetHintAction<JetSimpleNameExpression> imple
             @Nullable
             @Override
             public JetIntentionAction<JetSimpleNameExpression> createAction(@NotNull DiagnosticWithPsiElement diagnostic) {
-                assert diagnostic.getPsiElement() instanceof JetSimpleNameExpression;
-
+                // There could be different psi elements (i.e. JetArrayAccessExpression), but we can fix only JetSimpleNameExpression case
                 if (diagnostic.getPsiElement() instanceof JetSimpleNameExpression) {
                     JetSimpleNameExpression psiElement = (JetSimpleNameExpression) diagnostic.getPsiElement();
                     return new ImportClassFix(psiElement);

idea/testData/quickfix/autoImports/afterNoImportForIndex.kt

+// "Import Class" "false"
+
+package Teting
+
+class Some() {
+//    fun get(i : Int) : Int {
+//        return i
+//    }
+}
+
+fun main(args : Array<String>) {
+    val some = Some()
+    // Nothing should be changed
+    <caret>some[12]
+}
\ No newline at end of file

idea/testData/quickfix/autoImports/beforeNoImportForIndex.kt

+// "Import Class" "false"
+
+package Teting
+
+class Some() {
+//    fun get(i : Int) : Int {
+//        return i
+//    }
+}
+
+fun main(args : Array<String>) {
+    val some = Some()
+    // Nothing should be changed
+    <caret>some[12]
+}
\ No newline at end of file