~ Better checks in testRejectOnResumeInContext

Kotlin · Sep 14, 2020 · fda826c · fda826c
1 parent f09fc9d
commit fda826c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/kotlinx-coroutines-core/jvm/test/RejectedExecutionTest.kt b/kotlinx-coroutines-core/jvm/test/RejectedExecutionTest.kt
@@ -67,16 +67,18 @@ class RejectedExecutionTest : TestBase() {
                     try {
                         withContext(Dispatchers.Default) {
                             expect(3)
+                            assertDefaultDispatcherThread()
                         }
                         // cancelled on resume back
                     } finally {
+                        expect(4)
                         assertIoThread()
                     }
                     expectUnreached()
                 }
         }
         assertEquals(2, executor.submittedTasks)
-        finish(4)
+        finish(5)
     }
 
     @Test
@@ -135,6 +137,12 @@ class RejectedExecutionTest : TestBase() {
         if (!thread.name.startsWith(threadName)) error("Not an executor thread: $thread")
     }
 
+    private fun assertDefaultDispatcherThread() {
+        val thread = Thread.currentThread()
+        if (thread !is CoroutineScheduler.Worker) error("Not a thread from Dispatchers.Default: $thread")
+        assertEquals(CoroutineScheduler.WorkerState.CPU_ACQUIRED, thread.state)
+    }
+
     private fun assertIoThread() {
         val thread = Thread.currentThread()
         if (thread !is CoroutineScheduler.Worker) error("Not a thread from Dispatchers.IO: $thread")