blob: 501affb4d9083cc2f33785ee7f26952e02febbeb (
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
|
/*
* 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 kotlin.test.*
class LinkedListChannelTest : TestBase() {
@Test
fun testBasic() = runTest {
val c = Channel<Int>(Channel.UNLIMITED)
c.send(1)
assertTrue(c.trySend(2).isSuccess)
c.send(3)
check(c.close())
check(!c.close())
assertEquals(1, c.receive())
assertEquals(2, c.tryReceive().getOrNull())
assertEquals(3, c.receiveCatching().getOrNull())
assertNull(c.receiveCatching().getOrNull())
}
@Test
fun testConsumeAll() = runTest {
val q = Channel<Int>(Channel.UNLIMITED)
for (i in 1..10) {
q.send(i) // buffers
}
q.cancel()
check(q.isClosedForSend)
check(q.isClosedForReceive)
assertFailsWith<CancellationException> { q.receive() }
}
@Test
fun testCancelWithCause() = runTest({ it is TestCancellationException }) {
val channel = Channel<Int>(Channel.UNLIMITED)
channel.cancel(TestCancellationException())
channel.receive()
}
}
|