JS: add support of interceptResume function in coroutine controller

2016-12-05 15:30:04 +03:00
parent 9eff193ea4
commit 060f08a8dd
11 changed files with 132 additions and 18 deletions
@@ -0,0 +1,53 @@
+// NO_INTERCEPT_RESUME_TESTS
+
+class Controller {
+    var log = ""
+    var resumeIndex = 0
+
+    suspend fun <T> suspendWithValue(value: T): T = suspendWithCurrentContinuation { continuation ->
+        log += "suspend($value);"
+        continuation.resume(value)
+        Suspend
+    }
+
+    suspend fun suspendWithException(value: String): Unit = suspendWithCurrentContinuation { continuation ->
+        log += "error($value);"
+        continuation.resumeWithException(RuntimeException(value))
+        Suspend
+    }
+
+    operator fun interceptResume(block: () -> Unit) {
+        var id = resumeIndex++
+        log += "before $id;"
+        block()
+        log += "after $id;"
+    }
+}
+
+fun test(coroutine c: Controller.() -> Continuation<Unit>): String {
+    val controller = Controller()
+    c(controller).resume(Unit)
+    return controller.log
+}
+
+fun box(): String {
+    var result = test {
+        val o = suspendWithValue("O")
+        val k = suspendWithValue("K")
+        log += "$o$k;"
+    }
+    if (result != "before 0;suspend(O);before 1;suspend(K);before 2;OK;after 2;after 1;after 0;") return "fail1: $result"
+
+    result = test {
+        try {
+            suspendWithException("OK")
+            log += "ignore;"
+        }
+        catch (e: RuntimeException) {
+            log += "${e.message};"
+        }
+    }
+    if (result != "before 0;error(OK);before 1;OK;after 1;after 0;") return "fail2: $result"
+
+    return "OK"
+}