commonTest.CoroutineScopeTest.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
Show all versions of kotlinx-coroutines-core
Coroutines support libraries for Kotlin
/*
* Copyright 2016-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/
@file:Suppress("UNREACHABLE_CODE")
package kotlinx.coroutines
import kotlinx.coroutines.internal.*
import kotlin.coroutines.*
import kotlin.test.*
class CoroutineScopeTest : TestBase() {
@Test
fun testScope() = runTest {
suspend fun callJobScoped() = coroutineScope {
expect(2)
launch {
expect(4)
}
launch {
expect(5)
launch {
expect(7)
}
expect(6)
}
expect(3)
42
}
expect(1)
val result = callJobScoped()
assertEquals(42, result)
yield() // Check we're not cancelled
finish(8)
}
@Test
fun testScopeCancelledFromWithin() = runTest {
expect(1)
suspend fun callJobScoped() = coroutineScope {
launch {
expect(2)
delay(Long.MAX_VALUE)
}
launch {
expect(3)
throw TestException2()
}
}
try {
callJobScoped()
expectUnreached()
} catch (e: TestException2) {
expect(4)
}
yield() // Check we're not cancelled
finish(5)
}
@Test
fun testExceptionFromWithin() = runTest {
expect(1)
try {
expect(2)
coroutineScope {
expect(3)
throw TestException1()
}
expectUnreached()
} catch (e: TestException1) {
finish(4)
}
}
@Test
fun testScopeBlockThrows() = runTest {
expect(1)
suspend fun callJobScoped(): Unit = coroutineScope {
launch {
expect(2)
delay(Long.MAX_VALUE)
}
yield() // let launch sleep
throw TestException1()
}
try {
callJobScoped()
expectUnreached()
} catch (e: TestException1) {
expect(3)
}
yield() // Check we're not cancelled
finish(4)
}
@Test
fun testOuterJobIsCancelled() = runTest {
suspend fun callJobScoped() = coroutineScope {
launch {
expect(3)
try {
println("1")
delay(Long.MAX_VALUE)
} finally {
println("F")
expect(4)
}
}
println("E?")
expect(2)
println("3")
delay(Long.MAX_VALUE)
42
}
val outerJob = launch(NonCancellable) {
expect(1)
try {
println("2")
callJobScoped()
expectUnreached()
} catch (e: JobCancellationException) {
println("4")
expect(5)
if (RECOVER_STACK_TRACES) {
val cause = e.cause as JobCancellationException // shall be recovered JCE
assertNull(cause.cause)
} else {
assertNull(e.cause)
}
}
}
repeat(3) { yield() } // let everything to start properly
println("C")
outerJob.cancel()
println("J")
outerJob.join()
println("D")
finish(6)
}
@Test
fun testAsyncCancellationFirst() = runTest {
try {
expect(1)
failedConcurrentSumFirst()
expectUnreached()
} catch (e: TestException1) {
finish(6)
}
}
// First async child fails -> second is cancelled
private suspend fun failedConcurrentSumFirst(): Int = coroutineScope {
val one = async {
expect(3)
throw TestException1()
}
val two = async(start = CoroutineStart.ATOMIC) {
try {
expect(4)
delay(Long.MAX_VALUE) // Emulates very long computation
42
} finally {
expect(5)
}
}
expect(2)
one.await() + two.await()
}
@Test
fun testAsyncCancellationSecond() = runTest {
try {
expect(1)
failedConcurrentSumSecond()
expectUnreached()
} catch (e: TestException1) {
finish(6)
}
}
// Second async child fails -> fist is cancelled
private suspend fun failedConcurrentSumSecond(): Int = coroutineScope {
val one = async {
try {
expect(3)
delay(Long.MAX_VALUE) // Emulates very long computation
42
} finally {
expect(5)
}
}
val two = async(start = CoroutineStart.ATOMIC) {
expect(4)
throw TestException1()
}
expect(2)
one.await() + two.await()
}
@Test
@Suppress("UNREACHABLE_CODE")
fun testDocumentationExample() = runTest {
suspend fun loadData() = coroutineScope {
expect(1)
val data = async {
try {
delay(Long.MAX_VALUE)
} finally {
expect(3)
}
}
yield()
// UI updater
withContext(coroutineContext) {
expect(2)
throw TestException1()
data.await() // Actually unreached
expectUnreached()
}
}
try {
loadData()
expectUnreached()
} catch (e: TestException1) {
finish(4)
}
}
@Test
fun testCoroutineScopeCancellationVsException() = runTest {
expect(1)
var job: Job? = null
job = launch(start = CoroutineStart.UNDISPATCHED) {
expect(2)
try {
coroutineScope {
expect(3)
yield() // must suspend
expect(5)
job!!.cancel() // cancel this job _before_ it throws
throw TestException1()
}
} catch (e: TestException1) {
// must have caught TextException
expect(6)
}
}
expect(4)
yield() // to coroutineScope
finish(7)
}
@Test
fun testScopePlusContext() {
assertSame(EmptyCoroutineContext, scopePlusContext(EmptyCoroutineContext, EmptyCoroutineContext))
assertSame(Dispatchers.Default, scopePlusContext(EmptyCoroutineContext, Dispatchers.Default))
assertSame(Dispatchers.Default, scopePlusContext(Dispatchers.Default, EmptyCoroutineContext))
assertSame(Dispatchers.Default, scopePlusContext(Dispatchers.Default, Dispatchers.Default))
assertSame(Dispatchers.Default, scopePlusContext(Dispatchers.Unconfined, Dispatchers.Default))
assertSame(Dispatchers.Unconfined, scopePlusContext(Dispatchers.Default, Dispatchers.Unconfined))
assertSame(Dispatchers.Unconfined, scopePlusContext(Dispatchers.Unconfined, Dispatchers.Unconfined))
}
@Test
fun testIncompleteScopeState() = runTest {
lateinit var scopeJob: Job
coroutineScope {
scopeJob = coroutineContext[Job]!!
scopeJob.invokeOnCompletion { }
}
scopeJob.join()
assertTrue(scopeJob.isCompleted)
assertFalse(scopeJob.isActive)
assertFalse(scopeJob.isCancelled)
}
private fun scopePlusContext(c1: CoroutineContext, c2: CoroutineContext) =
(ContextScope(c1) + c2).coroutineContext
}