jvmTest.DelayJvmTest.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlinx-coroutines-core

Coroutines support libraries for Kotlin

There is a newer version: 1.9.0

/*
 * 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.junit.*
import java.util.concurrent.*
import kotlin.coroutines.*
import kotlin.test.assertEquals

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) {
            assertEquals(thread, Thread.currentThread())
            delay(100)
            assertEquals(thread, Thread.currentThread())
            42
        }
        assertEquals(42, c.await())
        pool.shutdown()
    }

    @Test
    fun testDelayWithoutDispatcher() = runBlocking(CoroutineName("testNoDispatcher.main")) {
        // launch w/o a specified dispatcher
        val c = async(CoroutineName("testNoDispatcher.inner")) {
            delay(100)
            42
        }
        assertEquals(42, c.await())
    }

    @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  interceptContinuation(continuation: Continuation): Continuation =
            Wrapper(pool, continuation)
    }

    class Wrapper(val pool: Executor, private val cont: Continuation) : Continuation {
        override val context: CoroutineContext
            get() = cont.context

        override fun resumeWith(result: Result) {
            pool.execute { cont.resumeWith(result) }
        }
    }
}