1 files changed, 141 insertions, 0 deletions
diff --git a/kotlinx-coroutines-debug/test/StartModeProbesTest.kt b/kotlinx-coroutines-debug/test/StartModeProbesTest.kt
new file mode 100644
index 00000000..a0297da3
--- /dev/null
+++ b/kotlinx-coroutines-debug/test/StartModeProbesTest.kt
@@ -0,0 +1,141 @@
+/*
+ * Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.coroutines.debug
+
+import kotlinx.coroutines.*
+import org.junit.*
+import org.junit.Test
+import kotlin.test.*
+
+class StartModeProbesTest : DebugTestBase() {
+
+    @Test
+    fun testUndispatched() = runTest {
+        expect(1)
+        val job = launch(start = CoroutineStart.UNDISPATCHED) {
+            expect(2)
+            undispatchedSleeping()
+            assertTrue(true)
+        }
+
+        yield()
+        expect(3)
+        verifyPartialDump(2, "StartModeProbesTest.undispatchedSleeping")
+        job.cancelAndJoin()
+        verifyPartialDump(1, "StartModeProbesTest\$testUndispatched")
+        finish(4)
+    }
+
+    private suspend fun undispatchedSleeping() {
+        delay(Long.MAX_VALUE)
+        assertTrue(true)
+    }
+
+    @Test
+    fun testWithTimeoutWithUndispatched() = runTest {
+        expect(1)
+        val job = launchUndispatched()
+
+        yield()
+        expect(3)
+        verifyPartialDump(
+            2,
+            "StartModeProbesTest\$launchUndispatched\$1.invokeSuspend",
+            "StartModeProbesTest.withTimeoutHelper",
+            "StartModeProbesTest\$withTimeoutHelper\$2.invokeSuspend"
+        )
+        job.cancelAndJoin()
+        verifyPartialDump(1, "StartModeProbesTest\$testWithTimeoutWithUndispatched")
+        finish(4)
+    }
+
+    private fun CoroutineScope.launchUndispatched(): Job {
+        return launch(start = CoroutineStart.UNDISPATCHED) {
+            withTimeoutHelper()
+            assertTrue(true)
+        }
+    }
+
+    private suspend fun withTimeoutHelper() {
+        withTimeout(Long.MAX_VALUE) {
+            expect(2)
+            delay(Long.MAX_VALUE)
+        }
+
+        assertTrue(true)
+    }
+
+    @Test
+    fun testWithTimeout() = runTest {
+        withTimeout(Long.MAX_VALUE) {
+            testActiveDump(
+                false,
+                "StartModeProbesTest\$testWithTimeout\$1.invokeSuspend",
+                "state: RUNNING"
+            )
+        }
+    }
+
+    @Test
+    fun testWithTimeoutAfterYield() = runTest {
+        withTimeout(Long.MAX_VALUE) {
+            testActiveDump(
+                true,
+                "StartModeProbesTest\$testWithTimeoutAfterYield\$1.invokeSuspend",
+                "StartModeProbesTest\$testWithTimeoutAfterYield\$1\$1.invokeSuspend",
+                "StartModeProbesTest.testActiveDump",
+                "state: RUNNING"
+            )
+        }
+    }
+
+    private suspend fun testActiveDump(shouldYield: Boolean, vararg expectedFrames: String) {
+        if (shouldYield) yield()
+        verifyPartialDump(1, *expectedFrames)
+        assertTrue(true)
+    }
+
+    @Test
+    fun testWithTailCall() = runTest {
+        expect(1)
+        val job = tailCallMethod()
+        yield()
+        expect(3)
+        verifyPartialDump(2, "StartModeProbesTest\$launchFromTailCall\$2")
+        job.cancelAndJoin()
+        verifyPartialDump(1, "StartModeProbesTest\$testWithTailCall")
+        finish(4)
+    }
+
+    private suspend fun CoroutineScope.tailCallMethod(): Job = launchFromTailCall()
+    private suspend fun CoroutineScope.launchFromTailCall(): Job = launch {
+        expect(2)
+        delay(Long.MAX_VALUE)
+    }
+
+    @Test
+    fun testCoroutineScope() = runTest {
+        expect(1)
+        val job = launch(start = CoroutineStart.UNDISPATCHED) {
+            runScope()
+        }
+
+        yield()
+        expect(3)
+        verifyPartialDump(
+            2,
+            "StartModeProbesTest\$runScope\$2.invokeSuspend",
+            "StartModeProbesTest\$testCoroutineScope\$1\$job\$1.invokeSuspend")
+        job.cancelAndJoin()
+        finish(4)
+    }
+
+    private suspend fun runScope() {
+        coroutineScope {
+            expect(2)
+            delay(Long.MAX_VALUE)
+        }
+    }
+}