From 2eaf22b89aa26de4e9f811588a0f87874f934b58 Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Thu, 23 May 2024 21:47:05 +0200
Subject: [PATCH] [kotlin] k2 extract function: test data for fixed return type
 approximation

^KTIJ-29779 fixed

GitOrigin-RevId: d468599b4b5052f906634038d5ff2b86d115a631
---
 .../introduce/ExtractionTestGenerated.java           |  5 +++++
 .../extractFunction/basic/approximatedReturnType.kt  |  8 ++++++++
 .../basic/approximatedReturnType.kt.after            | 12 ++++++++++++
 .../introduce/K2IntroduceFunctionTestGenerated.java  |  5 +++++
 4 files changed, 30 insertions(+)
 create mode 100644 plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt
 create mode 100644 plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt.after

diff --git a/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/introduce/ExtractionTestGenerated.java b/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/introduce/ExtractionTestGenerated.java
index d23d963f8b60..ab9d93b3e1ea 100644
--- a/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/introduce/ExtractionTestGenerated.java
+++ b/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/introduce/ExtractionTestGenerated.java
@@ -868,6 +868,11 @@ public abstract class ExtractionTestGenerated extends AbstractExtractionTest {
                 KotlinTestUtils.runTest(this::doExtractFunctionTest, this, testDataFilePath);
             }
 
+            @TestMetadata("approximatedReturnType.kt")
+            public void testApproximatedReturnType() throws Exception {
+                runTest("testData/refactoring/extractFunction/basic/approximatedReturnType.kt");
+            }
+
             @TestMetadata("callWithLambdaArgument.kt")
             public void testCallWithLambdaArgument() throws Exception {
                 runTest("testData/refactoring/extractFunction/basic/callWithLambdaArgument.kt");
diff --git a/plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt b/plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt
new file mode 100644
index 000000000000..6e342eede0ef
--- /dev/null
+++ b/plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt
@@ -0,0 +1,8 @@
+class SimpleClass
+
+fun findClasses() {
+    val classloader = Alias::class.java.classLoader
+    val obj = <selection>classloader?.loadClass("")?.kotlin</selection>
+}
+
+// IGNORE_K1
\ No newline at end of file
diff --git a/plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt.after b/plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt.after
new file mode 100644
index 000000000000..49f273995d33
--- /dev/null
+++ b/plugins/kotlin/idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt.after
@@ -0,0 +1,12 @@
+import kotlin.reflect.KClass
+
+class SimpleClass
+
+fun findClasses() {
+    val classloader = Alias::class.java.classLoader
+    val obj = klass(classloader)
+}
+
+private fun klass(classloader: ClassLoader?): KClass<out Any>? = classloader?.loadClass("")?.kotlin
+
+// IGNORE_K1
\ No newline at end of file
diff --git a/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/introduce/K2IntroduceFunctionTestGenerated.java b/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/introduce/K2IntroduceFunctionTestGenerated.java
index fee7b8ece058..9c0140a12243 100644
--- a/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/introduce/K2IntroduceFunctionTestGenerated.java
+++ b/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/introduce/K2IntroduceFunctionTestGenerated.java
@@ -26,6 +26,11 @@ public abstract class K2IntroduceFunctionTestGenerated extends AbstractK2Introdu
             KotlinTestUtils.runTest(this::doExtractFunctionTest, this, testDataFilePath);
         }
 
+        @TestMetadata("approximatedReturnType.kt")
+        public void testApproximatedReturnType() throws Exception {
+            runTest("../../idea/tests/testData/refactoring/extractFunction/basic/approximatedReturnType.kt");
+        }
+
         @TestMetadata("callWithLambdaArgument.kt")
         public void testCallWithLambdaArgument() throws Exception {
             runTest("../../idea/tests/testData/refactoring/extractFunction/basic/callWithLambdaArgument.kt");