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
|
/*
* Copyright 2016-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
package kotlinx.coroutines
import org.hamcrest.MatcherAssert.*
import org.hamcrest.core.*
import org.junit.*
import java.util.concurrent.*
import kotlin.coroutines.*
class DelayJvmTest : TestBase() {
/**
* Test that delay works properly in contexts with custom [ContinuationInterceptor]
*/
@Test
fun testDelayInArbitraryContext() = runBlocking {
var thread: Thread? = null
val pool = Executors.newFixedThreadPool(1) { runnable ->
Thread(runnable).also { thread = it }
}
val context = CustomInterceptor(pool)
val c = async(context) {
assertThat(Thread.currentThread(), IsEqual(thread))
delay(100)
assertThat(Thread.currentThread(), IsEqual(thread))
42
}
assertThat(c.await(), IsEqual(42))
pool.shutdown()
}
@Test
fun testDelayWithoutDispatcher() = runBlocking(CoroutineName("testNoDispatcher.main")) {
// launch w/o a specified dispatcher
val c = async(CoroutineName("testNoDispatcher.inner")) {
delay(100)
42
}
assertThat(c.await(), IsEqual(42))
}
@Test
fun testNegativeDelay() = runBlocking {
expect(1)
val job = async {
expect(3)
delay(0)
expect(4)
}
delay(-1)
expect(2)
job.await()
finish(5)
}
class CustomInterceptor(val pool: Executor) : AbstractCoroutineContextElement(ContinuationInterceptor), ContinuationInterceptor {
override fun <T> interceptContinuation(continuation: Continuation<T>): Continuation<T> =
Wrapper(pool, continuation)
}
class Wrapper<T>(val pool: Executor, private val cont: Continuation<T>) : Continuation<T> {
override val context: CoroutineContext
get() = cont.context
override fun resumeWith(result: Result<T>) {
pool.execute { cont.resumeWith(result) }
}
}
}
|
