K2: Fix type-alias expansion during suspend function type serialization

When suspend function type is serialized, there is special logic that
adds Continuation parameter, before doing so, type-aliases has to be
expanded, attributes for resulting type should also derive from
expanded type

KT-53193, KT-54062

compiler/fir/analysis-tests/legacy-fir-tests/tests-gen/org/jetbrains/kotlin/fir/FirLoadCompiledKotlinGenerated.java

@@ -1000,6 +1000,11 @@ public class FirLoadCompiledKotlinGenerated extends AbstractFirLoadCompiledKotli
         public void testBasic() throws Exception {
             runTest("compiler/testData/loadJava/compiledKotlin/coroutines/Basic.kt");
         }
+
+        @TestMetadata("TypeAliasFTSuspendWithReceiver.kt")
+        public void testTypeAliasFTSuspendWithReceiver() throws Exception {
+            runTest("compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt");
+        }
     }
 
     @TestMetadata("compiler/testData/loadJava/compiledKotlin/dataClass")

compiler/fir/analysis-tests/testData/loadCompiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.txt

@@ -0,0 +1,9 @@
+public final fun foo(f: R|suspend test/Context.() -> kotlin/Unit|): R|kotlin/Unit|
+
+public final class Context : R|kotlin/Any| {
+    public constructor(): R|test/Context|
+
+}
+
+public final typealias SuspendWithContext = R|suspend test/Context.() -> kotlin/Unit|
+

compiler/fir/providers/src/org/jetbrains/kotlin/fir/types/InferenceUtils.kt

@@ -123,7 +123,7 @@ fun ConeKotlinType.suspendFunctionTypeToFunctionTypeWithContinuation(session: Fi
             isNullable = false
         ) + lastTypeArgument).toTypedArray(),
         isNullable = false,
-        attributes = attributes
+        attributes = fullyExpandedType.attributes
     )
 }
 

compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt

@@ -0,0 +1,9 @@
+// NO_CHECK_SOURCE_VS_BINARY
+// Type-alias info is lost during serialization in K1
+package test
+
+class Context
+
+typealias SuspendWithContext = suspend Context.() -> Unit
+
+fun foo(f: SuspendWithContext) {}

compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.txt

@@ -0,0 +1,8 @@
+package test
+
+public fun foo(/*0*/ f: suspend test.Context.() -> kotlin.Unit): kotlin.Unit
+
+public final class Context {
+    /*primary*/ public constructor Context()
+}
+public typealias SuspendWithContext = suspend test.Context.() -> kotlin.Unit

compiler/tests-gen/org/jetbrains/kotlin/jvm/compiler/LoadJavaTestGenerated.java

@@ -2678,6 +2678,11 @@ public class LoadJavaTestGenerated extends AbstractLoadJavaTest {
             public void testBasic() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/coroutines/Basic.kt");
             }
+
+            @TestMetadata("TypeAliasFTSuspendWithReceiver.kt")
+            public void testTypeAliasFTSuspendWithReceiver() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt");
+            }
         }
 
         @TestMetadata("compiler/testData/loadJava/compiledKotlin/dataClass")

compiler/tests-gen/org/jetbrains/kotlin/jvm/compiler/LoadKotlinWithTypeTableTestGenerated.java

@@ -1000,6 +1000,11 @@ public class LoadKotlinWithTypeTableTestGenerated extends AbstractLoadKotlinWith
         public void testBasic() throws Exception {
             runTest("compiler/testData/loadJava/compiledKotlin/coroutines/Basic.kt");
         }
+
+        @TestMetadata("TypeAliasFTSuspendWithReceiver.kt")
+        public void testTypeAliasFTSuspendWithReceiver() throws Exception {
+            runTest("compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt");
+        }
     }
 
     @TestMetadata("compiler/testData/loadJava/compiledKotlin/dataClass")

compiler/tests-gen/org/jetbrains/kotlin/jvm/compiler/ir/IrLoadJavaTestGenerated.java

@@ -2679,6 +2679,11 @@ public class IrLoadJavaTestGenerated extends AbstractIrLoadJavaTest {
             public void testBasic() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/coroutines/Basic.kt");
             }
+
+            @TestMetadata("TypeAliasFTSuspendWithReceiver.kt")
+            public void testTypeAliasFTSuspendWithReceiver() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt");
+            }
         }
 
         @TestMetadata("compiler/testData/loadJava/compiledKotlin/dataClass")

compiler/tests-gen/org/jetbrains/kotlin/jvm/compiler/javac/LoadJavaUsingJavacTestGenerated.java

@@ -2678,6 +2678,11 @@ public class LoadJavaUsingJavacTestGenerated extends AbstractLoadJavaUsingJavacT
             public void testBasic() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/coroutines/Basic.kt");
             }
+
+            @TestMetadata("TypeAliasFTSuspendWithReceiver.kt")
+            public void testTypeAliasFTSuspendWithReceiver() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt");
+            }
         }
 
         @TestMetadata("compiler/testData/loadJava/compiledKotlin/dataClass")

core/descriptors.runtime/tests/org/jetbrains/kotlin/jvm/runtime/JvmRuntimeDescriptorLoaderTestGenerated.java

@@ -1002,6 +1002,11 @@ public class JvmRuntimeDescriptorLoaderTestGenerated extends AbstractJvmRuntimeD
             public void testBasic() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/coroutines/Basic.kt");
             }
+
+            @TestMetadata("TypeAliasFTSuspendWithReceiver.kt")
+            public void testTypeAliasFTSuspendWithReceiver() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/coroutines/TypeAliasFTSuspendWithReceiver.kt");
+            }
         }
 
         @TestMetadata("compiler/testData/loadJava/compiledKotlin/dataClass")