add tests for mockk#832

aSemy · Jul 25, 2022 · 7e7ddd6 · 7e7ddd6
1 parent 54e30ec
commit 7e7ddd6
Show file tree

Hide file tree

Showing 2 changed files with 98 additions and 0 deletions.
diff --git a/mockk/common/src/test/kotlin/io/mockk/it/SealedClassTest.kt b/mockk/common/src/test/kotlin/io/mockk/it/SealedClassTest.kt
@@ -0,0 +1,49 @@
+package io.mockk.it
+
+import io.mockk.every
+import io.mockk.mockk
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+
+class SealedClassTest {
+
+    @Test
+    fun serviceReturnsSealedClassImpl() {
+        val factory = mockk<Factory> {
+            every { create() } returns Leaf(1)
+        }
+
+        val result = factory.create()
+
+        assertEquals(Leaf(1), result)
+    }
+
+    @Test
+    fun serviceAnswersSealedClassImpl() {
+        val factory = mockk<Factory> {
+            every { create() } answers { Leaf(1) }
+        }
+
+        val result = factory.create()
+
+        assertEquals(Leaf(1), result)
+    }
+
+    companion object {
+
+        sealed class Node
+
+        data class Root(val id: Int) : Node()
+        data class Leaf(val id: Int) : Node()
+
+        interface Factory {
+            fun create(): Node
+        }
+
+        class FactoryImpl : Factory {
+            override fun create(): Node = Root(0)
+        }
+
+    }
+}
diff --git a/mockk/common/src/test/kotlin/io/mockk/it/SealedInterfaceTest.kt b/mockk/common/src/test/kotlin/io/mockk/it/SealedInterfaceTest.kt
@@ -0,0 +1,49 @@
+package io.mockk.it
+
+import io.mockk.every
+import io.mockk.mockk
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+
+class SealedInterfaceTest {
+
+    @Test
+    fun serviceReturnsSealedClassImpl() {
+        val factory = mockk<Factory> {
+            every { create() } returns Leaf(1)
+        }
+
+        val result = factory.create()
+
+        assertEquals(Leaf(1), result)
+    }
+
+    @Test
+    fun serviceAnswersSealedClassImpl() {
+        val factory = mockk<Factory> {
+            every { create() } answers { Leaf(1) }
+        }
+
+        val result = factory.create()
+
+        assertEquals(Leaf(1), result)
+    }
+
+    companion object {
+
+        sealed interface Node
+
+        data class Root(val id: Int) : Node
+        data class Leaf(val id: Int) : Node
+
+        interface Factory {
+            fun create(): Node
+        }
+
+        class FactoryImpl : Factory {
+            override fun create(): Node = Root(0)
+        }
+
+    }
+}