blob: d2ef3d2691682f3dd765b418a69a72e39691fdd1 (
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
/*
* Copyright 2016-2020 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 testReceiveOrNullCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
channel.receiveOrNull()
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testReceiveOrNullSelectCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
select<Unit> {
channel.onReceiveOrNull {
expectUnreached()
}
}
expectUnreached() // will be cancelled before it dispatches
}
channel.send(item)
job.cancel()
}
@Test
fun testReceiveOrClosedCancelledFail() = runTest(unhandled = shouldBeUnhandled) {
val channel = Channel(onUndeliveredElement = onCancelFail)
val job = launch(start = CoroutineStart.UNDISPATCHED) {
channel.receiveOrClosed()
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.onReceiveOrClosed {
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()
}
}
|
