def test_ui_confirm_withdraw(logged_in_wallet_user_browser: DriverAPI,
dbsession: Session, user_phone_number,
top_up_user, eth_asset_id):
"""Create new account through UI."""
# Go to address
b = logged_in_wallet_user_browser
b.find_by_css("#nav-wallet").click()
b.find_by_css(
"#row-asset-{} a.withdraw-asset".format(eth_asset_id)).click()
b.fill("address", TEST_ADDRESS)
b.fill("amount", "0.1")
b.find_by_css("button[name='process']").click()
# We should arrive to the confirmation page
assert b.is_element_present_by_css("#heading-confirm-withdraw")
# Peek into SMS code
with transaction.manager:
# Withdraw is the firt user op on the stack
withdraw = dbsession.query(UserCryptoOperation).join(
CryptoOperation).order_by(
CryptoOperation.created_at.desc()).first()
confirmation = UserWithdrawConfirmation.get_pending_confirmation(
withdraw)
sms_code = confirmation.other_data["sms_code"]
b.fill("code", sms_code)
b.find_by_css("button[name='confirm']").click()
# We got text box telling withdraw was success
assert b.is_element_present_by_css("#msg-withdraw-confirmed")
def test_ui_confirm_withdraw(
logged_in_wallet_user_browser: DriverAPI, dbsession: Session, user_phone_number, top_up_user, eth_asset_id
):
"""Create new account through UI."""
# Go to address
b = logged_in_wallet_user_browser
b.find_by_css("#nav-wallet").click()
b.find_by_css("#row-asset-{} a.withdraw-asset".format(eth_asset_id)).click()
b.fill("address", TEST_ADDRESS)
b.fill("amount", "0.1")
b.find_by_css("button[name='process']").click()
# We should arrive to the confirmation page
assert b.is_element_present_by_css("#heading-confirm-withdraw")
# Peek into SMS code
with transaction.manager:
# Withdraw is the firt user op on the stack
withdraw = (
dbsession.query(UserCryptoOperation)
.join(CryptoOperation)
.order_by(CryptoOperation.created_at.desc())
.first()
)
confirmation = UserWithdrawConfirmation.get_pending_confirmation(withdraw)
sms_code = confirmation.other_data["sms_code"]
b.fill("code", sms_code)
b.find_by_css("button[name='confirm']").click()
# We got text box telling withdraw was success
assert b.is_element_present_by_css("#msg-withdraw-confirmed")
def test_confirm_user_withdraw_timeout(dbsession, eth_network_id, eth_asset_id, user_id, topped_up_user):
"""User did not reply to withdraw confirmation within the timeout."""
with transaction.manager:
uca = dbsession.query(UserCryptoAddress).first()
asset = dbsession.query(Asset).get(eth_asset_id)
withdraw_op = uca.withdraw(asset, Decimal(5), eth_address_to_bin(TEST_ADDRESS), "Foobar", 1)
UserWithdrawConfirmation.require_confirmation(withdraw_op)
with transaction.manager:
ManualConfirmation.run_timeout_checks(dbsession, now() + timedelta(hours=12))
with transaction.manager:
confirmation = dbsession.query(UserWithdrawConfirmation).first()
assert confirmation.action_taken_at
assert confirmation.state == ManualConfirmationState.timed_out
assert confirmation.user_crypto_operation.crypto_operation.state == CryptoOperationState.cancelled
assert "error" in confirmation.user_crypto_operation.crypto_operation.other_data
def test_confirm_user_withdraw_cancelled(dbsession, eth_network_id, eth_asset_id, user_id, topped_up_user):
"""User cancels the withdraw confirmation."""
with transaction.manager:
uca = dbsession.query(UserCryptoAddress).first()
asset = dbsession.query(Asset).get(eth_asset_id)
original_balance = uca.get_crypto_account(asset).account.get_balance()
withdraw_op = uca.withdraw(asset, Decimal(7), eth_address_to_bin(TEST_ADDRESS), "Foobar", 1)
UserWithdrawConfirmation.require_confirmation(withdraw_op)
with transaction.manager:
confirmation = dbsession.query(UserWithdrawConfirmation).first()
confirmation.cancel()
assert confirmation.action_taken_at
assert confirmation.state == ManualConfirmationState.cancelled
assert confirmation.user_crypto_operation.crypto_operation.state == CryptoOperationState.cancelled
assert "error" in confirmation.user_crypto_operation.crypto_operation.other_data
# The balance was restored
uca = dbsession.query(UserCryptoAddress).first()
asset = dbsession.query(Asset).get(eth_asset_id)
assert uca.get_crypto_account(asset).account.get_balance() == original_balance
def test_confirm_user_withdraw_success(dbsession, eth_network_id, eth_asset_id, user_id, topped_up_user):
"""SMS confirmation success."""
with transaction.manager:
uca = dbsession.query(UserCryptoAddress).first()
asset = dbsession.query(Asset).get(eth_asset_id)
withdraw_op = uca.withdraw(asset, Decimal(5), eth_address_to_bin(TEST_ADDRESS), "Foobar", 1)
confirmation = UserWithdrawConfirmation.require_confirmation(withdraw_op)
assert confirmation.created_at
assert confirmation.confirmation_type == ManualConfirmationType.sms
assert confirmation.state == ManualConfirmationState.pending
with transaction.manager:
confirmation = dbsession.query(UserWithdrawConfirmation).first()
code = confirmation.other_data["sms_code"]
confirmation.resolve_sms(code, None)
assert confirmation.action_taken_at
assert confirmation.state == ManualConfirmationState.resolved
assert confirmation.user_crypto_operation.crypto_operation.state == CryptoOperationState.waiting