Fix unit tests

Author: hichamboushaba

WooCommerce/src/test/kotlin/com/woocommerce/android/ui/common/UserEligibilityErrorViewModelTest.kt

@@ -6,42 +6,32 @@ import com.woocommerce.android.R.string
 import com.woocommerce.android.extensions.takeIfNotEqualTo
 import com.woocommerce.android.model.toAppModel
 import com.woocommerce.android.ui.common.UserEligibilityErrorViewModel.ViewState
+import com.woocommerce.android.ui.login.AccountRepository
 import com.woocommerce.android.viewmodel.BaseUnitTest
-import com.woocommerce.android.viewmodel.MultiLiveEvent.Event.*
+import com.woocommerce.android.viewmodel.MultiLiveEvent.Event.Exit
+import com.woocommerce.android.viewmodel.MultiLiveEvent.Event.Logout
+import com.woocommerce.android.viewmodel.MultiLiveEvent.Event.ShowSnackbar
 import kotlinx.coroutines.ExperimentalCoroutinesApi
 import org.assertj.core.api.Assertions.assertThat
 import org.junit.Before
 import org.junit.Test
-import org.mockito.kotlin.KArgumentCaptor
 import org.mockito.kotlin.any
-import org.mockito.kotlin.argumentCaptor
-import org.mockito.kotlin.clearInvocations
 import org.mockito.kotlin.doReturn
 import org.mockito.kotlin.mock
-import org.mockito.kotlin.spy
 import org.mockito.kotlin.times
 import org.mockito.kotlin.verify
 import org.mockito.kotlin.whenever
-import org.wordpress.android.fluxc.Dispatcher
-import org.wordpress.android.fluxc.action.AccountAction.SIGN_OUT
-import org.wordpress.android.fluxc.action.SiteAction
-import org.wordpress.android.fluxc.annotations.action.Action
 import org.wordpress.android.fluxc.model.user.WCUserModel
-import org.wordpress.android.fluxc.store.AccountStore
-import org.wordpress.android.fluxc.store.AccountStore.OnAccountChanged
-import kotlin.test.assertEquals
 import kotlin.test.assertFalse
 import kotlin.test.assertTrue
 
 @ExperimentalCoroutinesApi
 class UserEligibilityErrorViewModelTest : BaseUnitTest() {
     private val appPrefsWrapper: AppPrefs = mock()
-    private val dispatcher: Dispatcher = mock()
-    private val accountStore: AccountStore = mock()
+    private val accountRepository: AccountRepository = mock()
     private val userEligibilityFetcher: UserEligibilityFetcher = mock()
 
     private lateinit var viewModel: UserEligibilityErrorViewModel
-    private lateinit var actionCaptor: KArgumentCaptor<Action<*>>
 
     private val testUser = WCUserModel().apply {
         remoteUserId = 1L
@@ -56,22 +46,11 @@ class UserEligibilityErrorViewModelTest : BaseUnitTest() {
 
     @Before
     fun setup() {
-        actionCaptor = argumentCaptor()
-
-        viewModel = spy(
-            UserEligibilityErrorViewModel(
-                SavedStateHandle(),
-                appPrefsWrapper,
-                dispatcher,
-                accountStore,
-                userEligibilityFetcher
-            )
-        )
-
-        clearInvocations(
-            viewModel,
-            userEligibilityFetcher,
-            appPrefsWrapper
+        viewModel = UserEligibilityErrorViewModel(
+            SavedStateHandle(),
+            appPrefsWrapper,
+            accountRepository,
+            userEligibilityFetcher
         )
     }
 
@@ -149,23 +128,17 @@ class UserEligibilityErrorViewModelTest : BaseUnitTest() {
     }
 
     @Test
-    fun `Handles logout button click correctly`() {
-        doReturn(false).whenever(accountStore).hasAccessToken()
+    fun `Handles logout button click correctly`() = testBlocking {
+        doReturn(true).whenever(accountRepository).logout()
 
         var logoutEvent: Logout? = null
         viewModel.event.observeForever {
-            if (it is Logout) logoutEvent = it
+            logoutEvent = it as? Logout
         }
 
         viewModel.onLogoutButtonClicked()
 
-        // note that we expect two dispatches because there's one to sign out the user and
-        // the other to remove WPcom and Jetpack sites from local db
-        verify(dispatcher, times(2)).dispatch(actionCaptor.capture())
-        assertEquals(SIGN_OUT, actionCaptor.firstValue.type)
-        assertEquals(SiteAction.REMOVE_WPCOM_AND_JETPACK_SITES, actionCaptor.secondValue.type)
-
-        viewModel.onAccountChanged(OnAccountChanged())
+        verify(accountRepository).logout()
         assertThat(logoutEvent).isNotNull
     }
 }