commonMain.org.kodein.mock.References.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockmp-runtime-jvm Show documentation

MocKMP runtime

The newest version!

package org.kodein.mock

import kotlin.reflect.KClass


internal class References {

    private val references = ArrayList()

    @Suppress("RemoveRedundantCallsOfConversionMethods")
    private val map = hashMapOf, Any>(
        Boolean::class to false,
        UByte::class to 0.toUByte(),
        Byte::class to 0.toByte(),
        UShort::class to 0.toUShort(),
        Short::class to 0.toShort(),
        Char::class to 0.toChar(),
        UInt::class to 0.toUInt(),
        Int::class to 0.toInt(),
        Float::class to 0.toFloat(),
        ULong::class to 0.toULong(),
        Long::class to 0.toLong(),
        Double::class to 0.toDouble()
    )

    fun addReference(r: Any) {
        references.add(r)
        map[r::class] = r
    }

    fun tryGetReference(cls: KClass<*>): Any? {
        map[cls]?.let { return it }
        var ref: Any? = null
        references.forEach {
            if (cls.isInstance(it)) ref = it
        }
        if (ref == null) ref = unsafeValue(cls)
        return ref
    }

    fun getReference(cls: KClass<*>): Any {
        val r = runCatching { tryGetReference(cls) }
        if (r.isFailure || r.getOrThrow() == null) {
            throw IllegalStateException("Could not create an instance of ${cls.bestName()}. Please use mocker.useReference(${cls.simpleName}) to set a reference.", r.exceptionOrNull())
        }
        return r.getOrNull()!!
    }
}