Skip to content

Commit e778a5a

Browse files
committed
Fixed tests
1 parent df435b6 commit e778a5a

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

app/src/androidTest/java/com/duckduckgo/app/browser/BrowserTabViewModelTest.kt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,8 @@ import com.duckduckgo.app.browser.viewstate.FindInPageViewState
128128
import com.duckduckgo.app.browser.viewstate.GlobalLayoutViewState
129129
import com.duckduckgo.app.browser.viewstate.HighlightableButton
130130
import com.duckduckgo.app.browser.viewstate.LoadingViewState
131+
import com.duckduckgo.app.browser.viewstate.VpnMenuState
132+
import com.duckduckgo.app.browser.menu.VpnMenuStateProvider
131133
import com.duckduckgo.app.browser.webview.MaliciousSiteBlockedWarningLayout.Action.LearnMore
132134
import com.duckduckgo.app.browser.webview.MaliciousSiteBlockedWarningLayout.Action.LeaveSite
133135
import com.duckduckgo.app.browser.webview.MaliciousSiteBlockedWarningLayout.Action.ReportError
@@ -450,6 +452,8 @@ class BrowserTabViewModelTest {
450452

451453
private val mockExternalIntentProcessingState: ExternalIntentProcessingState = mock()
452454

455+
private val mockVpnMenuStateProvider: VpnMenuStateProvider = mock()
456+
453457
private val mockHasPendingTabLaunchFlow = MutableStateFlow(false)
454458

455459
private val mockHasPendingDuckAiOpenFlow = MutableStateFlow(false)
@@ -660,6 +664,7 @@ class BrowserTabViewModelTest {
660664
whenever(mockDuckAiFeatureState.showInputScreenAutomaticallyOnNewTab).thenReturn(mockDuckAiFeatureStateInputScreenOpenAutomaticallyFlow)
661665
whenever(mockExternalIntentProcessingState.hasPendingTabLaunch).thenReturn(mockHasPendingTabLaunchFlow)
662666
whenever(mockExternalIntentProcessingState.hasPendingDuckAiOpen).thenReturn(mockHasPendingDuckAiOpenFlow)
667+
whenever(mockVpnMenuStateProvider.getVpnMenuState()).thenReturn(flowOf(VpnMenuState.Hidden))
663668
whenever(mockOnboardingDesignExperimentManager.isModifiedControlEnrolledAndEnabled()).thenReturn(false)
664669
whenever(mockOnboardingDesignExperimentManager.isBuckEnrolledAndEnabled()).thenReturn(false)
665670
whenever(mockOnboardingDesignExperimentManager.isBbEnrolledAndEnabled()).thenReturn(false)
@@ -810,6 +815,7 @@ class BrowserTabViewModelTest {
810815
serpEasterEggLogosToggles = mockSerpEasterEggLogoToggles,
811816
nonHttpAppLinkChecker = nonHttpAppLinkChecker,
812817
externalIntentProcessingState = mockExternalIntentProcessingState,
818+
vpnMenuStateProvider = mockVpnMenuStateProvider,
813819
)
814820

815821
testee.loadData("abc", null, false, false)

0 commit comments

Comments
 (0)