Fix unit tests missing isRegistered check.

This commit is contained in:
Alex Hart
2025-03-18 13:49:24 -03:00
committed by Cody Henthorne
parent f6aa324d41
commit e3356163bf
2 changed files with 19 additions and 0 deletions

View File

@@ -11,7 +11,9 @@ import io.mockk.mockkObject
import io.mockk.unmockkObject
import org.junit.rules.ExternalResource
import org.thoughtcrime.securesms.keyvalue.AccountValues
import org.thoughtcrime.securesms.keyvalue.BackupValues
import org.thoughtcrime.securesms.keyvalue.EmojiValues
import org.thoughtcrime.securesms.keyvalue.InAppPaymentValues
import org.thoughtcrime.securesms.keyvalue.PhoneNumberPrivacyValues
import org.thoughtcrime.securesms.keyvalue.RegistrationValues
import org.thoughtcrime.securesms.keyvalue.SignalStore
@@ -45,12 +47,20 @@ class MockSignalStoreRule(private val relaxed: Set<KClass<*>> = emptySet()) : Ex
lateinit var emoji: EmojiValues
private set
lateinit var inAppPayments: InAppPaymentValues
private set
lateinit var backup: BackupValues
private set
override fun before() {
account = mockk(relaxed = relaxed.contains(AccountValues::class), relaxUnitFun = true)
phoneNumberPrivacy = mockk(relaxed = relaxed.contains(PhoneNumberPrivacyValues::class), relaxUnitFun = true)
registration = mockk(relaxed = relaxed.contains(RegistrationValues::class), relaxUnitFun = true)
svr = mockk(relaxed = relaxed.contains(SvrValues::class), relaxUnitFun = true)
emoji = mockk(relaxed = relaxed.contains(EmojiValues::class), relaxUnitFun = true)
inAppPayments = mockk(relaxed = relaxed.contains(InAppPaymentValues::class), relaxUnitFun = true)
backup = mockk(relaxed = relaxed.contains(BackupValues::class), relaxUnitFun = true)
mockkObject(SignalStore)
every { SignalStore.account } returns account
@@ -58,6 +68,8 @@ class MockSignalStoreRule(private val relaxed: Set<KClass<*>> = emptySet()) : Ex
every { SignalStore.registration } returns registration
every { SignalStore.svr } returns svr
every { SignalStore.emoji } returns emoji
every { SignalStore.inAppPayments } returns inAppPayments
every { SignalStore.backup } returns backup
}
override fun after() {