Fix unit tests

wordpress-mobile · May 20, 2024 · 12e9ef7 · 12e9ef7
1 parent b150ba3
commit 12e9ef7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/WordPress/src/test/java/org/wordpress/android/ui/reader/subfilter/SubFilterViewModelTest.kt b/WordPress/src/test/java/org/wordpress/android/ui/reader/subfilter/SubFilterViewModelTest.kt
@@ -302,6 +302,8 @@ class SubFilterViewModelTest : BaseUnitTest() {
 
     @Test
     fun `view model updates the tags and sites and asks to show the bottom sheet when filters button is tapped`() {
+        mockReaderTableEmpty()
+
         var updateTasks: EnumSet<UpdateTask>? = null
 
         viewModel.updateTagsAndSites.observeForever { updateTasks = it.peekContent() }
@@ -318,6 +320,8 @@ class SubFilterViewModelTest : BaseUnitTest() {
 
     @Test
     fun `view model asks to show the bottom sheet when filters button is tapped`() {
+        mockReaderTableEmpty()
+
         var uiState: BottomSheetUiState? = null
 
         viewModel.bottomSheetUiState.observeForever { uiState = it.peekContent() }
@@ -539,4 +543,11 @@ class SubFilterViewModelTest : BaseUnitTest() {
             assertThat(viewModel.isTitleContainerVisible.getOrAwaitValue()).isEqualTo(isTitleContainerVisible)
         }
     }
+
+    private fun mockReaderTableEmpty() {
+        whenever(initialTag.organization).thenReturn(Organization.NO_ORGANIZATION)
+        whenever(accountStore.hasAccessToken()).thenReturn(true)
+        whenever(readerTagTableWrapper.getFollowedTags()).thenReturn(ReaderTagList())
+        whenever(readerBlogTableWrapper.getFollowedBlogs()).thenReturn(emptyList())
+    }
 }