def test_transaction_update():
sender_wallet = Wallet()
first_recipient = 'first_recipient'
first_amount = 50
transaction = Transaction(sender_wallet, first_recipient, first_amount)
next_recipient = 'next_recipient'
next_amount = 75
transaction.update(sender_wallet, next_recipient, next_amount)
assert transaction.output[next_recipient] == next_amount
assert transaction.output[sender_wallet.address] ==\
sender_wallet.balance - first_amount - next_amount
assert Wallet.verify(transaction.input['public_key'], transaction.output,
transaction.input['signature'])
to_first_again_amount = 25
transaction.update(sender_wallet, first_recipient, to_first_again_amount)
assert transaction.output[first_recipient] ==\
first_amount + to_first_again_amount
assert transaction.output[sender_wallet.address] ==\
sender_wallet.balance - first_amount - next_amount - to_first_again_amount
assert Wallet.verify(transaction.input['public_key'], transaction.output,
transaction.input['signature'])
def test_transaction():
sender_wallet = Wallet()
recipient = 'recipient'
amount = 50
transaction = Transaction(sender_wallet, recipient, amount)
assert transaction.output[recipient] == amount
assert transaction.output[
sender_wallet.address] == sender_wallet.balance - amount
assert 'timestamp' in transaction.input
assert transaction.input['amount'] == sender_wallet.balance
assert transaction.input['address'] == sender_wallet.address
assert transaction.input['public_key'] == sender_wallet.public_key
assert Wallet.verify(transaction.input['public_key'], transaction.output,
transaction.input['signature'])
def is_valid_transaction(transaction):
"""
Validate a transaction.
Raise an exception for invalid transactions.
"""
if transaction.input == MINING_REWARD_INPUT:
if list(transaction.output.values()) != [MINING_REWARD]:
raise Exception('Invalid mining reward')
return
output_total = sum(transaction.output.values())
if transaction.input['amount'] != output_total:
raise Exception('Invalid transaction output values')
if not Wallet.verify(transaction.input['public_key'],
transaction.output,
transaction.input['signature']):
raise Exception('Invalid signature')
def test_verify_valid_signature():
data = {'foo': 'test_data'}
wallet = Wallet()
signature = wallet.sign(data)
assert Wallet.verify(wallet.public_key, data, signature)