KT-11104 : quick fix "surround with null check"

idea/testData/quickfix/surroundWithNullCheck/chainedUnsafeCall.kt

@@ -0,0 +1,9 @@
+// "Surround with null check" "false"
+// ACTION: Introduce local variable
+// ACTION: Add non-null asserted (!!) call
+// ACTION: Replace with safe (?.) call
+// ERROR: Only safe (?.) or non-null asserted (!!.) calls are allowed on a nullable receiver of type Int?
+
+fun foo(arg: Int?) {
+    arg?.hashCode()<caret>.toLong()
+}

idea/testData/quickfix/surroundWithNullCheck/expressionUnsafeCall.kt

@@ -0,0 +1,8 @@
+// "Surround with null check" "false"
+// ACTION: Add non-null asserted (!!) call
+// ACTION: Convert to block body
+// ACTION: Introduce local variable
+// ACTION: Replace with safe (?.) call
+// ERROR: Only safe (?.) or non-null asserted (!!.) calls are allowed on a nullable receiver of type Int?
+
+fun foo(arg: Int?) = arg<caret>.hashCode()

idea/testData/quickfix/surroundWithNullCheck/infixUnsafeCall.kt

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+infix fun Int.op(arg: Int) = this
+
+fun foo(arg: Int?) {
+    arg <caret>op 42
+}

idea/testData/quickfix/surroundWithNullCheck/infixUnsafeCall.kt.after

@@ -0,0 +1,9 @@
+// "Surround with null check" "true"
+
+infix fun Int.op(arg: Int) = this
+
+fun foo(arg: Int?) {
+    if (arg != null) {
+        arg op 42
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/invokeFuncUnsafe.kt

@@ -0,0 +1,5 @@
+// "Surround with null check" "true"
+
+fun foo(exec: (() -> Unit)?) {
+    <caret>exec()
+}

idea/testData/quickfix/surroundWithNullCheck/invokeFuncUnsafe.kt.after

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+fun foo(exec: (() -> Unit)?) {
+    if (exec != null) {
+        exec()
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/invokeUnsafe.kt

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+operator fun Int.invoke() = this
+
+fun foo(arg: Int?) {
+    <caret>arg()
+}

idea/testData/quickfix/surroundWithNullCheck/invokeUnsafe.kt.after

@@ -0,0 +1,9 @@
+// "Surround with null check" "true"
+
+operator fun Int.invoke() = this
+
+fun foo(arg: Int?) {
+    if (arg != null) {
+        arg()
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/simpleUnsafeCall.kt

@@ -0,0 +1,5 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?) {
+    arg<caret>.hashCode()
+}

idea/testData/quickfix/surroundWithNullCheck/simpleUnsafeCall.kt.after

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?) {
+    if (arg != null) {
+        arg.hashCode()
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInBinary.kt

@@ -0,0 +1,5 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?) {
+    42 + arg<caret>.hashCode() - 13
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInBinary.kt.after

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?) {
+    if (arg != null) {
+        42 + arg.hashCode() - 13
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInDeclaration.kt

@@ -0,0 +1,8 @@
+// "Surround with null check" "false"
+// ACTION: Add non-null asserted (!!) call
+// ACTION: Replace with safe (?.) call
+// ERROR: Only safe (?.) or non-null asserted (!!.) calls are allowed on a nullable receiver of type String?
+
+fun foo(s: String?) {
+    val x = s<caret>.hashCode()
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInGetter.kt

@@ -0,0 +1,11 @@
+// "Surround with null check" "false"
+// ACTION: Add non-null asserted (!!) call
+// ACTION: Convert to block body
+// ACTION: Introduce local variable
+// ACTION: Replace with safe (?.) call
+// ERROR: Only safe (?.) or non-null asserted (!!.) calls are allowed on a nullable receiver of type String?
+
+class My(val x: String?) {
+    val y: Int
+        get() = x<caret>.hashCode()
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideAnonymous.kt

@@ -0,0 +1,12 @@
+// "Surround with null check" "false"
+// WITH_RUNTIME
+// ACTION: Add 'block =' to argument
+// ACTION: Add non-null asserted (!!) call
+// ACTION: Convert to block body
+// ACTION: Introduce local variable
+// ACTION: Replace with safe (?.) call
+// ERROR: Only safe (?.) or non-null asserted (!!.) calls are allowed on a nullable receiver of type Int?
+
+fun foo(arg: Int?) {
+    run(fun() = arg<caret>.hashCode())
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideIf.kt

@@ -0,0 +1,5 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?, flag: Boolean) {
+    if (flag) arg<caret>.hashCode()
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideIf.kt.after

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?, flag: Boolean) {
+    if (flag) if (arg != null) {
+        arg.hashCode()
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideWhen.kt

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?, flag: Boolean) {
+    when (flag) {
+        true -> arg<caret>.hashCode()
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideWhen.kt.after

@@ -0,0 +1,9 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?, flag: Boolean) {
+    when (flag) {
+        true -> if (arg != null) {
+            arg.hashCode()
+        }
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideWhile.kt

@@ -0,0 +1,5 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?, flag: Boolean) {
+    while (flag) arg<caret>.hashCode()
+}

idea/testData/quickfix/surroundWithNullCheck/unsafeCallInsideWhile.kt.after

@@ -0,0 +1,7 @@
+// "Surround with null check" "true"
+
+fun foo(arg: Int?, flag: Boolean) {
+    while (flag) if (arg != null) {
+        arg.hashCode()
+    }
+}

idea/testData/quickfix/surroundWithNullCheck/unstableValue.kt

@@ -0,0 +1,12 @@
+// "Surround with null check" "false"
+// ACTION: Introduce local variable
+// ACTION: Add non-null asserted (!!) call
+// ACTION: Replace with safe (?.) call
+// ERROR: Only safe (?.) or non-null asserted (!!.) calls are allowed on a nullable receiver of type Int?
+
+class My(var x: Int?) {
+
+    fun foo() {
+        x<caret>.hashCode()
+    }
+}