GiftedHacker/tangem-app-android-audited
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Updated on 2026-08-14

Browse source
This commit is contained in:
Tangem 2025-05-29 17:54:17 +04:00
parent 8b0a7f01cf
commit 9c81f9ca51
4 changed files with 27 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

11
data/staking/src/main/java/com/tangem/data/staking/single/DefaultSingleYieldBalanceProducer.kt
View file

@ -45,12 +45,13 @@ internal class DefaultSingleYieldBalanceProducer @AssistedInject constructor(
return multiYieldBalanceSupplier(
params = MultiYieldBalanceProducer.Params(userWalletId = params.userWalletId),
)
.mapNotNull {
val currentStakingIds = getStakingIds()
it.firstOrNull { balance ->
currentStakingIds.contains(balance.getStakingId())
.mapNotNull { balances ->
val currentStakingIds = getStakingIds().ifEmpty {
return@mapNotNull YieldBalance.Unsupported
}
balances.firstOrNull { currentStakingIds.contains(it.getStakingId()) }
?: YieldBalance.Unsupported
}
.distinctUntilChanged()
.flowOn(dispatchers.default)

14
data/staking/src/test/kotlin/com/tangem/data/staking/single/DefaultSingleYieldBalanceProducerTest.kt
View file

@ -184,10 +184,10 @@ internal class DefaultSingleYieldBalanceProducerTest {
fun `test if flow doesn't contain network from params`() = runTest {
val balance = MockYieldBalanceWrapperDTOFactory.createWithBalance(solanaId).toDomain()
val expected = flowOf(setOf(balance))
val yieldBalancesFlow = flowOf(setOf(balance))
val multiParams = MultiYieldBalanceProducer.Params(userWalletId = params.userWalletId)
every { multiNetworkStatusSupplier(multiParams) } returns expected
every { multiNetworkStatusSupplier(multiParams) } returns yieldBalancesFlow
coEvery { stakingIdFactory.create(params.userWalletId, params.currencyId, params.network) } returns stakingIds
val actual = producer.produce()
@ -198,17 +198,18 @@ internal class DefaultSingleYieldBalanceProducerTest {
coVerify { stakingIdFactory.create(params.userWalletId, params.currencyId, params.network) }
Truth.assertThat(values.size).isEqualTo(0)
val expected = YieldBalance.Unsupported
Truth.assertThat(values.first()).isEqualTo(expected)
}
@Test
fun `test if wallet manager facade returns empty set`() = runTest {
val balance = MockYieldBalanceWrapperDTOFactory.createWithBalance(tonId).toDomain()
val expected = flowOf(setOf(balance))
val yieldBalancesFlow = flowOf(setOf(balance))
val multiParams = MultiYieldBalanceProducer.Params(userWalletId = params.userWalletId)
every { multiNetworkStatusSupplier(multiParams) } returns expected
every { multiNetworkStatusSupplier(multiParams) } returns yieldBalancesFlow
coEvery { stakingIdFactory.create(params.userWalletId, params.currencyId, params.network) } returns emptySet()
val actual = producer.produce()
@ -219,7 +220,8 @@ internal class DefaultSingleYieldBalanceProducerTest {
coVerify { stakingIdFactory.create(params.userWalletId, params.currencyId, params.network) }
Truth.assertThat(values.size).isEqualTo(0)
val expected = YieldBalance.Unsupported
Truth.assertThat(values.first()).isEqualTo(expected)
}
private companion object {