Update unit tests

wordpress-mobile · May 10, 2024 · ca7cf66 · ca7cf66
1 parent 5b45a34
commit ca7cf66
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/...s/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt b/...s/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt
@@ -24,16 +24,20 @@ import org.wordpress.android.models.ReaderPost
 import org.wordpress.android.models.ReaderPostList
 import org.wordpress.android.models.ReaderTag
 import org.wordpress.android.models.ReaderTagType
+import org.wordpress.android.ui.pages.SnackbarMessageHolder
 import org.wordpress.android.ui.reader.ReaderTestUtils
 import org.wordpress.android.ui.reader.discover.ReaderNavigationEvents
 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.ReaderPostRepository
 import org.wordpress.android.ui.reader.repository.usecases.PostLikeUseCase
 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.tagsfeed.TagsFeedPostItem
+import org.wordpress.android.util.DisplayUtilsWrapper
 import org.wordpress.android.viewmodel.Event
 import kotlin.test.assertIs
 
@@ -54,9 +58,21 @@ class ReaderTagsFeedViewModelTest : BaseUnitTest() {
     @Mock
     lateinit var postLikeUseCase: PostLikeUseCase
 
+    @Mock
+    lateinit var readerPostMoreButtonUiStateBuilder: ReaderPostMoreButtonUiStateBuilder
+
+    @Mock
+    lateinit var readerPostUiStateBuilder: ReaderPostUiStateBuilder
+
+    @Mock
+    lateinit var displayUtilsWrapper: DisplayUtilsWrapper
+
     @Mock
     lateinit var navigationEvents: MediatorLiveData<Event<ReaderNavigationEvents>>
 
+    @Mock
+    lateinit var snackbarEvents: MediatorLiveData<Event<SnackbarMessageHolder>>
+
     private lateinit var viewModel: ReaderTagsFeedViewModel
 
     private val collectedUiStates: MutableList<ReaderTagsFeedViewModel.UiState> = mutableListOf()
@@ -81,9 +97,14 @@ class ReaderTagsFeedViewModelTest : BaseUnitTest() {
             readerPostCardActionsHandler = readerPostCardActionsHandler,
             readerPostTableWrapper = readerPostTableWrapper,
             postLikeUseCase = postLikeUseCase,
+            readerPostMoreButtonUiStateBuilder = readerPostMoreButtonUiStateBuilder,
+            readerPostUiStateBuilder = readerPostUiStateBuilder,
+            displayUtilsWrapper = displayUtilsWrapper,
         )
         whenever(readerPostCardActionsHandler.navigationEvents)
             .thenReturn(navigationEvents)
+        whenever(readerPostCardActionsHandler.snackbarEvents)
+            .thenReturn(snackbarEvents)
         observeActionEvents()
         observeNavigationEvents()
     }

diff --git a/...va/org/wordpress/android/ui/reader/viewmodels/tagsfeed/ReaderTagsFeedUiStateMapperTest.kt b/...va/org/wordpress/android/ui/reader/viewmodels/tagsfeed/ReaderTagsFeedUiStateMapperTest.kt
@@ -55,7 +55,7 @@ class ReaderTagsFeedUiStateMapperTest : BaseUnitTest() {
         val onSiteClick: (TagsFeedPostItem) -> Unit = {}
         val onPostCardClick: (TagsFeedPostItem) -> Unit = {}
         val onPostLikeClick: (TagsFeedPostItem) -> Unit = {}
-        val onPostMoreMenuClick = {}
+        val onPostMoreMenuClick: (TagsFeedPostItem) -> Unit = {}
 
         val dateLine = "dateLine"
         val numberLikesText = "numberLikesText"