/
CancellableContinuationJvmTest.kt
82 lines (69 loc) · 2.08 KB
/
CancellableContinuationJvmTest.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
* Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
package kotlinx.coroutines
import kotlin.coroutines.*
import kotlin.test.*
class CancellableContinuationJvmTest : TestBase() {
@Test
fun testToString() = runTest {
checkToString()
}
private suspend fun checkToString() {
suspendCancellableCoroutine<Unit> {
it.resume(Unit)
assertTrue(it.toString().contains("kotlinx.coroutines.CancellableContinuationJvmTest.checkToString(CancellableContinuationJvmTest.kt"))
}
suspend {}() // Eliminate tail-call optimization
}
@Test
fun testExceptionIsNotReported() = runTest({ it is CancellationException }) {
val ctx = coroutineContext
suspendCancellableCoroutine<Unit> {
ctx.cancel()
it.resumeWith(Result.failure(TestException()))
}
}
@Test
fun testBlockingIntegration() = runTest {
val source = BlockingSource()
val job = launch(Dispatchers.Default) {
source.await()
}
source.cancelAndJoin(job)
}
@Test
fun testBlockingIntegrationAlreadyCancelled() = runTest {
val source = BlockingSource()
val job = launch(Dispatchers.Default) {
cancel()
source.await()
}
source.cancelAndJoin(job)
}
private suspend fun BlockingSource.cancelAndJoin(job: Job) {
while (!hasSubscriber) {
Thread.sleep(10)
}
job.cancelAndJoin()
}
private suspend fun BlockingSource.await() = suspendCancellableCoroutine<Unit> {
it.invokeOnCancellation { this.cancel() }
subscribe()
}
private class BlockingSource {
@Volatile
private var isCancelled = false
@Volatile
public var hasSubscriber = false
public fun subscribe() {
hasSubscriber = true
while (!isCancelled) {
Thread.sleep(10)
}
}
public fun cancel() {
isCancelled = true
}
}
}