For #11841: Removed topsite bug on private browsing (#12020)

mozilla-mobile · Jun 26, 2020 · f3b44c0 · f3b44c0
1 parent 005ad6f
commit f3b44c0
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/app/src/main/java/org/mozilla/fenix/home/HomeFragment.kt b/app/src/main/java/org/mozilla/fenix/home/HomeFragment.kt
@@ -248,11 +248,17 @@ class HomeFragment : Fragment() {
      * data in our store. The [View.consumeFrom] coroutine dispatch
      * doesn't get run right away which means that we won't draw on the first layout pass.
      */
-    fun updateSessionControlView(view: View) {
-        sessionControlView?.update(homeFragmentStore.state)
+    private fun updateSessionControlView(view: View) {
+        if (browsingModeManager.mode == BrowsingMode.Private) {
+            view.consumeFrom(homeFragmentStore, viewLifecycleOwner) {
+                sessionControlView?.update(it)
+            }
+        } else {
+            sessionControlView?.update(homeFragmentStore.state)
 
-        view.consumeFrom(homeFragmentStore, viewLifecycleOwner) {
-            sessionControlView?.update(it)
+            view.consumeFrom(homeFragmentStore, viewLifecycleOwner) {
+                sessionControlView?.update(it)
+            }
         }
     }