@@ -128,6 +128,8 @@ import com.duckduckgo.app.browser.viewstate.FindInPageViewState
128128import com.duckduckgo.app.browser.viewstate.GlobalLayoutViewState
129129import com.duckduckgo.app.browser.viewstate.HighlightableButton
130130import com.duckduckgo.app.browser.viewstate.LoadingViewState
131+ import com.duckduckgo.app.browser.viewstate.VpnMenuState
132+ import com.duckduckgo.app.browser.menu.VpnMenuStateProvider
131133import com.duckduckgo.app.browser.webview.MaliciousSiteBlockedWarningLayout.Action.LearnMore
132134import com.duckduckgo.app.browser.webview.MaliciousSiteBlockedWarningLayout.Action.LeaveSite
133135import 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