@@ -19,9 +19,7 @@ package com.duckduckgo.app.onboarding.ui
1919import android.arch.core.executor.testing.InstantTaskExecutorRule
2020import com.duckduckgo.app.browser.defaultBrowsing.DefaultBrowserDetector
2121import com.duckduckgo.app.onboarding.store.OnboardingStore
22- import com.duckduckgo.app.statistics.Variant
2322import com.duckduckgo.app.statistics.VariantManager
24- import com.duckduckgo.app.statistics.VariantManager.VariantFeature.DefaultBrowserFeature.ShowInOnboarding
2523import com.nhaarman.mockito_kotlin.mock
2624import com.nhaarman.mockito_kotlin.never
2725import com.nhaarman.mockito_kotlin.verify
@@ -43,7 +41,7 @@ class OnboardingViewModelTest {
4341 private var variantManager: VariantManager = mock()
4442
4543 private val testee: OnboardingViewModel by lazy {
46- OnboardingViewModel (onboardingStore, mockDefaultBrowserDetector, variantManager )
44+ OnboardingViewModel (onboardingStore, mockDefaultBrowserDetector)
4745 }
4846
4947 @Test
@@ -66,33 +64,17 @@ class OnboardingViewModelTest {
6664 }
6765
6866 @Test
69- fun whenThirdPageRequestedWithFeatureEnabledAndDefaultBrowserCapableThenDefaultBrowserPageReturned () {
70- whenever(variantManager.getVariant()).thenReturn(variantWithOnboardingEnabled())
67+ fun whenThirdPageRequestedWithDefaultBrowserCapableThenDefaultBrowserPageReturned () {
7168 whenever(mockDefaultBrowserDetector.deviceSupportsDefaultBrowserConfiguration()).thenReturn(true )
7269 val page = testee.getItem(2 )
7370 assertTrue(page is OnboardingPageFragment .DefaultBrowserPage )
7471 }
7572
76- @Test
77- fun whenThirdPageRequestedWithFeatureDisabledAndDefaultBrowserCapableThenNoPageReturned () {
78- whenever(variantManager.getVariant()).thenReturn(variantWithOnboardingDisabled())
79- whenever(mockDefaultBrowserDetector.deviceSupportsDefaultBrowserConfiguration()).thenReturn(true )
80- val page = testee.getItem(2 )
81- assertNull(page)
82- }
83-
8473 @Test
8574 fun whenThirdPageRequestedButDefaultBrowserNotCapableThenNoPageReturned () {
86- whenever(variantManager.getVariant()).thenReturn(variantWithOnboardingEnabled())
8775 whenever(mockDefaultBrowserDetector.deviceSupportsDefaultBrowserConfiguration()).thenReturn(false )
8876 val page = testee.getItem(2 )
8977 assertNull(page)
9078 }
9179
92- private fun variantWithOnboardingEnabled (): Variant =
93- Variant (" " , 0.0 , listOf (ShowInOnboarding ))
94-
95- private fun variantWithOnboardingDisabled (): Variant =
96- Variant (" " , 0.0 , listOf ())
97-
9880}
0 commit comments