diff --git a/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt b/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt index f81950465a4e..3894acccfe46 100644 --- a/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt +++ b/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/ReaderTagsFeedViewModelTest.kt @@ -24,9 +24,12 @@ 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 @@ -34,6 +37,7 @@ import org.wordpress.android.ui.reader.viewmodels.tagsfeed.ReaderTagsFeedUiState 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> + @Mock + lateinit var snackbarEvents: MediatorLiveData> + private lateinit var viewModel: ReaderTagsFeedViewModel private val collectedUiStates: MutableList = 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/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/tagsfeed/ReaderTagsFeedUiStateMapperTest.kt b/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/tagsfeed/ReaderTagsFeedUiStateMapperTest.kt index dd58f3f52b9a..ec6dd4b5b97f 100644 --- a/WordPress/src/test/java/org/wordpress/android/ui/reader/viewmodels/tagsfeed/ReaderTagsFeedUiStateMapperTest.kt +++ b/WordPress/src/test/java/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"