[FIR] Add deserialization of inheritors of sealed classes

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/deserialization/ClassDeserialization.kt

@@ -51,7 +51,8 @@ fun deserializeClassToSymbol(
         isData = Flags.IS_DATA.get(classProto.flags)
         isInline = Flags.IS_INLINE_CLASS.get(classProto.flags)
     }
-    val firClass = if (modality == Modality.SEALED) {
+    val isSealed = modality == Modality.SEALED
+    val firClass = if (isSealed) {
         FirSealedClassImpl(
             null,
             session,
@@ -131,6 +132,11 @@ fun deserializeClassToSymbol(
                 symbol.fir
             }
         )
+
+        if (isSealed) {
+            classProto.sealedSubclassFqNameList.mapTo((firClass as FirSealedClassImpl).inheritors) {
+                ClassId.fromString(nameResolver.getQualifiedClassName(it))
+            }
+        }
     }
 }
-

compiler/fir/resolve/testData/loadCompiledKotlin/class/SealedClass.txt

@@ -0,0 +1,20 @@
+public final class Inheritor2 : R|test/SealedClass| {
+    public constructor(): R|test/Inheritor2|
+
+}
+
+public final class Inheritor3 : R|test/SealedClass| {
+    public constructor(): R|test/Inheritor3|
+
+}
+
+public sealed class SealedClass : R|kotlin/Any| {
+    private constructor(): R|test/SealedClass|
+
+    public final class Inheritor1 : R|test/SealedClass| {
+        public constructor(): R|test/SealedClass.Inheritor1|
+
+    }
+
+}
+

compiler/fir/resolve/tests/org/jetbrains/kotlin/fir/FirLoadCompiledKotlinGenerated.java

@@ -690,6 +690,11 @@ public class FirLoadCompiledKotlinGenerated extends AbstractFirLoadCompiledKotli
             runTest("compiler/testData/loadJava/compiledKotlin/class/RecursiveGeneric.kt");
         }
 
+        @TestMetadata("SealedClass.kt")
+        public void testSealedClass() throws Exception {
+            runTest("compiler/testData/loadJava/compiledKotlin/class/SealedClass.kt");
+        }
+
         @TestMetadata("SingleAbstractMethod.kt")
         public void testSingleAbstractMethod() throws Exception {
             runTest("compiler/testData/loadJava/compiledKotlin/class/SingleAbstractMethod.kt");

compiler/testData/loadJava/compiledKotlin/class/SealedClass.kt

@@ -0,0 +1,9 @@
+package test
+
+class Inheritor3 : SealedClass()
+
+sealed class SealedClass {
+    class Inheritor1 : SealedClass()
+}
+
+class Inheritor2 : SealedClass()

compiler/testData/loadJava/compiledKotlin/class/SealedClass.txt

@@ -0,0 +1,17 @@
+package test
+
+public final class Inheritor2 : test.SealedClass {
+    /*primary*/ public constructor Inheritor2()
+}
+
+public final class Inheritor3 : test.SealedClass {
+    /*primary*/ public constructor Inheritor3()
+}
+
+public sealed class SealedClass {
+    /*primary*/ private constructor SealedClass()
+
+    public final class Inheritor1 : test.SealedClass {
+        /*primary*/ public constructor Inheritor1()
+    }
+}

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

@@ -2381,6 +2381,11 @@ public class LoadJavaTestGenerated extends AbstractLoadJavaTest {
                 runTest("compiler/testData/loadJava/compiledKotlin/class/RecursiveGeneric.kt");
             }
 
+            @TestMetadata("SealedClass.kt")
+            public void testSealedClass() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/class/SealedClass.kt");
+            }
+
             @TestMetadata("SingleAbstractMethod.kt")
             public void testSingleAbstractMethod() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/class/SingleAbstractMethod.kt");

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

@@ -690,6 +690,11 @@ public class LoadKotlinWithTypeTableTestGenerated extends AbstractLoadKotlinWith
             runTest("compiler/testData/loadJava/compiledKotlin/class/RecursiveGeneric.kt");
         }
 
+        @TestMetadata("SealedClass.kt")
+        public void testSealedClass() throws Exception {
+            runTest("compiler/testData/loadJava/compiledKotlin/class/SealedClass.kt");
+        }
+
         @TestMetadata("SingleAbstractMethod.kt")
         public void testSingleAbstractMethod() throws Exception {
             runTest("compiler/testData/loadJava/compiledKotlin/class/SingleAbstractMethod.kt");

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

@@ -2381,6 +2381,11 @@ public class IrLoadJavaTestGenerated extends AbstractIrLoadJavaTest {
                 runTest("compiler/testData/loadJava/compiledKotlin/class/RecursiveGeneric.kt");
             }
 
+            @TestMetadata("SealedClass.kt")
+            public void testSealedClass() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/class/SealedClass.kt");
+            }
+
             @TestMetadata("SingleAbstractMethod.kt")
             public void testSingleAbstractMethod() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/class/SingleAbstractMethod.kt");

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

@@ -2381,6 +2381,11 @@ public class LoadJavaUsingJavacTestGenerated extends AbstractLoadJavaUsingJavacT
                 runTest("compiler/testData/loadJava/compiledKotlin/class/RecursiveGeneric.kt");
             }
 
+            @TestMetadata("SealedClass.kt")
+            public void testSealedClass() throws Exception {
+                runTest("compiler/testData/loadJava/compiledKotlin/class/SealedClass.kt");
+            }
+
             @TestMetadata("SingleAbstractMethod.kt")
             public void testSingleAbstractMethod() throws Exception {
                 runTest("compiler/testData/loadJava/compiledKotlin/class/SingleAbstractMethod.kt");