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
|
/*
* Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
package kotlinx.coroutines.selects
import kotlinx.coroutines.*
import kotlinx.coroutines.sync.*
import kotlin.test.*
class SelectMutexTest : TestBase() {
@Test
fun testSelectLock() = runTest {
val mutex = Mutex()
expect(1)
launch { // ensure that it is not scheduled earlier than needed
finish(4) // after main exits
}
val res = select<String> {
mutex.onLock {
assertTrue(mutex.isLocked)
expect(2)
"OK"
}
}
assertEquals("OK", res)
expect(3)
// will wait for the first coroutine
}
@Test
fun testSelectLockWait() = runTest {
val mutex = Mutex(true) // locked
expect(1)
launch {
expect(3)
val res = select<String> {
// will suspended
mutex.onLock {
assertTrue(mutex.isLocked)
expect(6)
"OK"
}
}
assertEquals("OK", res)
expect(7)
}
expect(2)
yield() // to launched coroutine
expect(4)
mutex.unlock()
expect(5)
yield() // to resumed select
finish(8)
}
}
|
