commonMain.com.copperleaf.ballast.test.internal.ViewModelTestScenarioInputSequenceScopeImpl.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ballast-test Show documentation

null

There is a newer version: 4.2.1

package com.copperleaf.ballast.test.internal

import com.copperleaf.ballast.test.ViewModelTestScenarioInputSequenceScope
import com.copperleaf.ballast.test.internal.vm.TestViewModel
import kotlinx.coroutines.CompletableDeferred

internal class ViewModelTestScenarioInputSequenceScopeImpl(
    private val logger: (String) -> Unit,
    private val vm: TestViewModel
) : ViewModelTestScenarioInputSequenceScope {

    override suspend fun send(input: Inputs) {
        logger("        before send ProcessInput")
        val deferred = CompletableDeferred()
        vm.send(
            TestViewModel.Inputs.ProcessInput(input, deferred)
        )
        logger("        after send ProcessInput")
        deferred.await()
        logger("        after await ProcessInput")
    }

    override suspend fun sendAndAwait(input: Inputs): State {
        logger("        before send AwaitInput")
        val deferred = CompletableDeferred()
        vm.send(
            TestViewModel.Inputs.AwaitInput(input, deferred)
        )
        logger("        after send AwaitInput")
        deferred.await()
        val latestStateAfterProcessing = vm.observeStates().value
        logger("        after await AwaitInput: $latestStateAfterProcessing")
        return latestStateAfterProcessing
    }

    internal suspend fun finish() {
        logger("        before send TestCompleted")
        val deferred = CompletableDeferred()
        vm.send(
            TestViewModel.Inputs.TestCompleted(deferred)
        )
        logger("        after send TestCompleted")

        // close the channel to prevent new items from coming in and cancelling the TestCompleted event
        vm.close()

        deferred.await()
        logger("        after await TestCompleted")
    }
}