suspending block capture test

mfwgenerics · Jan 7, 2023 · b153740 · b153740
1 parent d5d8d3b
commit b153740
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/testing/build.gradle.kts b/testing/build.gradle.kts
@@ -16,6 +16,7 @@ application {
 
 dependencies {
     testImplementation(kotlin("test"))
+    testImplementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.6.4")
 }
 
 tasks.test {

diff --git a/testing/src/test/kotlin/CapturedBlockTests.kt b/testing/src/test/kotlin/CapturedBlockTests.kt
@@ -1,6 +1,9 @@
 import io.koalaql.kapshot.Capturable
 import io.koalaql.kapshot.CapturedBlock
+import kotlinx.coroutines.runBlocking
 import org.junit.jupiter.api.Test
+import kotlin.coroutines.resume
+import kotlin.coroutines.suspendCoroutine
 import kotlin.test.assertEquals
 
 class CapturedBlockTests {
@@ -104,4 +107,23 @@ i""",
             listOf(1, 2, 3).sourceyMap { it*2 }
         )
     }
+
+    fun interface SuspendTest<R>: Capturable<SuspendTest<R>> {
+        suspend operator fun invoke(): R
+
+        override fun withSource(source: String): SuspendTest<R> = object : SuspendTest<R> by this {
+            override fun source(): String = source
+        }
+    }
+
+    @Test
+    fun `user defined suspend capture`() {
+        suspend fun suspends(block: SuspendTest<Int>): String = "${block.source()}=${10 + block()})"
+
+        runBlocking {
+            assertEquals("suspendCoroutine { it.resume(32) }=42)", suspends {
+                suspendCoroutine { it.resume(32) }
+            })
+        }
+    }
 }