Fix race in tests

Kotlin · May 27, 2019 · ad0d42d · ad0d42d
1 parent c34c1d5
commit ad0d42d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kotlinx-coroutines-core/jvm/test/flow/CallbackFlowTest.kt b/kotlinx-coroutines-core/jvm/test/flow/CallbackFlowTest.kt
@@ -41,7 +41,7 @@ class CallbackFlowTest : TestBase() {
 
         val flow = channelFlow<Int>(16) {
             api.start(channel)
-            channel.invokeOnClose {
+            awaitClose {
                 api.stop()
             }
         }
@@ -85,7 +85,7 @@ class CallbackFlowTest : TestBase() {
 
         val flow = channelFlow<Int> {
             api.start(channel)
-            channel.invokeOnClose {
+            awaitClose {
                 api.stop()
             }
         }