KT-12628: !! is now correctly added to array expressions

(cherry picked from commit 56c9d7e)

idea/src/org/jetbrains/kotlin/idea/quickfix/ExclExclCallFixes.kt

@@ -95,6 +95,9 @@ class AddExclExclCallFix(val psiElement: PsiElement) : ExclExclCallFix() {
                 return sibling
             }
         }
+        else if (psiElement is KtArrayAccessExpression) {
+            return psiElement.arrayExpression
+        }
         else if (psiElement is KtExpression) {
             return psiElement
         }

idea/testData/quickfix/expressions/unsafeCall4.kt

@@ -0,0 +1,4 @@
+// "Add non-null asserted (!!) call" "true"
+
+operator fun Int.get(row: Int, column: Int) = this
+fun foo(arg: Int?) = arg<caret>[42, 13]

idea/testData/quickfix/expressions/unsafeCall4.kt.after

@@ -0,0 +1,4 @@
+// "Add non-null asserted (!!) call" "true"
+
+operator fun Int.get(row: Int, column: Int) = this
+fun foo(arg: Int?) = arg!![42, 13]

idea/testData/quickfix/expressions/unsafeCall5.kt

@@ -0,0 +1,4 @@
+// "Add non-null asserted (!!) call" "true"
+
+operator fun Int.get(row: Int, column: Int) = if (row == column) this else null
+fun foo(arg: Int) = arg[42, 13]<caret>.hashCode()

idea/testData/quickfix/expressions/unsafeCall5.kt.after

@@ -0,0 +1,4 @@
+// "Add non-null asserted (!!) call" "true"
+
+operator fun Int.get(row: Int, column: Int) = if (row == column) this else null
+fun foo(arg: Int) = arg[42, 13]!!.hashCode()

idea/tests/org/jetbrains/kotlin/idea/quickfix/QuickFixTestGenerated.java

@@ -4809,6 +4809,18 @@ public class QuickFixTestGenerated extends AbstractQuickFixTest {
             doTest(fileName);
         }
 
+        @TestMetadata("unsafeCall4.kt")
+        public void testUnsafeCall4() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/quickfix/expressions/unsafeCall4.kt");
+            doTest(fileName);
+        }
+
+        @TestMetadata("unsafeCall5.kt")
+        public void testUnsafeCall5() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/quickfix/expressions/unsafeCall5.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("uselessCastStaticAssertIsFine.kt")
         public void testUselessCastStaticAssertIsFine() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("idea/testData/quickfix/expressions/uselessCastStaticAssertIsFine.kt");