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
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*
* 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 CancellableContinuationHandlersTest : TestBase() {
@Test
fun testDoubleSubscription() = runTest({ it is IllegalStateException }) {
suspendCancellableCoroutine<Unit> { c ->
c.invokeOnCancellation { finish(1) }
c.invokeOnCancellation { expectUnreached() }
}
}
@Test
fun testDoubleSubscriptionAfterCompletion() = runTest {
suspendCancellableCoroutine<Unit> { c ->
c.resume(Unit)
// Nothing happened
c.invokeOnCancellation { expectUnreached() }
c.invokeOnCancellation { expectUnreached() }
}
}
@Test
fun testDoubleSubscriptionAfterCancellation() = runTest {
try {
suspendCancellableCoroutine<Unit> { c ->
c.cancel()
c.invokeOnCancellation {
assertTrue(it is CancellationException)
expect(1)
}
c.invokeOnCancellation {
assertTrue(it is CancellationException)
expect(2)
}
}
} catch (e: CancellationException) {
finish(3)
}
}
@Test
fun testDoubleSubscriptionAfterCancellationWithCause() = runTest {
try {
suspendCancellableCoroutine<Unit> { c ->
c.cancel(AssertionError())
c.invokeOnCancellation {
require(it is AssertionError)
expect(1)
}
c.invokeOnCancellation {
require(it is AssertionError)
expect(2)
}
}
} catch (e: AssertionError) {
finish(3)
}
}
@Test
fun testDoubleSubscriptionMixed() = runTest {
try {
suspendCancellableCoroutine<Unit> { c ->
c.invokeOnCancellation {
require(it is IndexOutOfBoundsException)
expect(1)
}
c.cancel(IndexOutOfBoundsException())
c.invokeOnCancellation {
require(it is IndexOutOfBoundsException)
expect(2)
}
}
} catch (e: IndexOutOfBoundsException) {
finish(3)
}
}
@Test
fun testExceptionInHandler() = runTest({it is CompletionHandlerException}) {
suspendCancellableCoroutine<Unit> { c ->
c.invokeOnCancellation { throw AssertionError() }
c.cancel()
}
}
}
|
