def test_tx_update():
sender_wallet = Wallet()
recipient = "b64e8ac4"
amount = 50
transaction = Transaction(sender_wallet, recipient, amount)
next_recipient = "a13b2bf4"
next_amount = 60
# update Tx with new recipient
transaction.tx_update(sender_wallet, next_recipient, next_amount)
assert transaction.output[next_recipient] == next_amount
assert transaction.output[sender_wallet.address] == (
sender_wallet.balance - amount - next_amount)
assert Wallet.verify_signature(transaction.input["public_key"],
transaction.output,
transaction.input["signature"])
# update Tx with existing recipient; repeat of first Tx
transaction.tx_update(sender_wallet, recipient, amount)
assert transaction.output[recipient] == amount * 2
assert transaction.output[sender_wallet.address] == (
sender_wallet.balance - amount * 2 - next_amount)
assert Wallet.verify_signature(transaction.input["public_key"],
transaction.output,
transaction.input["signature"])
def test_tx_update_when_exceeds_balance():
sender_wallet = Wallet()
transaction = Transaction(sender_wallet, "b64e8ac4", 50)
with pytest.raises(Exception, match="Input amount exceeds balance."):
transaction.tx_update(sender_wallet, "a13b2bf4", 10000)
