jvmMain.io.mockk.MockK.kt Maven / Gradle / Ivy

Go to download
@file:JvmName("JVMMockKKt")
package io.mockk

import io.mockk.impl.JvmMockKGateway
import kotlin.reflect.KFunction
import kotlin.reflect.KProperty
import kotlin.reflect.jvm.javaMethod

actual object MockK {
    actual inline fun  useImpl(block: () -> T): T {
        MockKGateway.implementation = JvmMockKGateway.defaultImplementationBuilder
        return block()
    }
}

/**
 * Returns the defining top-level extension function's [Class].
 */
inline val KFunction<*>.declaringKotlinFile
    get() = checkNotNull(javaMethod) { "$this is not a top-level extension function" }
            .declaringClass.kotlin

/**
 * Builds a static mock. Any mocks of this function's declaring class are cancelled before it's mocked
 */
fun mockkStatic(vararg functions: KFunction<*>) =
        mockkStatic(*functions.map { it.declaringKotlinFile }.toTypedArray())

/**
 * Builds a static mock. Any mocks of this function's declaring class are cancelled before it's mocked
 */
fun mockkStatic(vararg functions: KProperty<*>) =
    mockkStatic(*functions.map(KProperty<*>::getter).toTypedArray())

/**
 * Cancel static mocks.
 */
fun unmockkStatic(vararg functions: KFunction<*>) =
        unmockkStatic(*functions.map { it.declaringKotlinFile }.toTypedArray())

/**
 * Cancel static mocks.
 */
fun unmockkStatic(vararg functions: KProperty<*>) =
    unmockkStatic(*functions.map(KProperty<*>::getter).toTypedArray())

/**
 * Clear static mocks.
 */
fun clearStaticMockk(vararg functions: KFunction<*>) =
    clearStaticMockk(*functions.map { it.declaringKotlinFile }.toTypedArray())

/**
 * Clear static mocks.
 */
fun clearStaticMockk(vararg functions: KProperty<*>) =
    clearStaticMockk(*functions.map(KProperty<*>::getter).toTypedArray())

/**
 * Builds a static mock and unmocks it after the block has been executed.
 */
inline fun mockkStatic(vararg functions: KFunction<*>, block: () -> Unit) =
        mockkStatic(*functions.map { it.declaringKotlinFile }.toTypedArray(), block = block)

/**
 * Builds a static mock and unmocks it after the block has been executed.
 */
inline fun mockkStatic(vararg functions: KProperty<*>, block: () -> Unit) =
    mockkStatic(*functions.map(KProperty<*>::getter).toTypedArray(), block = block)