Fix race in testOnSendCancelled

Kotlin · Oct 4, 2021 · c806d9e · c806d9e
1 parent 9c516a0
commit c806d9e
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/reactive/kotlinx-coroutines-reactive/test/PublishTest.kt b/reactive/kotlinx-coroutines-reactive/test/PublishTest.kt
@@ -296,12 +296,10 @@ class PublishTest : TestBase() {
             expect(2)
             // Collector is ready
             send(1)
-            expect(3)
             try {
                 send(2)
                 expectUnreached()
             } catch (e: CancellationException) {
-                expect(7)
                 // publisher cancellation is async
                 latch.countDown()
                 throw e
@@ -312,15 +310,14 @@ class PublishTest : TestBase() {
         val collectorLatch = Mutex(true)
         val job = launch {
             published.asFlow().buffer(0).collect {
-                expect(4)
                 collectorLatch.unlock()
-                hang { expect(6) }
+                hang { expect(4) }
             }
         }
         collectorLatch.lock()
-        expect(5)
+        expect(3)
         job.cancelAndJoin()
         latch.await()
-        finish(8)
+        finish(5)
     }
 }