@@ -21,9 +21,9 @@ import androidx.arch.core.executor.testing.InstantTaskExecutorRule
2121import androidx.test.ext.junit.runners.AndroidJUnit4
2222import app.cash.turbine.test
2323import com.duckduckgo.app.appearance.AppearanceViewModel.Command
24+ import com.duckduckgo.app.browser.omnibar.OmnibarFeatureRepository
2425import com.duckduckgo.app.icon.api.AppIcon
2526import com.duckduckgo.app.pixels.AppPixelName
26- import com.duckduckgo.app.pixels.remoteconfig.AndroidBrowserConfigFeature
2727import com.duckduckgo.app.settings.clear.FireAnimation
2828import com.duckduckgo.app.settings.db.SettingsDataStore
2929import com.duckduckgo.app.statistics.pixels.Pixel
@@ -33,8 +33,6 @@ import com.duckduckgo.common.test.CoroutineTestRule
3333import com.duckduckgo.common.ui.DuckDuckGoTheme
3434import com.duckduckgo.common.ui.store.AppTheme
3535import com.duckduckgo.common.ui.store.ThemingDataStore
36- import com.duckduckgo.feature.toggles.api.FakeFeatureToggleFactory
37- import com.duckduckgo.feature.toggles.api.Toggle.State
3836import kotlinx.coroutines.flow.flowOf
3937import kotlinx.coroutines.test.runTest
4038import org.junit.Assert.assertEquals
@@ -74,7 +72,8 @@ internal class AppearanceViewModelTest {
7472 @Mock
7573 private lateinit var mockTabSwitcherDataStore: TabSwitcherDataStore
7674
77- private val androidBrowserConfigFeature = FakeFeatureToggleFactory .create(AndroidBrowserConfigFeature ::class .java)
75+ @Mock
76+ private lateinit var mockOmnibarFeatureRepository: OmnibarFeatureRepository
7877
7978 @SuppressLint(" DenyListedApi" )
8079 @Before
@@ -86,9 +85,7 @@ internal class AppearanceViewModelTest {
8685 whenever(mockAppSettingsDataStore.selectedFireAnimation).thenReturn(FireAnimation .HeroFire )
8786 whenever(mockAppSettingsDataStore.omnibarType).thenReturn(OmnibarType .SINGLE_TOP )
8887 whenever(mockTabSwitcherDataStore.isTrackersAnimationInfoTileHidden()).thenReturn(flowOf(false ))
89-
90- androidBrowserConfigFeature.useUnifiedOmnibarLayout().setRawStoredState(State (enable = false ))
91- androidBrowserConfigFeature.splitOmnibar().setRawStoredState(State (enable = false ))
88+ whenever(mockOmnibarFeatureRepository.isSplitOmnibarEnabled).thenReturn(false )
9289
9390 initializeViewModel()
9491 }
@@ -101,7 +98,7 @@ internal class AppearanceViewModelTest {
10198 mockPixel,
10299 coroutineTestRule.testDispatcherProvider,
103100 mockTabSwitcherDataStore,
104- androidBrowserConfigFeature ,
101+ mockOmnibarFeatureRepository ,
105102 )
106103 }
107104
@@ -243,6 +240,14 @@ internal class AppearanceViewModelTest {
243240 verify(mockPixel).fire(AppPixelName .SETTINGS_ADDRESS_BAR_POSITION_SELECTED_TOP )
244241 }
245242
243+ @Test
244+ fun whenOmnibarPositionUpdatedToSplit () =
245+ runTest {
246+ testee.onOmnibarTypeSelected(OmnibarType .SPLIT )
247+ verify(mockAppSettingsDataStore).omnibarType = OmnibarType .SPLIT
248+ verify(mockPixel).fire(AppPixelName .SETTINGS_ADDRESS_BAR_POSITION_SELECTED_SPLIT_TOP )
249+ }
250+
246251 @Test
247252 fun whenFullSiteAddressEnabled () =
248253 runTest {
0 commit comments