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

package kotlinx.coroutines.flow

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

class OnCompletionInterceptedReleaseTest : TestBase() {
    @Test
    fun testLeak() = runTest {
        expect(1)
        var cont: Continuation? = null
        val interceptor = CountingInterceptor()
        val job = launch(interceptor, start = CoroutineStart.UNDISPATCHED) {
            emptyFlow()
                .onCompletion { emit(1) }
                .collect { value ->
                    expect(2)
                    assertEquals(1, value)
                    suspendCoroutine { cont = it }
                }
        }
        cont!!.resume(Unit)
        assertTrue(job.isCompleted)
        assertEquals(interceptor.intercepted, interceptor.released)
        finish(3)
    }

    class CountingInterceptor : AbstractCoroutineContextElement(ContinuationInterceptor), ContinuationInterceptor {
        var intercepted = 0
        var released = 0
        override fun  interceptContinuation(continuation: Continuation): Continuation {
            intercepted++
            return Continuation(continuation.context) { continuation.resumeWith(it) }
        }

        override fun releaseInterceptedContinuation(continuation: Continuation<*>) {
            released++
        }
    }
}