package kotlinx.coroutines.flow

import kotlinx.coroutines.testing.*
import kotlinx.coroutines.*
import kotlin.test.*

class CancellableTest : TestBase() {

    @Test
    fun testCancellable() = runTest {
        var sum = 0
        val flow = (0..1000).asFlow()
            .onEach {
                if (it != 0) currentCoroutineContext().cancel()
                sum += it
            }

        flow.launchIn(this).join()
        assertEquals(500500, sum)
        
        sum = 0
        flow.cancellable().launchIn(this).join()
        assertEquals(1, sum)
    }

    @Test
    fun testFastPath() {
        val flow = listOf(1).asFlow()
        assertNotSame(flow, flow.cancellable())

        val cancellableFlow = flow { emit(42) }
        assertSame(cancellableFlow, cancellableFlow.cancellable())
    }
}