Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
@file:Suppress("InvalidPackageDeclaration")
package org.mozilla.experiments.nimbus
import android.content.Context
import org.json.JSONObject
class MockNimbus(override val context: Context, val map: Map<String, JSONObject>) : FeaturesInterface {
constructor(vararg pairs: Pair<String, String>, context: Context = Context()) : this(
context,
pairs.toMap().mapValues { entry ->
JSONObject(entry.value)
},
)
init {
NullVariables.instance.setContext(context)
}
override fun getVariables(featureId: String, recordExposureEvent: Boolean): Variables =
map[featureId]?.let { json -> JSONVariables(context, json) } ?: NullVariables.instance
private var exposureCounts = mutableMapOf<String, Int>()
override fun recordExposureEvent(featureId: String, experimentSlug: String?) {
if (map[featureId] != null) {
exposureCounts[featureId] = getExposureCount(featureId) + 1
}
}
fun getExposureCount(featureId: String) = exposureCounts.getOrDefault(featureId, 0)
fun isExposed(featureId: String) = getExposureCount(featureId) > 0
}