104 lines
3.0 KiB
Kotlin
104 lines
3.0 KiB
Kotlin
package co.electriccoin.zcash.ui.screen.send
|
|
|
|
import androidx.activity.compose.BackHandler
|
|
import androidx.compose.runtime.Composable
|
|
import androidx.compose.runtime.mutableStateOf
|
|
import androidx.compose.runtime.saveable.rememberSaveable
|
|
import androidx.compose.ui.test.junit4.ComposeContentTestRule
|
|
import cash.z.ecc.android.sdk.model.ZecSend
|
|
import cash.z.ecc.sdk.fixture.ZatoshiFixture
|
|
import co.electriccoin.zcash.ui.design.theme.ZcashTheme
|
|
import co.electriccoin.zcash.ui.screen.send.ext.Saver
|
|
import co.electriccoin.zcash.ui.screen.send.model.SendStage
|
|
import co.electriccoin.zcash.ui.screen.send.view.Send
|
|
import kotlinx.coroutines.flow.MutableStateFlow
|
|
import kotlinx.coroutines.flow.update
|
|
import java.util.concurrent.atomic.AtomicInteger
|
|
|
|
class SendViewTestSetup(
|
|
private val composeTestRule: ComposeContentTestRule,
|
|
private val initialState: SendStage,
|
|
private val initialZecSend: ZecSend?
|
|
) {
|
|
private val onBackCount = AtomicInteger(0)
|
|
private val onCreateCount = AtomicInteger(0)
|
|
val mutableActionExecuted = MutableStateFlow(false)
|
|
|
|
@Volatile
|
|
private var lastSendStage: SendStage? = null
|
|
|
|
@Volatile
|
|
private var lastZecSend: ZecSend? = null
|
|
|
|
fun getOnBackCount(): Int {
|
|
composeTestRule.waitForIdle()
|
|
return onBackCount.get()
|
|
}
|
|
|
|
fun getOnCreateCount(): Int {
|
|
composeTestRule.waitForIdle()
|
|
return onCreateCount.get()
|
|
}
|
|
|
|
fun getLastZecSend(): ZecSend? {
|
|
composeTestRule.waitForIdle()
|
|
return lastZecSend
|
|
}
|
|
|
|
fun getLastSendStage(): SendStage? {
|
|
composeTestRule.waitForIdle()
|
|
return lastSendStage
|
|
}
|
|
|
|
@Composable
|
|
@Suppress("TestFunctionName")
|
|
fun DefaultContent() {
|
|
val (sendStage, setSendStage) =
|
|
rememberSaveable { mutableStateOf(initialState) }
|
|
|
|
lastSendStage = sendStage
|
|
|
|
val onBackAction = {
|
|
onBackCount.incrementAndGet()
|
|
when (sendStage) {
|
|
SendStage.Form -> {}
|
|
SendStage.Confirmation -> setSendStage(SendStage.Form)
|
|
SendStage.Sending -> {}
|
|
SendStage.SendFailure -> setSendStage(SendStage.Form)
|
|
SendStage.SendSuccessful -> {}
|
|
}
|
|
}
|
|
|
|
BackHandler {
|
|
onBackAction()
|
|
}
|
|
|
|
val (zecSend, setZecSend) =
|
|
rememberSaveable(stateSaver = ZecSend.Saver) { mutableStateOf(initialZecSend) }
|
|
|
|
lastZecSend = zecSend
|
|
|
|
ZcashTheme {
|
|
Send(
|
|
mySpendableBalance = ZatoshiFixture.new(),
|
|
sendStage = sendStage,
|
|
onSendStageChange = setSendStage,
|
|
zecSend = zecSend,
|
|
onZecSendChange = setZecSend,
|
|
onBack = onBackAction,
|
|
onCreateAndSend = {
|
|
onCreateCount.incrementAndGet()
|
|
lastZecSend = it
|
|
mutableActionExecuted.update { true }
|
|
}
|
|
)
|
|
}
|
|
}
|
|
|
|
fun setDefaultContent() {
|
|
composeTestRule.setContent {
|
|
DefaultContent()
|
|
}
|
|
}
|
|
}
|