added missing tests

Author: malmstein

duckchat/duckchat-impl/src/main/java/com/duckduckgo/duckchat/impl/ui/DuckChatWebViewFragment.kt

@@ -350,6 +350,7 @@ open class DuckChatWebViewFragment : DuckDuckGoFragment(R.layout.activity_duck_c
     }
 
     private fun configureLegacyOmnibar() {
+        root.findViewById<View>(R.id.includeToolbar).show()
         toolbar?.let {
             it.setNavigationIcon(com.duckduckgo.mobile.android.R.drawable.ic_arrow_left_24)
             it.setNavigationOnClickListener {

duckchat/duckchat-impl/src/test/java/com/duckduckgo/duckchat/impl/ui/DuckChatWebViewViewModelTest.kt

@@ -19,6 +19,7 @@ package com.duckduckgo.duckchat.impl.ui
 import androidx.test.ext.junit.runners.AndroidJUnit4
 import app.cash.turbine.test
 import com.duckduckgo.common.test.CoroutineTestRule
+import com.duckduckgo.duckchat.impl.DuckChatInternal
 import com.duckduckgo.duckchat.impl.ui.DuckChatWebViewViewModel.Command
 import com.duckduckgo.subscriptions.api.SubscriptionStatus
 import com.duckduckgo.subscriptions.api.SubscriptionStatus.AUTO_RENEWABLE
@@ -27,7 +28,9 @@ import com.duckduckgo.subscriptions.api.SubscriptionStatus.INACTIVE
 import com.duckduckgo.subscriptions.api.SubscriptionStatus.UNKNOWN
 import com.duckduckgo.subscriptions.api.Subscriptions
 import kotlinx.coroutines.flow.MutableSharedFlow
+import kotlinx.coroutines.flow.flowOf
 import kotlinx.coroutines.test.runTest
+import org.junit.Assert.assertFalse
 import org.junit.Assert.assertTrue
 import org.junit.Before
 import org.junit.Rule
@@ -43,14 +46,17 @@ class DuckChatWebViewViewModelTest {
     val coroutineTestRule: CoroutineTestRule = CoroutineTestRule()
 
     private val subscriptions: Subscriptions = mock()
+    private val duckChat: DuckChatInternal = mock()
     private val subscriptionStatusFlow = MutableSharedFlow<SubscriptionStatus>()
 
     private lateinit var viewModel: DuckChatWebViewViewModel
 
     @Before
     fun setup() {
         whenever(subscriptions.getSubscriptionStatusFlow()).thenReturn(subscriptionStatusFlow)
-        viewModel = DuckChatWebViewViewModel(subscriptions)
+        whenever(duckChat.observeEnableDuckChatUserSetting()).thenReturn(flowOf(true))
+        whenever(duckChat.observeFullscreenModeUserSetting()).thenReturn(flowOf(true))
+        viewModel = DuckChatWebViewViewModel(subscriptions, duckChat)
     }
 
     @Test
@@ -135,4 +141,38 @@ class DuckChatWebViewViewModelTest {
             }
         }
     }
+
+    @Test
+    fun `fullscreen mode - when flags enabled, then viewstate enabled`() =
+        runTest {
+            whenever(duckChat.observeEnableDuckChatUserSetting()).thenReturn(flowOf(true))
+            whenever(duckChat.observeFullscreenModeUserSetting()).thenReturn(flowOf(true))
+            viewModel = DuckChatWebViewViewModel(
+                subscriptions,
+                duckChat = duckChat,
+            )
+
+            viewModel.viewState.test {
+                val state = awaitItem()
+                assertTrue(state.isDuckChatUserEnabled)
+                assertTrue(state.isFullScreenModeEnabled)
+            }
+        }
+
+    @Test
+    fun `fullscreen mode - when flags disabled, then viewstate disabled`() =
+        runTest {
+            whenever(duckChat.observeEnableDuckChatUserSetting()).thenReturn(flowOf(false))
+            whenever(duckChat.observeFullscreenModeUserSetting()).thenReturn(flowOf(false))
+            viewModel = DuckChatWebViewViewModel(
+                subscriptions,
+                duckChat = duckChat,
+            )
+
+            viewModel.viewState.test {
+                val state = awaitItem()
+                assertFalse(state.isDuckChatUserEnabled)
+                assertFalse(state.isFullScreenModeEnabled)
+            }
+        }
 }