blob: ae05fb8d743410e1e54f234a0d5d4eab21226188 (
plain)
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* Copyright 2016-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
package kotlinx.coroutines.channels
import kotlinx.coroutines.*
import kotlinx.coroutines.internal.*
import kotlinx.coroutines.selects.*
import kotlin.test.*
/**
* Tests for failures inside `onUndeliveredElement` handler in [Channel].
*/
class ChannelUndeliveredElementFailureTest : TestBase() {
private val item = "LOST"
private val onCancelFail: (String) -> Unit = { throw TestException(it) }
private val shouldBeUnhandled: List<(Throwable) -> Boolean> = listOf({ it.isElementCancelException() })
private fun Throwable.isElementCancelException() =
this is UndeliveredElementException && cause is TestException && cause!!.message == item
@Test
fun testSendCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
channel.send(item)
expectUnreached()
}
job.cancel()
}
@Test
fun testSendSelectCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
select {
channel.onSend(item) {
expectUnreached()
}
}
}
job.cancel()
}
@Test
fun testReceiveCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
channel.receive()
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testReceiveSelectCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
select<Unit> {
channel.onReceive {
expectUnreached()
}
}
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testReceiveCatchingCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
channel.receiveCatching()
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testReceiveOrClosedSelectCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
select<Unit> {
channel.onReceiveCatching {
expectUnreached()
}
}
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testHasNextCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
channel.iterator().hasNext()
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testChannelCancelledFail() = runTest(expected = { it.isElementCancelException()}) {
val channel = Channel(1, onUndeliveredElement = onCancelFail)
channel.send(item)
channel.cancel()
expectUnreached()
}
}
|