Delete the users vault data on logout (#418)

This commit is contained in:
David Perez
2023-12-19 12:29:34 -06:00
committed by GitHub
parent 89389e8a9c
commit ea4825b8b8
5 changed files with 32 additions and 1 deletions

View File

@@ -74,8 +74,9 @@ class AuthRepositoryTest {
private val identityService: IdentityService = mockk()
private val haveIBeenPwnedService: HaveIBeenPwnedService = mockk()
private val mutableVaultStateFlow = MutableStateFlow(VAULT_STATE)
private val vaultRepository: VaultRepository = mockk() {
private val vaultRepository: VaultRepository = mockk {
every { vaultStateFlow } returns mutableVaultStateFlow
every { deleteVaultData(any()) } just runs
every { lockVaultIfNecessary(any()) } just runs
every { clearUnlockedData() } just runs
}
@@ -999,6 +1000,7 @@ class AuthRepositoryTest {
userId = USER_ID_1,
organizationKeys = null,
)
verify { vaultRepository.deleteVaultData(userId = USER_ID_1) }
verify { vaultRepository.clearUnlockedData() }
verify { vaultRepository.lockVaultIfNecessary(userId = USER_ID_1) }
}
@@ -1081,6 +1083,7 @@ class AuthRepositoryTest {
userId = USER_ID_1,
organizationKeys = null,
)
verify { vaultRepository.deleteVaultData(userId = USER_ID_1) }
verify { vaultRepository.clearUnlockedData() }
verify { vaultRepository.lockVaultIfNecessary(userId = USER_ID_1) }
}
@@ -1131,6 +1134,7 @@ class AuthRepositoryTest {
userId = USER_ID_2,
organizationKeys = null,
)
verify { vaultRepository.deleteVaultData(userId = USER_ID_2) }
verify(exactly = 0) { vaultRepository.clearUnlockedData() }
verify { vaultRepository.lockVaultIfNecessary(userId = USER_ID_2) }
}

View File

@@ -243,6 +243,18 @@ class VaultRepositoryTest {
}
}
@Test
fun `deleteVaultData should call deleteVaultData on VaultDiskSource`() {
val userId = "userId-1234"
coEvery { vaultDiskSource.deleteVaultData(userId) } just runs
vaultRepository.deleteVaultData(userId = userId)
coVerify(exactly = 1) {
vaultDiskSource.deleteVaultData(userId)
}
}
@Suppress("MaxLineLength")
@Test
fun `sync with syncService Success should unlock the vault for orgs if necessary and update AuthDiskSource and DataStateFlows`() =