main.wisp.deployment.FakeEnvironmentVariableLoader.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of wisp-deployment-testing Show documentation

a module containing deployment testing helpers, e.g. fakes

There is a newer version: 2024.09.17.200749-4708422

Show newest version

package wisp.deployment

/**
 * A Fake [EnvironmentVariableLoader] that loads from an in memory map
 */
class FakeEnvironmentVariableLoader @JvmOverloads constructor(val vars: MutableMap = mutableMapOf()) :
    EnvironmentVariableLoader {

    override fun getEnvironmentVariable(name: String): String {
        return vars[name] ?: throw IllegalStateException("$name environment variable not set")
    }

    override fun getEnvironmentVariableOrDefault(name: String, defaultValue: String): String {
        return vars[name] ?: defaultValue
    }
}