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

Go to download
package io.mockk.impl.eval

import io.mockk.CapturingSlot
import io.mockk.MockKGateway.CallRecorder
import io.mockk.MockKGateway.Stubber
import io.mockk.MockKMatcherScope
import io.mockk.MockKStubScope
import io.mockk.impl.recording.AutoHinter

class EveryBlockEvaluator(callRecorder: () -> CallRecorder,
                          autoHinterFactory: () -> AutoHinter) : RecordedBlockEvaluator(callRecorder, autoHinterFactory), Stubber {

    override fun  every(mockBlock: (MockKMatcherScope.() -> T)?,
                           coMockBlock: (suspend MockKMatcherScope.() -> T)?): MockKStubScope {

        callRecorder().startStubbing()

        val lambda = CapturingSlot>()
        val scope = MockKMatcherScope(callRecorder(), lambda)

        record(scope, mockBlock, coMockBlock)

        return MockKStubScope(callRecorder(), lambda)
    }
}