secant-android-wallet/ui-lib/src/androidTest/java/co/electriccoin/zcash/ui/screen/send/SendViewTestSetup.kt

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.platform.LocalContext
import androidx.compose.ui.platform.LocalFocusManager
import androidx.compose.ui.test.junit4.ComposeContentTestRule
import cash.z.ecc.android.sdk.fixture.WalletBalanceFixture
import cash.z.ecc.android.sdk.model.MonetarySeparators
import cash.z.ecc.android.sdk.model.Zatoshi
import cash.z.ecc.android.sdk.model.ZecSend
import cash.z.ecc.android.sdk.type.AddressType
import co.electriccoin.zcash.ui.common.model.WalletRestoringState
import co.electriccoin.zcash.ui.design.theme.ZcashTheme
import co.electriccoin.zcash.ui.fixture.BalanceStateFixture
import co.electriccoin.zcash.ui.fixture.WalletSnapshotFixture
import co.electriccoin.zcash.ui.screen.send.ext.Saver
import co.electriccoin.zcash.ui.screen.send.model.AmountState
import co.electriccoin.zcash.ui.screen.send.model.MemoState
import co.electriccoin.zcash.ui.screen.send.model.RecipientAddressState
import co.electriccoin.zcash.ui.screen.send.model.SendStage
import co.electriccoin.zcash.ui.screen.send.view.Send
import kotlinx.coroutines.flow.MutableStateFlow
import java.util.Locale
import java.util.concurrent.atomic.AtomicInteger

class SendViewTestSetup(
    private val composeTestRule: ComposeContentTestRule,
    private val initialState: SendStage,
    private val initialZecSend: ZecSend?,
    private val hasCameraFeature: Boolean
) {
    private val onBackCount = AtomicInteger(0)
    private val onSettingsCount = AtomicInteger(0)
    private val onCreateCount = AtomicInteger(0)
    private val onScannerCount = 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 getOnSettingsCount(): Int {
        composeTestRule.waitForIdle()
        return onSettingsCount.get()
    }

    fun getOnCreateCount(): Int {
        composeTestRule.waitForIdle()
        return onCreateCount.get()
    }

    fun getOnScannerCount(): Int {
        composeTestRule.waitForIdle()
        return onScannerCount.get()
    }

    fun getLastZecSend(): ZecSend? {
        composeTestRule.waitForIdle()
        return lastZecSend
    }

    fun getLastSendStage(): SendStage? {
        composeTestRule.waitForIdle()
        return lastSendStage
    }

    @Composable
    @Suppress("TestFunctionName")
    fun DefaultContent() {
        val context = LocalContext.current

        val monetarySeparators = MonetarySeparators.current(Locale.getDefault())

        val (sendStage, setSendStage) =
            rememberSaveable(stateSaver = SendStage.Saver) { mutableStateOf(initialState) }

        lastSendStage = sendStage

        val onBackAction = {
            onBackCount.incrementAndGet()
            when (sendStage) {
                SendStage.Form -> {}
                SendStage.Proposing -> {}
                is SendStage.SendFailure -> setSendStage(SendStage.Form)
            }
        }

        BackHandler {
            onBackAction()
        }

        val (zecSend, setZecSend) =
            rememberSaveable(stateSaver = ZecSend.Saver) { mutableStateOf(initialZecSend) }

        lastZecSend = zecSend

        ZcashTheme {
            // TODO [#1260]: Cover Send.Form screen UI with tests
            // TODO [#1260]: https://github.com/Electric-Coin-Company/zashi-android/issues/1260
            Send(
                balanceState = BalanceStateFixture.new(),
                sendStage = sendStage,
                onCreateZecSend = setZecSend,
                focusManager = LocalFocusManager.current,
                onBack = onBackAction,
                onSettings = { onSettingsCount.incrementAndGet() },
                onQrScannerOpen = {
                    onScannerCount.incrementAndGet()
                },
                goBalances = {
                    // TODO [#1194]: Cover Current balances UI widget with tests
                    // TODO [#1194]: https://github.com/Electric-Coin-Company/zashi-android/issues/1194
                },
                hasCameraFeature = hasCameraFeature,
                recipientAddressState = RecipientAddressState("", AddressType.Invalid()),
                onRecipientAddressChange = {
                    // TODO [#1260]: Cover Send.Form screen UI with tests
                    // TODO [#1260]: https://github.com/Electric-Coin-Company/zashi-android/issues/1260
                },
                setAmountState = {},
                amountState = AmountState.new(context, monetarySeparators, "", false),
                setMemoState = {},
                memoState = MemoState.new(""),
                walletRestoringState = WalletRestoringState.NONE,
                walletSnapshot =
                    WalletSnapshotFixture.new(
                        saplingBalance =
                            WalletBalanceFixture.new(
                                available = Zatoshi(Zatoshi.MAX_INCLUSIVE.div(100))
                            )
                    ),
            )
        }
    }

    fun setDefaultContent() {
        composeTestRule.setContent {
            DefaultContent()
        }
    }
}