Update unit tests for ReaderTagsFeedViewModel

Author: Thomas Horta

WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt

@@ -14,7 +14,9 @@ import org.junit.Test
 import org.mockito.Mock
 import org.mockito.Mockito
 import org.mockito.kotlin.any
+import org.mockito.kotlin.anyOrNull
 import org.mockito.kotlin.doSuspendableAnswer
+import org.mockito.kotlin.mock
 import org.mockito.kotlin.times
 import org.mockito.kotlin.verify
 import org.mockito.kotlin.verifyNoInteractions
@@ -35,12 +37,14 @@ import org.wordpress.android.ui.reader.discover.ReaderPostCardActionsHandler
 import org.wordpress.android.ui.reader.discover.ReaderPostMoreButtonUiStateBuilder
 import org.wordpress.android.ui.reader.discover.ReaderPostUiStateBuilder
 import org.wordpress.android.ui.reader.exceptions.ReaderPostFetchException
+import org.wordpress.android.ui.reader.repository.ReaderAnnouncementRepository
 import org.wordpress.android.ui.reader.repository.ReaderPostRepository
 import org.wordpress.android.ui.reader.repository.usecases.PostLikeUseCase
 import org.wordpress.android.ui.reader.tracker.ReaderTracker
 import org.wordpress.android.ui.reader.viewmodels.tagsfeed.ReaderTagsFeedUiStateMapper
 import org.wordpress.android.ui.reader.viewmodels.tagsfeed.ReaderTagsFeedViewModel
 import org.wordpress.android.ui.reader.viewmodels.tagsfeed.ReaderTagsFeedViewModel.ActionEvent
+import org.wordpress.android.ui.reader.views.compose.ReaderAnnouncementCardItemData
 import org.wordpress.android.ui.reader.views.compose.tagsfeed.TagsFeedPostItem
 import org.wordpress.android.util.DisplayUtilsWrapper
 import org.wordpress.android.util.NetworkUtilsWrapper
@@ -86,6 +90,9 @@ class ReaderTagsFeedViewModelTest : BaseUnitTest() {
     @Mock
     lateinit var networkUtilsWrapper: NetworkUtilsWrapper
 
+    @Mock
+    lateinit var readerAnnouncementRepository: ReaderAnnouncementRepository
+
     private lateinit var viewModel: ReaderTagsFeedViewModel
 
     private val collectedUiStates: MutableList<ReaderTagsFeedViewModel.UiState> = mutableListOf()
@@ -116,6 +123,7 @@ class ReaderTagsFeedViewModelTest : BaseUnitTest() {
             displayUtilsWrapper = displayUtilsWrapper,
             readerTracker = readerTracker,
             networkUtilsWrapper = networkUtilsWrapper,
+            readerAnnouncementRepository = readerAnnouncementRepository,
         )
         whenever(readerPostCardActionsHandler.navigationEvents)
             .thenReturn(navigationEvents)
@@ -126,6 +134,69 @@ class ReaderTagsFeedViewModelTest : BaseUnitTest() {
         observeErrorMessageEvents()
     }
 
+    @Test
+    fun `when tags changed, then UI state should update properly`() = testCollectingUiStates {
+        // Given
+        val tag = ReaderTestUtils.createTag("tag")
+        mockMapInitialTagFeedItems()
+
+        // When
+        viewModel.onTagsChanged(listOf(tag))
+        advanceUntilIdle()
+
+        // Then
+        assertThat(collectedUiStates).contains(
+            ReaderTagsFeedViewModel.UiState.Loaded(
+                data = listOf(getInitialTagFeedItem(tag)),
+            )
+        )
+    }
+
+    @Test
+    fun `given has announcement, when tags changed, then UI state should update properly`() = testCollectingUiStates {
+        // Given
+        val tag = ReaderTestUtils.createTag("tag")
+        val announcementItems = listOf<ReaderAnnouncementCardItemData>(mock(), mock())
+        mockMapInitialTagFeedItems()
+        whenever(readerAnnouncementRepository.hasReaderAnnouncement()).thenReturn(true)
+        whenever(readerAnnouncementRepository.getReaderAnnouncementItems()).thenReturn(announcementItems)
+
+        // When
+        viewModel.onTagsChanged(listOf(tag))
+        advanceUntilIdle()
+
+        // Then
+        val loadedState = collectedUiStates.last() as ReaderTagsFeedViewModel.UiState.Loaded
+        assertThat(loadedState.data).isEqualTo(listOf(getInitialTagFeedItem(tag)))
+        assertThat(loadedState.announcementItem!!.items).isEqualTo(announcementItems)
+    }
+
+    @Test
+    fun `given has announcement, when done clicked, then UI state should update properly`() = testCollectingUiStates {
+        // Given
+        val tag = ReaderTestUtils.createTag("tag")
+        val announcementItems = listOf<ReaderAnnouncementCardItemData>(mock(), mock())
+        mockMapInitialTagFeedItems()
+        whenever(readerAnnouncementRepository.hasReaderAnnouncement()).thenReturn(true)
+        whenever(readerAnnouncementRepository.getReaderAnnouncementItems()).thenReturn(announcementItems)
+
+        viewModel.onTagsChanged(listOf(tag))
+        advanceUntilIdle()
+
+        // When
+        val loadedState = collectedUiStates.last() as ReaderTagsFeedViewModel.UiState.Loaded
+        loadedState.announcementItem!!.onDoneClicked()
+        advanceUntilIdle()
+
+        // Then
+        verify(readerAnnouncementRepository).dismissReaderAnnouncement()
+        assertThat(collectedUiStates.last()).isEqualTo(
+            ReaderTagsFeedViewModel.UiState.Loaded(
+                data = listOf(getInitialTagFeedItem(tag)),
+            )
+        )
+    }
+
     @Test
     fun `given valid tag, when loaded, then UI state should update properly`() = testCollectingUiStates {
         // Given
@@ -899,13 +970,18 @@ class ReaderTagsFeedViewModelTest : BaseUnitTest() {
         }
 
     private fun mockMapInitialTagFeedItems() {
-        whenever(readerTagsFeedUiStateMapper.mapInitialPostsUiState(any(), any(), any(), any(), any(), any()))
-            .thenAnswer {
-                val tags = it.getArgument<List<ReaderTag>>(0)
-                ReaderTagsFeedViewModel.UiState.Loaded(
-                    tags.map { tag -> getInitialTagFeedItem(tag) }
-                )
-            }
+        whenever(
+            readerTagsFeedUiStateMapper.mapInitialPostsUiState(
+                any(), anyOrNull(), any(), any(), any(), any(), any()
+            )
+        ).thenAnswer {
+            val tags = it.getArgument<List<ReaderTag>>(0)
+            val announcementItem = it.getArgument<ReaderTagsFeedViewModel.ReaderAnnouncementItem?>(1)
+            ReaderTagsFeedViewModel.UiState.Loaded(
+                data = tags.map { tag -> getInitialTagFeedItem(tag) },
+                announcementItem = announcementItem,
+            )
+        }
     }
 
     private fun mockMapLoadingTagFeedItems() {