diff --git a/app/src/test/java/org/mozilla/fenix/search/DefaultSearchControllerTest.kt b/app/src/test/java/org/mozilla/fenix/search/DefaultSearchControllerTest.kt index 77f7fd1a3..932f84c7c 100644 --- a/app/src/test/java/org/mozilla/fenix/search/DefaultSearchControllerTest.kt +++ b/app/src/test/java/org/mozilla/fenix/search/DefaultSearchControllerTest.kt @@ -204,6 +204,24 @@ class DefaultSearchControllerTest { verify { navController.navigate(directions) } } + @Test + fun handleSearchShortcutsButtonClicked_alreadyOpen() { + every { store.state.showSearchShortcuts } returns true + + controller.handleSearchShortcutsButtonClicked() + + verify { store.dispatch(SearchFragmentAction.ShowSearchShortcutEnginePicker(false)) } + } + + @Test + fun handleSearchShortcutsButtonClicked_notYetOpen() { + every { store.state.showSearchShortcuts } returns false + + controller.handleSearchShortcutsButtonClicked() + + verify { store.dispatch(SearchFragmentAction.ShowSearchShortcutEnginePicker(true)) } + } + @Test fun handleExistingSessionSelected() { val session: Session = mockk(relaxed = true) diff --git a/app/src/test/java/org/mozilla/fenix/search/SearchInteractorTest.kt b/app/src/test/java/org/mozilla/fenix/search/SearchInteractorTest.kt index a64d83432..d33c1f860 100644 --- a/app/src/test/java/org/mozilla/fenix/search/SearchInteractorTest.kt +++ b/app/src/test/java/org/mozilla/fenix/search/SearchInteractorTest.kt @@ -189,6 +189,16 @@ class SearchInteractorTest { verify { store.dispatch(SearchFragmentAction.SearchShortcutEngineSelected(searchEngine)) } } + @Test + fun onSearchShortcutsButtonClicked() { + val searchController: SearchController = mockk(relaxed = true) + val interactor = SearchInteractor(searchController) + + interactor.onSearchShortcutsButtonClicked() + + verify { searchController.handleSearchShortcutsButtonClicked() } + } + @Test fun onClickSearchEngineSettings() { val navController: NavController = mockk()