Add tests for show hide blaze overlay view

wordpress-mobile · Jul 26, 2023 · c7a6523 · c7a6523
1 parent b8eaa92
commit c7a6523
Showing 1 changed file with 101 additions and 0 deletions.
diff --git a/...src/test/java/org/wordpress/android/ui/mysite/cards/dashboard/blaze/BlazeViewModelTest.kt b/...src/test/java/org/wordpress/android/ui/mysite/cards/dashboard/blaze/BlazeViewModelTest.kt
@@ -7,12 +7,15 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import org.mockito.Mock
 import org.mockito.junit.MockitoJUnitRunner
+import org.mockito.kotlin.mock
 import org.mockito.kotlin.whenever
 import org.wordpress.android.BaseUnitTest
 import org.wordpress.android.fluxc.Dispatcher
 import org.wordpress.android.fluxc.store.MediaStore
 import org.wordpress.android.ui.blaze.BlazeFeatureUtils
 import org.wordpress.android.ui.blaze.BlazeFlowSource
+import org.wordpress.android.ui.blaze.BlazeUiState
+import org.wordpress.android.ui.blaze.PageUIModel
 import org.wordpress.android.ui.blaze.PostUIModel
 import org.wordpress.android.ui.blaze.blazeoverlay.BlazeViewModel
 import org.wordpress.android.ui.mysite.SelectedSiteRepository
@@ -34,6 +37,8 @@ class BlazeViewModelTest : BaseUnitTest() {
 
     private lateinit var blazeViewModel: BlazeViewModel
 
+    private lateinit var uiStates: MutableList<BlazeUiState>
+    private lateinit var promoteUiState: MutableList<BlazeUiState>
     @Before
     fun setUp() {
         blazeViewModel = BlazeViewModel(
@@ -42,6 +47,14 @@ class BlazeViewModelTest : BaseUnitTest() {
             mediaStore,
             selectedSiteRepository
         )
+        uiStates = mutableListOf()
+        promoteUiState = mutableListOf()
+        blazeViewModel.promoteUiState.observeForever {
+            promoteUiState.add(it)
+        }
+        blazeViewModel.uiState.observeForever {
+            uiStates.add(it)
+        }
     }
 
     @Test
@@ -55,4 +68,92 @@ class BlazeViewModelTest : BaseUnitTest() {
 
         Assertions.assertThat(result).isNotNull
     }
+
+    @Test
+    fun `given blaze overlay shown is false, when started from post list, then uiState is set to promote post`() {
+        whenever(selectedSiteRepository.getSelectedSite()).thenReturn(mock())
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(false)
+
+        val model = PostUIModel(postId = 1L, title = "title", featuredImageId = 1L,
+            url = "url", featuredImageUrl = "featuredImageUrl"
+        )
+        blazeViewModel.start(BlazeFlowSource.POSTS_LIST, model)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.PromoteScreen.PromotePost::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is true, when started from post list, then uiState is set to webview screen`() {
+        whenever(selectedSiteRepository.getSelectedSite()).thenReturn(mock())
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(true)
+
+        val model = PostUIModel(postId = 1L, title = "title", featuredImageId = 1L,
+            url = "url", featuredImageUrl = "featuredImageUrl"
+        )
+        blazeViewModel.start(BlazeFlowSource.POSTS_LIST, model)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.WebViewScreen::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is false, when started from page list, then uiState is set to promote post`() {
+        whenever(selectedSiteRepository.getSelectedSite()).thenReturn(mock())
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(false)
+
+        val model = PageUIModel(pageId = 1L, title = "title", featuredImageId = 1L,
+            url = "url", featuredImageUrl = "featuredImageUrl"
+        )
+        blazeViewModel.start(BlazeFlowSource.PAGES_LIST, model)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.PromoteScreen.PromotePage::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is true, when started from page list, then uiState is set to webview screen`() {
+        whenever(selectedSiteRepository.getSelectedSite()).thenReturn(mock())
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(true)
+
+        val model = PageUIModel(pageId = 1L, title = "title", featuredImageId = 1L,
+            url = "url", featuredImageUrl = "featuredImageUrl"
+        )
+        blazeViewModel.start(BlazeFlowSource.PAGES_LIST, model)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.WebViewScreen::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is false, when started from dashboard card, then uiState is set to site`() {
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(false)
+
+        blazeViewModel.start(BlazeFlowSource.DASHBOARD_CARD, null)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.PromoteScreen.Site::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is true, when started from dashboard card, then uiState is set to webview screen`() {
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(true)
+
+        blazeViewModel.start(BlazeFlowSource.DASHBOARD_CARD, null)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.WebViewScreen::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is false, when started from no campaigns list, then uiState is set to site`() {
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(false)
+
+        blazeViewModel.start(BlazeFlowSource.CAMPAIGN_LISTING_PAGE, null)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.PromoteScreen.Site::class.java)
+    }
+
+    @Test
+    fun `given blaze overlay shown is true, when started from no campaigns list, then uiState is set to webview`() {
+        whenever(blazeFeatureUtils.shouldHideBlazeOverlay()).thenReturn(true)
+
+        blazeViewModel.start(BlazeFlowSource.CAMPAIGN_LISTING_PAGE, null)
+
+        Assertions.assertThat(uiStates.last()).isInstanceOf(BlazeUiState.WebViewScreen::class.java)
+    }
 }