io.mockk.impl.eval.RecordedBlockEvaluator.kt Maven / Gradle / Ivy

Go to download
package io.mockk.impl.eval

import io.mockk.InternalPlatformDsl
import io.mockk.MockKException
import io.mockk.MockKGateway.CallRecorder
import io.mockk.MockKMatcherScope
import io.mockk.impl.InternalPlatform
import io.mockk.impl.recording.AutoHinter

abstract class RecordedBlockEvaluator(
    val callRecorder: () -> CallRecorder,
    val autoHinterFactory: () -> AutoHinter
) {

    fun  record(
        scope: S,
        mockBlock: (S.() -> T)?,
        coMockBlock: (suspend S.() -> T)?
    ) {
        try {
            val block: () -> T = if (mockBlock != null) {
                { scope.mockBlock() }
            } else if (coMockBlock != null) {
                { InternalPlatformDsl.runCoroutine { scope.coMockBlock() } }
            } else {
                { throw MockKException("You should specify either 'mockBlock' or 'coMockBlock'") }
            }

            val autoHinter = autoHinterFactory()

            autoHinter.autoHint(
                callRecorder(),
                0,
                64,
                block
            )

            val n = callRecorder().estimateCallRounds();
            for (i in 1 until n) {
                autoHinter.autoHint(
                    callRecorder(),
                    i,
                    n,
                    block
                )
            }
            callRecorder().round(n, n)
            callRecorder().done()
        } catch (ex: Throwable) {
            throw InternalPlatform.prettifyRecordingException(ex)
        }
    }
}