Commit 5f01583f authored by mcarare's avatar mcarare Committed by ekager
Browse files

For #15688: Update unit tests.

parent 9274558e

app/src/test/java/org/mozilla/fenix/components/toolbar/DefaultBrowserToolbarControllerTest.kt

+17 −4
Original line number Original line Diff line number Diff line
@@ -45,30 +45,43 @@ class DefaultBrowserToolbarControllerTest { 




    @RelaxedMockK

    @RelaxedMockK

    private lateinit var activity: HomeActivity

    private lateinit var activity: HomeActivity



    @MockK(relaxUnitFun = true)

    @MockK(relaxUnitFun = true)

    private lateinit var navController: NavController

    private lateinit var navController: NavController



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var onTabCounterClicked: () -> Unit

    private lateinit var onTabCounterClicked: () -> Unit



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var onCloseTab: (Session) -> Unit

    private lateinit var onCloseTab: (Session) -> Unit



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var sessionManager: SessionManager

    private lateinit var sessionManager: SessionManager



    @MockK(relaxUnitFun = true)

    @MockK(relaxUnitFun = true)

    private lateinit var engineView: EngineView

    private lateinit var engineView: EngineView



    @MockK

    @MockK

    private lateinit var currentSession: Session

    private lateinit var currentSession: Session



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var metrics: MetricController

    private lateinit var metrics: MetricController



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var searchUseCases: SearchUseCases

    private lateinit var searchUseCases: SearchUseCases



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var sessionUseCases: SessionUseCases

    private lateinit var sessionUseCases: SessionUseCases



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var browserAnimator: BrowserAnimator

    private lateinit var browserAnimator: BrowserAnimator



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var topSitesUseCase: TopSitesUseCases

    private lateinit var topSitesUseCase: TopSitesUseCases



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var readerModeController: ReaderModeController

    private lateinit var readerModeController: ReaderModeController



    @RelaxedMockK

    @RelaxedMockK

    private lateinit var homeViewModel: HomeScreenViewModel

    private lateinit var homeViewModel: HomeScreenViewModel
@@ -233,12 +246,12 @@ class DefaultBrowserToolbarControllerTest { 
        val item = TabCounterMenu.Item.NewTab(BrowsingMode.Normal)

        val item = TabCounterMenu.Item.NewTab(BrowsingMode.Normal)





        every { activity.browsingModeManager } returns browsingModeManager

        every { activity.browsingModeManager } returns browsingModeManager

        every { navController.popBackStack(R.id.homeFragment, any()) } returns true

        every { navController.navigate(BrowserFragmentDirections.actionGlobalHome(focusOnAddressBar = true)) } just Runs





        val controller = createController()

        val controller = createController()

        controller.handleTabCounterItemInteraction(item)

        controller.handleTabCounterItemInteraction(item)

        assertEquals(BrowsingMode.Normal, browsingModeManager.mode)

        assertEquals(BrowsingMode.Normal, browsingModeManager.mode)

        verify { navController.popBackStack(R.id.homeFragment, false) }

        verify { navController.navigate(BrowserFragmentDirections.actionGlobalHome(focusOnAddressBar = true)) }

    }

    }





    @Test

    @Test
@@ -247,12 +260,12 @@ class DefaultBrowserToolbarControllerTest { 
        val item = TabCounterMenu.Item.NewTab(BrowsingMode.Private)

        val item = TabCounterMenu.Item.NewTab(BrowsingMode.Private)





        every { activity.browsingModeManager } returns browsingModeManager

        every { activity.browsingModeManager } returns browsingModeManager

        every { navController.popBackStack(R.id.homeFragment, any()) } returns true

        every { navController.navigate(BrowserFragmentDirections.actionGlobalHome(focusOnAddressBar = true)) } just Runs





        val controller = createController()

        val controller = createController()

        controller.handleTabCounterItemInteraction(item)

        controller.handleTabCounterItemInteraction(item)

        assertEquals(BrowsingMode.Private, browsingModeManager.mode)

        assertEquals(BrowsingMode.Private, browsingModeManager.mode)

        verify { navController.popBackStack(R.id.homeFragment, false) }

        verify { navController.navigate(BrowserFragmentDirections.actionGlobalHome(focusOnAddressBar = true)) }

    }

    }





    @Test

    @Test