def test_apply_transaction_raises_if_cant_be_applied_to_sender_wallet(redis, mocker):
manager = WalletManager()
block = Block()
transaction = TransferTransaction()
transaction.type = TRANSACTION_TYPE_TRANSFER
transaction.fee = 10000
transaction.amount = 430000
transaction.id = "hehe"
transaction.sender_public_key = (
"020f5df4d2bc736d12ce43af5b1663885a893fade7ee5e62b3cc59315a63e6a325"
)
transaction.recipient_id = "AZYnpgXS3x43nxqhT4q29sZScRwZeNKLpW"
redis.set(
"wallets:address:AThM5PNSKdU9pu1ydqQnzRWVeNCGr8HKof",
Wallet(
{
"address": "AThM5PNSKdU9pu1ydqQnzRWVeNCGr8HKof",
"public_key": (
"020f5df4d2bc736d12ce43af5b1663885a893fade7ee5e62b3cc59315a63e6a325"
),
"balance": 5,
}
).to_json(),
)
with pytest.raises(Exception) as excinfo:
manager.apply_transaction(transaction, block)
assert str(excinfo.value) == (
"Can't apply transaction hehe from sender AThM5PNSKdU9pu1ydqQnzRWVeNCGr8HKof"
)
def test_apply_transaction_force_apply_skips_can_be_applied_check(redis, mocker):
manager = WalletManager()
balances_mock = mocker.patch(
"chain.plugins.database.wallet_manager.WalletManager._update_vote_balances"
)
mocker.patch(
"chain.plugins.database.wallet_manager.is_transaction_exception",
return_value=True,
)
block = Block()
transaction = TransferTransaction()
transaction.type = TRANSACTION_TYPE_TRANSFER
transaction.fee = 10000
transaction.amount = 430000
transaction.id = "hehe"
transaction.sender_public_key = (
"020f5df4d2bc736d12ce43af5b1663885a893fade7ee5e62b3cc59315a63e6a325"
)
transaction.recipient_id = "AZYnpgXS3x43nxqhT4q29sZScRwZeNKLpW"
redis.set(
"wallets:address:AThM5PNSKdU9pu1ydqQnzRWVeNCGr8HKof",
Wallet(
{
"address": "AThM5PNSKdU9pu1ydqQnzRWVeNCGr8HKof",
"public_key": (
"020f5df4d2bc736d12ce43af5b1663885a893fade7ee5e62b3cc59315a63e6a325"
),
"balance": 0,
}
).to_json(),
)
manager.apply_transaction(transaction, block)
# updated sender wallet is saved back to redis
sender = Wallet(
json.loads(redis.get("wallets:address:AThM5PNSKdU9pu1ydqQnzRWVeNCGr8HKof"))
)
assert sender.balance == -440000
# updated recipient wallet is saved back to redis
recipient = Wallet(
json.loads(redis.get("wallets:address:AZYnpgXS3x43nxqhT4q29sZScRwZeNKLpW"))
)
assert recipient.balance == 430000
balances_mock.assert_called_once()