From 76f15a1f7fae218fc231e1067ec3586df34cf1af Mon Sep 17 00:00:00 2001
From: mcarare <mihai.carare.dev@gmail.com>
Date: Mon, 30 Mar 2020 14:41:48 +0300
Subject: [PATCH] For #9498: Update unit tests with new constructor argument

---
 .../DefaultBrowserToolbarControllerTest.kt    | 46 +++++++++++++------
 1 file changed, 31 insertions(+), 15 deletions(-)

diff --git a/app/src/test/java/org/mozilla/fenix/components/toolbar/DefaultBrowserToolbarControllerTest.kt b/app/src/test/java/org/mozilla/fenix/components/toolbar/DefaultBrowserToolbarControllerTest.kt
index 20aff69d1..4e6cafce7 100644
--- a/app/src/test/java/org/mozilla/fenix/components/toolbar/DefaultBrowserToolbarControllerTest.kt
+++ b/app/src/test/java/org/mozilla/fenix/components/toolbar/DefaultBrowserToolbarControllerTest.kt
@@ -109,7 +109,8 @@ class DefaultBrowserToolbarControllerTest {
             bookmarkTapped = mockk(),
             readerModeController = mockk(),
             sessionManager = mockk(),
-            store = mockk()
+            store = mockk(),
+            sharedViewModel = mockk()
         )
 
         mockkStatic(
@@ -203,7 +204,8 @@ class DefaultBrowserToolbarControllerTest {
             bookmarkTapped = mockk(),
             readerModeController = mockk(),
             sessionManager = mockk(),
-            store = mockk()
+            store = mockk(),
+            sharedViewModel = mockk()
         )
 
         controller.handleBrowserMenuDismissed(itemList)
@@ -221,10 +223,10 @@ class DefaultBrowserToolbarControllerTest {
 
         verify { metrics.track(Event.SearchBarTapped(Event.SearchBarTapped.Source.BROWSER)) }
         verify {
-                val directions = BrowserFragmentDirections.actionBrowserFragmentToSearchFragment(
-                    sessionId = "1"
-                )
-                navController.nav(R.id.browserFragment, directions)
+            val directions = BrowserFragmentDirections.actionBrowserFragmentToSearchFragment(
+                sessionId = "1"
+            )
+            navController.nav(R.id.browserFragment, directions)
         }
     }
 
@@ -355,7 +357,8 @@ class DefaultBrowserToolbarControllerTest {
             bookmarkTapped = mockk(),
             readerModeController = mockk(),
             sessionManager = mockk(),
-            store = mockk()
+            store = mockk(),
+            sharedViewModel = mockk()
         )
         controller.ioScope = this
 
@@ -439,10 +442,16 @@ class DefaultBrowserToolbarControllerTest {
 
         controller.handleToolbarItemInteraction(item)
 
-        verify { metrics.track(
-            Event.BrowserMenuItemTapped(Event.BrowserMenuItemTapped.Item.SAVE_TO_COLLECTION)) }
-        verify { metrics.track(
-            Event.CollectionSaveButtonPressed(DefaultBrowserToolbarController.TELEMETRY_BROWSER_IDENTIFIER)) }
+        verify {
+            metrics.track(
+                Event.BrowserMenuItemTapped(Event.BrowserMenuItemTapped.Item.SAVE_TO_COLLECTION)
+            )
+        }
+        verify {
+            metrics.track(
+                Event.CollectionSaveButtonPressed(DefaultBrowserToolbarController.TELEMETRY_BROWSER_IDENTIFIER)
+            )
+        }
         verify {
             val directions =
                 BrowserFragmentDirections.actionBrowserFragmentToCreateCollectionFragment(
@@ -465,8 +474,13 @@ class DefaultBrowserToolbarControllerTest {
         controller.handleToolbarItemInteraction(item)
 
         verify { metrics.track(Event.BrowserMenuItemTapped(Event.BrowserMenuItemTapped.Item.SAVE_TO_COLLECTION)) }
-        verify { metrics.track(Event.CollectionSaveButtonPressed(
-            DefaultBrowserToolbarController.TELEMETRY_BROWSER_IDENTIFIER)) }
+        verify {
+            metrics.track(
+                Event.CollectionSaveButtonPressed(
+                    DefaultBrowserToolbarController.TELEMETRY_BROWSER_IDENTIFIER
+                )
+            )
+        }
         verify {
             val directions =
                 BrowserFragmentDirections.actionBrowserFragmentToCreateCollectionFragment(
@@ -498,7 +512,8 @@ class DefaultBrowserToolbarControllerTest {
             bookmarkTapped = mockk(),
             readerModeController = mockk(),
             sessionManager = mockk(),
-            store = mockk()
+            store = mockk(),
+            sharedViewModel = mockk()
         )
 
         val sessionManager: SessionManager = mockk(relaxed = true)
@@ -539,7 +554,8 @@ class DefaultBrowserToolbarControllerTest {
             bookmarkTapped = mockk(),
             readerModeController = mockk(),
             sessionManager = mockk(),
-            store = mockk()
+            store = mockk(),
+            sharedViewModel = mockk()
         )
 
         controller.handleToolbarItemInteraction(item)