jvmTest.exceptions.FlowSuppressionTest.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-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */

package kotlinx.coroutines.exceptions

import kotlinx.coroutines.*
import kotlinx.coroutines.flow.*
import org.junit.*
import org.junit.Test
import kotlin.test.*

class FlowSuppressionTest : TestBase() {
    @Test
    fun testSuppressionForPrimaryException() = runTest {
        val flow = flow {
            try {
                emit(1)
            } finally {
                throw TestException()
            }
        }.catch { expectUnreached() }.onEach { throw TestException2() }

        try {
            flow.collect()
        } catch (e: Throwable) {
            assertIs(e)
            assertIs(e.suppressed[0])
        }
    }

    @Test
    fun testSuppressionForPrimaryExceptionRetry() = runTest {
        val flow = flow {
            try {
                emit(1)
            } finally {
                throw TestException()
            }
        }.retry { expectUnreached(); true }.onEach { throw TestException2() }

        try {
            flow.collect()
        } catch (e: Throwable) {
            assertIs(e)
            assertIs(e.suppressed[0])

        }
    }

    @Test
    fun testCancellationSuppression() = runTest {
        val flow = flow {
            try {
                expect(1)
                emit(1)
            } finally {
                expect(3)
                throw CancellationException("")
            }
        }.catch { expectUnreached() }.onEach {
            expect(2)
            throw TestException("")
        }

        try {
            flow.collect()
        } catch (e: Throwable) {
            assertIs(e)
            assertIs(e.suppressed[0])
        }
        finish(4)
    }
}