diff --git a/app/src/test/java/com/duckduckgo/app/generalsettings/showonapplaunch/store/FakeShowOnAppLaunchOptionDataStore.kt b/app/src/test/java/com/duckduckgo/app/generalsettings/showonapplaunch/store/FakeShowOnAppLaunchOptionDataStore.kt index bbb7b11517c3..51261a3d6e70 100644 --- a/app/src/test/java/com/duckduckgo/app/generalsettings/showonapplaunch/store/FakeShowOnAppLaunchOptionDataStore.kt +++ b/app/src/test/java/com/duckduckgo/app/generalsettings/showonapplaunch/store/FakeShowOnAppLaunchOptionDataStore.kt @@ -26,9 +26,17 @@ class FakeShowOnAppLaunchOptionDataStore : ShowOnAppLaunchOptionDataStore { private var currentOptionStateFlow = MutableStateFlow(null) + private var currentSpecificPageUrl = MutableStateFlow("duckduckgo.com") + override val optionFlow: Flow = currentOptionStateFlow.asStateFlow().filterNotNull() + override val specificPageUrlFlow: Flow = currentSpecificPageUrl.asStateFlow() + override suspend fun setShowOnAppLaunchOption(showOnAppLaunchOption: ShowOnAppLaunchOption) { currentOptionStateFlow.value = showOnAppLaunchOption } + + override suspend fun setSpecificPageUrl(url: String) { + currentSpecificPageUrl.value = url + } }