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
|
/*
* 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 org.hamcrest.core.*
import org.junit.*
import org.junit.Assert.*
import org.junit.runner.*
import org.junit.runners.*
import kotlin.coroutines.*
@RunWith(Parameterized::class)
class SimpleSendReceiveJvmTest(
private val kind: TestChannelKind,
val n: Int,
val concurrent: Boolean
) : TestBase() {
companion object {
@Parameterized.Parameters(name = "{0}, n={1}, concurrent={2}")
@JvmStatic
fun params(): Collection<Array<Any>> = TestChannelKind.values().flatMap { kind ->
listOf(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 100, 1000).flatMap { n ->
listOf(false, true).map { concurrent ->
arrayOf<Any>(kind, n, concurrent)
}
}
}
}
val channel = kind.create()
@Test
fun testSimpleSendReceive() = runBlocking {
val ctx = if (concurrent) Dispatchers.Default else coroutineContext
launch(ctx) {
repeat(n) { channel.send(it) }
channel.close()
}
var expected = 0
for (x in channel) {
if (!kind.isConflated) {
assertThat(x, IsEqual(expected++))
} else {
assertTrue(x >= expected)
expected = x + 1
}
}
if (!kind.isConflated) {
assertThat(expected, IsEqual(n))
}
}
}
|
