-
Notifications
You must be signed in to change notification settings - Fork 1.8k
/
ChannelReceiveOrClosedTest.kt
141 lines (120 loc) · 4.23 KB
/
ChannelReceiveOrClosedTest.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
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
/*
* Copyright 2016-2018 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 kotlin.test.*
class ChannelReceiveOrClosedTest : TestBase() {
@Test
fun testChannelOfThrowables() = runTest {
val channel = Channel<Throwable>()
launch {
channel.send(TestException1())
channel.close(TestException2())
}
val element = channel.receiveOrClosed()
assertTrue(element.isValue)
assertTrue(element.value is TestException1)
assertTrue(element.valueOrNull is TestException1)
val closed = channel.receiveOrClosed()
assertTrue(closed.isClosed)
assertTrue(closed.closeCause is TestException2)
}
@Test
@Suppress("ReplaceAssertBooleanWithAssertEquality") // inline classes test
fun testNullableIntChanel() = runTest {
val channel = Channel<Int?>()
launch {
expect(2)
channel.send(1)
expect(3)
channel.send(null)
expect(6)
channel.close()
}
expect(1)
val element = channel.receiveOrClosed()
assertTrue(element.isValue)
assertEquals(1, element.value)
assertEquals(1, element.valueOrNull)
assertEquals("Value(1)", element.toString())
assertTrue(ValueOrClosed.value(1) == element) // Don't box
expect(4)
val nullElement = channel.receiveOrClosed()
assertTrue(nullElement.isValue)
assertNull(nullElement.value)
assertNull(nullElement.valueOrNull)
assertEquals("Value(null)", nullElement.toString())
assertTrue(ValueOrClosed.value(null) == nullElement) // Don't box
expect(5)
val closed = channel.receiveOrClosed()
assertTrue(closed.isClosed)
val closed2 = channel.receiveOrClosed()
assertTrue(closed2.isClosed)
assertTrue(closed2.closeCause is ClosedReceiveChannelException)
finish(7)
}
@Test
@ExperimentalUnsignedTypes
fun testUIntChannel() = runTest {
val channel = Channel<UInt>()
launch {
expect(2)
channel.send(1u)
yield()
expect(4)
channel.send((Long.MAX_VALUE - 1).toUInt())
expect(5)
}
expect(1)
val element = channel.receiveOrClosed()
assertEquals(1u, element.value)
expect(3)
val element2 = channel.receiveOrClosed()
assertEquals((Long.MAX_VALUE - 1).toUInt(), element2.value)
finish(6)
}
@Test
fun testCancelChannel() = runTest {
val channel = Channel<Boolean>()
launch {
expect(2)
channel.cancel()
}
expect(1)
val closed = channel.receiveOrClosed()
assertTrue(closed.isClosed)
assertTrue(closed.closeCause is ClosedReceiveChannelException)
finish(3)
}
@Test
@ExperimentalUnsignedTypes
fun testReceiveResultChannel() = runTest {
val channel = Channel<ValueOrClosed<UInt>>()
launch {
channel.send(ValueOrClosed.value(1u))
channel.send(ValueOrClosed.closed(TestException1()))
channel.close(TestException2())
}
val intResult = channel.receiveOrClosed()
assertTrue(intResult.isValue)
assertEquals(1u, intResult.value.value)
val closeCauseResult = channel.receiveOrClosed()
assertTrue(closeCauseResult.isValue)
assertTrue(closeCauseResult.value.closeCause is TestException1)
val closeCause = channel.receiveOrClosed()
assertTrue(closeCause.isClosed)
assertTrue(closeCause.closeCause is TestException2)
assertFailsWith<TestException2> { closeCause.valueOrThrow }
}
@Test
fun testToString() = runTest {
val channel = Channel<String>(1)
channel.send("message")
channel.close(TestException1())
assertEquals("Value(message)", channel.receiveOrClosed().toString())
// toString implementation for exception differs on every platform
val str = channel.receiveOrClosed().toString()
assertTrue(str.matches("Closed\\(.*TestException1\\)".toRegex()))
}
}