test sourceOf on different declaration types

mfwgenerics · Jan 2, 2023 · d5d8d3b · d5d8d3b
1 parent 974cc9f
commit d5d8d3b
Showing 1 changed file with 60 additions and 0 deletions.
diff --git a/testing/src/test/kotlin/CaptureSourceTests.kt b/testing/src/test/kotlin/CaptureSourceTests.kt
@@ -79,4 +79,64 @@ class CaptureSourceTests {
             "fun five() = 5", sourceOf(Inner::five)
         )
     }
+
+    @CaptureSource
+    interface CapturedInterface {
+        fun type() = "interface"
+    }
+
+    @CaptureSource
+    object CapturedObj {
+        val test = "123"
+    }
+
+    @CaptureSource
+    sealed class CapturedSealed {
+        object Left: CapturedSealed()
+        object Right: CapturedSealed()
+
+        @CaptureSource
+        companion object { }
+    }
+
+    @Test
+    fun `capture assorted declarations`() {
+        assertEquals(
+            """
+            interface CapturedInterface {
+                fun type() = "interface"
+            }
+            """.trimIndent(),
+            sourceOf<CapturedInterface>()
+        )
+
+        assertEquals(
+            """
+            object CapturedObj {
+                val test = "123"
+            }
+            """.trimIndent(),
+            sourceOf<CapturedObj>()
+        )
+
+        assertEquals(
+            """
+            sealed class CapturedSealed {
+                object Left: CapturedSealed()
+                object Right: CapturedSealed()
+
+                @CaptureSource
+                companion object { }
+            }
+            """.trimIndent(),
+            sourceOf<CapturedSealed>()
+        )
+
+        assertEquals(
+            """
+                companion object { }
+            """.trimIndent(),
+            sourceOf<CapturedSealed.Companion>()
+        )
+    }
 }