def diff_account_db(expected_state: AccountState,
account_db: BaseAccountDB) -> AccountDiff:
for account, account_data in sorted(expected_state.items()):
expected_balance = account_data['balance']
expected_nonce = account_data['nonce']
expected_code = account_data['code']
actual_nonce = account_db.get_nonce(account)
actual_code = account_db.get_code(account)
actual_balance = account_db.get_balance(account)
if actual_nonce != expected_nonce:
yield (account, 'nonce', actual_nonce, expected_nonce)
if actual_code != expected_code:
yield (account, 'code', actual_code, expected_code)
if actual_balance != expected_balance:
yield (account, 'balance', actual_balance, expected_balance)
for slot, expected_storage_value in sorted(
account_data['storage'].items()):
actual_storage_value = account_db.get_storage(account, slot)
if actual_storage_value != expected_storage_value:
yield (
account,
'storage[{0}]'.format(slot),
actual_storage_value,
expected_storage_value,
)
def setup_account_db(desired_state: AccountState, account_db: BaseAccountDB) -> None:
for account, account_data in desired_state.items():
for slot, value in account_data['storage'].items():
account_db.set_storage(account, slot, value)
nonce = account_data['nonce']
code = account_data['code']
balance = account_data['balance']
account_db.set_nonce(account, nonce)
account_db.set_code(account, code)
account_db.set_balance(account, balance)
account_db.persist()
def validate_frontier_transaction(account_db: BaseAccountDB,
transaction: BaseOrSpoofTransaction) -> None:
gas_cost = transaction.gas * transaction.gas_price
sender_balance = account_db.get_balance(transaction.sender)
if sender_balance < gas_cost:
raise ValidationError(
"Sender account balance cannot afford txn gas: `{0}`".format(
transaction.sender))
total_cost = transaction.value + gas_cost
if sender_balance < total_cost:
raise ValidationError("Sender account balance cannot afford txn")
if account_db.get_nonce(transaction.sender) != transaction.nonce:
raise ValidationError("Invalid transaction nonce")
def apply_state_dict(account_db: BaseAccountDB,
state_dict: AccountState) -> BaseAccountDB:
for account, account_data in state_dict.items():
account_db.set_balance(account, account_data["balance"])
account_db.set_nonce(account, account_data["nonce"])
account_db.set_code(account, account_data["code"])
for slot, value in account_data["storage"].items():
account_db.set_storage(account, slot, value)
return account_db