@@ -23,6 +23,7 @@ import app.cash.turbine.test
2323import com.duckduckgo.app.appearance.AppearanceViewModel.Command
2424import com.duckduckgo.app.icon.api.AppIcon
2525import com.duckduckgo.app.pixels.AppPixelName
26+ import com.duckduckgo.app.pixels.remoteconfig.AndroidBrowserConfigFeature
2627import com.duckduckgo.app.settings.clear.FireAnimation
2728import com.duckduckgo.app.settings.db.SettingsDataStore
2829import com.duckduckgo.app.statistics.pixels.Pixel
@@ -32,6 +33,8 @@ import com.duckduckgo.common.test.CoroutineTestRule
3233import com.duckduckgo.common.ui.DuckDuckGoTheme
3334import com.duckduckgo.common.ui.store.AppTheme
3435import com.duckduckgo.common.ui.store.ThemingDataStore
36+ import com.duckduckgo.feature.toggles.api.FakeFeatureToggleFactory
37+ import com.duckduckgo.feature.toggles.api.Toggle.State
3538import kotlinx.coroutines.flow.flowOf
3639import kotlinx.coroutines.test.runTest
3740import org.junit.Assert.assertEquals
@@ -71,6 +74,8 @@ internal class AppearanceViewModelTest {
7174 @Mock
7275 private lateinit var mockTabSwitcherDataStore: TabSwitcherDataStore
7376
77+ private val androidBrowserConfigFeature = FakeFeatureToggleFactory .create(AndroidBrowserConfigFeature ::class .java)
78+
7479 @SuppressLint(" DenyListedApi" )
7580 @Before
7681 fun before () {
@@ -82,6 +87,9 @@ internal class AppearanceViewModelTest {
8287 whenever(mockAppSettingsDataStore.omnibarType).thenReturn(OmnibarType .SINGLE_TOP )
8388 whenever(mockTabSwitcherDataStore.isTrackersAnimationInfoTileHidden()).thenReturn(flowOf(false ))
8489
90+ androidBrowserConfigFeature.useUnifiedOmnibarLayout().setRawStoredState(State (enable = false ))
91+ androidBrowserConfigFeature.splitOmnibar().setRawStoredState(State (enable = false ))
92+
8593 initializeViewModel()
8694 }
8795
@@ -93,6 +101,7 @@ internal class AppearanceViewModelTest {
93101 mockPixel,
94102 coroutineTestRule.testDispatcherProvider,
95103 mockTabSwitcherDataStore,
104+ androidBrowserConfigFeature,
96105 )
97106 }
98107
0 commit comments