[WIP] Updates: the MySiteViewModelTest

wordpress-mobile · Mar 13, 2024 · 42ec262 · 42ec262
1 parent 02192d8
commit 42ec262
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/WordPress/src/test/java/org/wordpress/android/ui/mysite/MySiteViewModelTest.kt b/WordPress/src/test/java/org/wordpress/android/ui/mysite/MySiteViewModelTest.kt
@@ -404,7 +404,7 @@ class MySiteViewModelTest : BaseUnitTest() {
 
         verify(quickStartUtilsWrapper)
             .startQuickStart(site.id, false, quickStartRepository.quickStartType, quickStartTracker)
-//        verify(mySiteSourceManager).refreshQuickStart()
+        verify(dashboardCardsViewModelSlice).startQuickStart(site)
     }
 
     @Test
@@ -425,7 +425,7 @@ class MySiteViewModelTest : BaseUnitTest() {
 
         viewModel.onPostUploaded(postUploadedEvent)
 
-//        verify(mySiteSourceManager).refreshBloggingPrompts(true)
+        verify(dashboardCardsViewModelSlice).refreshBloggingPrompt()
     }
 
     @Test
@@ -450,13 +450,6 @@ class MySiteViewModelTest : BaseUnitTest() {
         verify(analyticsTrackerWrapper, times(0)).track(Stat.MY_SITE_PULL_TO_REFRESH)
     }
 
-    /* CLEARED */
-    @Test
-    fun `when vm cleared() is invoked, then MySiteSource clear() is invoked`() {
-        viewModel.invokeOnCleared()
-
-//        verify(mySiteSourceManager).clear()
-    }
 
     /* LAND ON THE EDITOR A/B EXPERIMENT */
     @Test
@@ -502,6 +495,17 @@ class MySiteViewModelTest : BaseUnitTest() {
             assertThat(viewModel.onShowJetpackIndividualPluginOverlay.value?.peekContent()).isNull()
         }
 
+
+    @Test
+    fun `when onCleared is called, then clears all the vm slices`() {
+        viewModel.invokeOnCleared()
+
+        verify(siteInfoHeaderCardViewModelSlice).onCleared()
+        verify(accountDataViewModelSlice).onCleared()
+        verify(dashboardCardsViewModelSlice).onCleared()
+        verify(dashboardItemsViewModelSlice).onCleared()
+    }
+
     @Suppress("LongParameterList")
     private fun initSelectedSite(
         isQuickStartInProgress: Boolean = false,