FIR IDE: Use KtFirExpressionTypeProvider.getExpectedType() in

ReplaceCallFix.

idea/testData/quickfix/replaceWithSafeCall/assignmentToPropertyWithNoExplicitType.kt

@@ -0,0 +1,6 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+class T(s: String?) {
+    var i = s<caret>.length
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/assignmentToPropertyWithNoExplicitType.kt.after

@@ -0,0 +1,6 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+class T(s: String?) {
+    var i = s?.length
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/assignmentToPropertyWithNoExplicitType.kt.fir.after

@@ -0,0 +1,6 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+class T(s: String?) {
+    var i = s?.length ?: <caret>
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/comment.kt

@@ -5,5 +5,4 @@ fun foo(a: String?) {
             // comment2
             .<caret>length
 }
-// FIR_IDENTICAL
-/* FIR_COMPARISON */
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/comment.kt.after

@@ -5,5 +5,4 @@ fun foo(a: String?) {
         // comment2
         ?.length
 }
-// FIR_IDENTICAL
-/* FIR_COMPARISON */
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/comment.kt.fir.after

@@ -0,0 +1,8 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+fun foo(a: String?) {
+    val b = a // comment1
+        // comment2
+        ?.length ?: <caret>
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/functionExpressionBody.kt

@@ -0,0 +1,9 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+// ERROR: Type mismatch: inferred type is Int? but Int was expected
+
+// Note: There should be no error in FIR but errors are not currently checked for FIR
+class T {
+    fun foo(s: String?): Int = s<caret>.length
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/functionExpressionBody.kt.after

@@ -0,0 +1,9 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+// ERROR: Type mismatch: inferred type is Int? but Int was expected
+
+// Note: There should be no error in FIR but errors are not currently checked for FIR
+class T {
+    fun foo(s: String?): Int = s?.length
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/functionExpressionBody.kt.fir.after

@@ -0,0 +1,9 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+// ERROR: Type mismatch: inferred type is Int? but Int was expected
+
+// Note: There should be no error in FIR but errors are not currently checked for FIR
+class T {
+    fun foo(s: String?): Int = s?.length ?: <caret>
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/functionExpressionBodyWithNoExplicitType.kt

@@ -0,0 +1,6 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+class T {
+    fun foo(s: String?) = s<caret>.length
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/functionExpressionBodyWithNoExplicitType.kt.after

@@ -0,0 +1,6 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+class T {
+    fun foo(s: String?) = s?.length
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/functionExpressionBodyWithNoExplicitType.kt.fir.after

@@ -0,0 +1,6 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+class T {
+    fun foo(s: String?) = s?.length ?: <caret>
+}
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/lineBreak.kt

@@ -4,5 +4,4 @@ fun foo(a: String?) {
     val b = a
             .<caret>length
 }
-// FIR_IDENTICAL
-/* FIR_COMPARISON */
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/lineBreak.kt.after

@@ -4,5 +4,4 @@ fun foo(a: String?) {
     val b = a
         ?.length
 }
-// FIR_IDENTICAL
-/* FIR_COMPARISON */
+/* FIR_COMPARISON */

idea/testData/quickfix/replaceWithSafeCall/lineBreak.kt.fir.after

@@ -0,0 +1,7 @@
+// "Replace with safe (?.) call" "true"
+// WITH_RUNTIME
+fun foo(a: String?) {
+    val b = a
+        ?.length ?: <caret>
+}
+/* FIR_COMPARISON */