io.mockk.proxy.common.transformation.ClassTransformationSpec.kt Maven / Gradle / Ivy

Go to download
package io.mockk.proxy.common.transformation

data class ClassTransformationSpec(
    val cls: Class<*>,
    val simpleIntercept: Int = 0,
    val staticIntercept: Int = 0,
    val constructorIntercept: Int = 0
) {
    val shouldDoSimpleIntercept: Boolean
        get() = simpleIntercept > 0

    val shouldDoStaticIntercept: Boolean
        get() = staticIntercept > 0

    val shouldDoConstructorIntercept: Boolean
        get() = constructorIntercept > 0

    val shouldDoSomething
        get() = shouldDoSimpleIntercept ||
                shouldDoStaticIntercept ||
                shouldDoConstructorIntercept

    private data class Categories(
        val simple: Boolean,
        val static: Boolean,
        val constructor: Boolean
    )

    private fun categories() = Categories(
        shouldDoSimpleIntercept,
        shouldDoStaticIntercept,
        shouldDoConstructorIntercept
    )

    infix fun sameTransforms(other: ClassTransformationSpec) = categories() == other.categories()

    override fun toString(): String {
        val lst = mutableListOf()
        if (shouldDoSimpleIntercept) {
            lst.add("mockk")
        }
        if (shouldDoStaticIntercept) {
            lst.add("mockkStatic")
        }
        if (shouldDoConstructorIntercept) {
            lst.add("mockkConstructor")
        }
        return lst.joinToString(", ")
    }

}