Skip to content

Commit 17be05f

Browse files
committed
chore: update tests
1 parent 692d7fc commit 17be05f

File tree

2 files changed

+6
-61
lines changed

2 files changed

+6
-61
lines changed

app/src/test/java/to/bitkit/repositories/LightningRepoTest.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ class LightningRepoTest : BaseUnitTest() {
6363

6464
@Before
6565
fun setUp() {
66-
wheneverBlocking { coreService.checkGeoBlock() }.thenReturn(Pair(false, false))
66+
wheneverBlocking { coreService.isGeoBlocked() }.thenReturn(false)
6767
sut = LightningRepo(
6868
bgDispatcher = testDispatcher,
6969
lightningService = lightningService,

app/src/test/java/to/bitkit/repositories/WalletRepoTest.kt

Lines changed: 5 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ class WalletRepoTest : BaseUnitTest() {
7373

7474
@Before
7575
fun setUp() = runBlocking {
76-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(false, false))
76+
whenever(coreService.isGeoBlocked()).thenReturn(false)
7777
whenever(cacheStore.data).thenReturn(flowOf(AppCacheData(bolt11 = "", onchainAddress = ADDRESS)))
7878
whenever(lightningRepo.lightningState).thenReturn(MutableStateFlow(LightningState()))
7979
whenever(lightningRepo.nodeEvents).thenReturn(MutableSharedFlow())
@@ -176,28 +176,6 @@ class WalletRepoTest : BaseUnitTest() {
176176
verify(lightningRepo).newAddress()
177177
}
178178

179-
@Test
180-
fun `refreshBip21 should set receiveOnSpendingBalance false when shouldBlockLightning is true`() = test {
181-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(true, true))
182-
whenever(lightningRepo.newAddress()).thenReturn(Result.success(ADDRESS_NEW))
183-
184-
val result = sut.refreshBip21()
185-
186-
assertTrue(result.isSuccess)
187-
assertEquals(false, sut.walletState.value.receiveOnSpendingBalance)
188-
}
189-
190-
@Test
191-
fun `refreshBip21 should set receiveOnSpendingBalance true when shouldBlockLightning is false`() = test {
192-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(true, false))
193-
whenever(lightningRepo.newAddress()).thenReturn(Result.success(ADDRESS_NEW))
194-
195-
val result = sut.refreshBip21()
196-
197-
assertTrue(result.isSuccess)
198-
assertEquals(true, sut.walletState.value.receiveOnSpendingBalance)
199-
}
200-
201179
@Test
202180
fun `refreshBip21 should generate new address when current has transactions`() = test {
203181
whenever(cacheStore.data).thenReturn(flowOf(AppCacheData(onchainAddress = ADDRESS)))
@@ -333,28 +311,6 @@ class WalletRepoTest : BaseUnitTest() {
333311
assertTrue(result.contains("lightning=testInvoice"))
334312
}
335313

336-
@Test
337-
fun `toggleReceiveOnSpendingBalance should toggle state`() = test {
338-
val initialValue = sut.walletState.value.receiveOnSpendingBalance
339-
340-
sut.toggleReceiveOnSpendingBalance()
341-
342-
assertEquals(!initialValue, sut.walletState.value.receiveOnSpendingBalance)
343-
}
344-
345-
@Test
346-
fun `toggleReceiveOnSpendingBalance should return failure if shouldBlockLightning is true`() = test {
347-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(true, true))
348-
349-
if (sut.walletState.value.receiveOnSpendingBalance) {
350-
sut.toggleReceiveOnSpendingBalance()
351-
}
352-
353-
val result = sut.toggleReceiveOnSpendingBalance()
354-
355-
assert(result.isFailure)
356-
}
357-
358314
@Test
359315
fun `addTagToSelected should add tag and update lastUsedTags`() = test {
360316
// Set address in wallet state so paymentId() returns it
@@ -428,20 +384,9 @@ class WalletRepoTest : BaseUnitTest() {
428384
assertEquals(error, result.exceptionOrNull())
429385
}
430386

431-
@Test
432-
fun `shouldRequestAdditionalLiquidity should return false when geoBlocked is true`() = test {
433-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(true, false))
434-
sut.toggleReceiveOnSpendingBalance() // Set to false (initial is true)
435-
436-
val result = sut.shouldRequestAdditionalLiquidity()
437-
438-
assertTrue(result.isSuccess)
439-
assertFalse(result.getOrThrow())
440-
}
441-
442387
@Test
443388
fun `shouldRequestAdditionalLiquidity should return false when geo status is true`() = test {
444-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(true, false))
389+
whenever(coreService.isGeoBlocked()).thenReturn(true)
445390

446391
val result = sut.shouldRequestAdditionalLiquidity()
447392

@@ -451,7 +396,7 @@ class WalletRepoTest : BaseUnitTest() {
451396

452397
@Test
453398
fun `shouldRequestAdditionalLiquidity should return true when amount exceeds inbound capacity`() = test {
454-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(false, false))
399+
whenever(coreService.isGeoBlocked()).thenReturn(false)
455400
whenever(lightningRepo.lightningState).thenReturn(MutableStateFlow(LightningState(channels = channels)))
456401
sut.updateBip21Invoice(amountSats = 1000uL)
457402

@@ -463,7 +408,7 @@ class WalletRepoTest : BaseUnitTest() {
463408

464409
@Test
465410
fun `should not request additional liquidity for 0 channels`() = test {
466-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(false, false))
411+
whenever(coreService.isGeoBlocked()).thenReturn(false)
467412
whenever(lightningRepo.lightningState).thenReturn(MutableStateFlow(LightningState()))
468413
sut.updateBip21Invoice(amountSats = 1000uL)
469414

@@ -475,7 +420,7 @@ class WalletRepoTest : BaseUnitTest() {
475420

476421
@Test
477422
fun `shouldRequestAdditionalLiquidity should return false when amount is less than inbound capacity`() = test {
478-
whenever(coreService.checkGeoBlock()).thenReturn(Pair(false, false))
423+
whenever(coreService.isGeoBlocked()).thenReturn(false)
479424
whenever(lightningRepo.lightningState).thenReturn(MutableStateFlow(LightningState(channels = channels)))
480425
sut.updateBip21Invoice(amountSats = 900uL)
481426

0 commit comments

Comments
 (0)