tests: add mock of CoinJoinConfig

dashpay · Oct 30, 2023 · 7da5e2d · 7da5e2d
1 parent 4ec8988
commit 7da5e2d
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/wallet/test/de/schildbach/wallet/util/services/SendCoinsTaskRunnerTest.kt b/wallet/test/de/schildbach/wallet/util/services/SendCoinsTaskRunnerTest.kt
@@ -18,11 +18,16 @@
 package de.schildbach.wallet.util.services
 
 import de.schildbach.wallet.WalletApplication
+import de.schildbach.wallet.data.CoinJoinConfig
 import de.schildbach.wallet.payments.SendCoinsTaskRunner
+import de.schildbach.wallet.service.CoinJoinMode
+import io.mockk.coEvery
 import io.mockk.every
 import io.mockk.mockk
 import junit.framework.TestCase.assertTrue
 import kotlinx.coroutines.ExperimentalCoroutinesApi
+import kotlinx.coroutines.flow.MutableStateFlow
+import kotlinx.coroutines.flow.StateFlow
 import org.bitcoinj.core.Address
 import org.bitcoinj.core.Coin
 import org.bitcoinj.core.Context
@@ -37,10 +42,12 @@ class SendCoinsTaskRunnerTest {
     @Test
     fun sendCoins_coinSelectorSet_correctCoinSelector() {
         val wallet = mockk<Wallet>()
+        val coinJoinConfig = mockk<CoinJoinConfig>()
         every { wallet.context } returns Context(MainNetParams.get())
+        coEvery { coinJoinConfig.observeMode() } returns MutableStateFlow(CoinJoinMode.NONE)
         val application = mockk<WalletApplication>()
 
-        val sendCoinsTaskRunner = SendCoinsTaskRunner(application, mockk(), mockk(), mockk())
+        val sendCoinsTaskRunner = SendCoinsTaskRunner(application, mockk(), mockk(), mockk(), coinJoinConfig)
         val request = sendCoinsTaskRunner.createSendRequest(
             Address.fromBase58(MainNetParams.get(), "XjBya4EnibUyxubEA8D2Y8KSrBMW1oHq5U"),
             Coin.COIN,
@@ -54,11 +61,13 @@ class SendCoinsTaskRunnerTest {
     @Test
     fun sendCoins_nullCoinSelector_zeroConfSelectorByDefault() {
         val wallet = mockk<Wallet>()
+        val coinJoinConfig = mockk<CoinJoinConfig>()
         every { wallet.context } returns Context(MainNetParams.get())
+        coEvery { coinJoinConfig.observeMode() } returns MutableStateFlow(CoinJoinMode.NONE)
         val application = mockk<WalletApplication>()
         every { application.wallet } returns wallet
 
-        val sendCoinsTaskRunner = SendCoinsTaskRunner(application, mockk(), mockk(), mockk())
+        val sendCoinsTaskRunner = SendCoinsTaskRunner(application, mockk(), mockk(), mockk(), coinJoinConfig)
         val request = sendCoinsTaskRunner.createSendRequest(
             Address.fromBase58(MainNetParams.get(), "XjBya4EnibUyxubEA8D2Y8KSrBMW1oHq5U"),
             Coin.COIN,