Add KClass.sealedSubclasses to get direct subclasses of sealed class

#KT-14657 Fixed
2018-08-03 13:23:53 +02:00
parent 1b1713a849
commit cbc92bc9a1
12 changed files with 95 additions and 0 deletions
@@ -0,0 +1,41 @@
+// IGNORE_BACKEND: JS, JS_IR, JVM_IR, NATIVE
+// WITH_REFLECT
+
+import kotlin.reflect.KClass
+import kotlin.reflect.jvm.*
+import kotlin.test.assertEquals
+
+// --
+
+sealed class SealedClassWithTopLevelSubclasses {
+    class NotASealedSubclass : TL2()
+}
+object TL1 : SealedClassWithTopLevelSubclasses()
+open class TL2 : SealedClassWithTopLevelSubclasses()
+
+// --
+
+sealed class SealedClassWithNestedSubclasses {
+    data class N1(val x: Unit) : SealedClassWithNestedSubclasses()
+    object N2 : SealedClassWithNestedSubclasses()
+}
+
+// --
+
+sealed class SealedClassWithNoSubclasses
+
+// --
+
+fun sealedSubclassNames(c: KClass<*>) = c.sealedSubclasses.map { it.simpleName ?: throw AssertionError("Unnamed class: ${it.java}") }.sorted()
+
+fun box(): String {
+    assertEquals(listOf("TL1", "TL2"), sealedSubclassNames(SealedClassWithTopLevelSubclasses::class))
+    assertEquals(listOf("N1", "N2"), sealedSubclassNames(SealedClassWithNestedSubclasses::class))
+    assertEquals(emptyList(), sealedSubclassNames(SealedClassWithNoSubclasses::class))
+
+    assertEquals(emptyList(), sealedSubclassNames(String::class))
+    assertEquals(emptyList(), sealedSubclassNames(Thread::class))
+    assertEquals(emptyList(), sealedSubclassNames(FloatArray::class))
+
+    return "OK"
+}