From be81a14b04ee4b3fe4b0c0c9d0522d2dc99c3946 Mon Sep 17 00:00:00 2001
From: Mihai Branescu <branescu.mihai2@gmail.com>
Date: Fri, 18 Oct 2019 15:43:39 +0300
Subject: [PATCH] For #6018 - Added unit tests for interactor and controller

---
 .../search/DefaultSearchControllerTest.kt      | 18 ++++++++++++++++++
 .../fenix/search/SearchInteractorTest.kt       | 10 ++++++++++
 2 files changed, 28 insertions(+)

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()