def test_get_state_change_with_balance_proof():
""" All state changes which contain a balance proof must be found by when
querying the database.
"""
serializer = JSONSerializer
storage = SQLiteStorage(':memory:', serializer)
counter = itertools.count()
lock_expired = ReceiveLockExpired(
balance_proof=make_signed_balance_proof_from_counter(counter),
secrethash=sha3(factories.make_secret(next(counter))),
message_identifier=next(counter),
)
unlock = ReceiveUnlock(
message_identifier=next(counter),
secret=sha3(factories.make_secret(next(counter))),
balance_proof=make_signed_balance_proof_from_counter(counter),
)
transfer_refund = ReceiveTransferRefund(
transfer=make_signed_transfer_from_counter(counter),
routes=list(),
)
transfer_refund_cancel_route = ReceiveTransferRefundCancelRoute(
routes=list(),
transfer=make_signed_transfer_from_counter(counter),
secret=sha3(factories.make_secret(next(counter))),
)
mediator_from_route, mediator_signed_transfer = make_from_route_from_counter(
counter)
action_init_mediator = ActionInitMediator(
routes=list(),
from_route=mediator_from_route,
from_transfer=mediator_signed_transfer,
)
target_from_route, target_signed_transfer = make_from_route_from_counter(
counter)
action_init_target = ActionInitTarget(
route=target_from_route,
transfer=target_signed_transfer,
)
statechanges_balanceproofs = [
(lock_expired, lock_expired.balance_proof),
(unlock, unlock.balance_proof),
(transfer_refund, transfer_refund.transfer.balance_proof),
(transfer_refund_cancel_route,
transfer_refund_cancel_route.transfer.balance_proof),
(action_init_mediator,
action_init_mediator.from_transfer.balance_proof),
(action_init_target, action_init_target.transfer.balance_proof),
]
timestamp = datetime.utcnow().isoformat(timespec='milliseconds')
for state_change, _ in statechanges_balanceproofs:
storage.write_state_change(state_change, timestamp)
# Make sure state changes are returned in the correct order in which they were stored
stored_statechanges = storage.get_statechanges_by_identifier(1, 'latest')
assert isinstance(stored_statechanges[0], ReceiveLockExpired)
assert isinstance(stored_statechanges[1], ReceiveUnlock)
assert isinstance(stored_statechanges[2], ReceiveTransferRefund)
assert isinstance(stored_statechanges[3], ReceiveTransferRefundCancelRoute)
assert isinstance(stored_statechanges[4], ActionInitMediator)
assert isinstance(stored_statechanges[5], ActionInitTarget)
# Make sure state changes are returned in the correct order in which they were stored
stored_statechanges = storage.get_statechanges_by_identifier(1, 2)
assert isinstance(stored_statechanges[0], ReceiveLockExpired)
assert isinstance(stored_statechanges[1], ReceiveUnlock)
for state_change, balance_proof in statechanges_balanceproofs:
state_change_record = get_state_change_with_balance_proof_by_balance_hash(
storage=storage,
chain_id=balance_proof.chain_id,
token_network_identifier=balance_proof.token_network_identifier,
channel_identifier=balance_proof.channel_identifier,
sender=balance_proof.sender,
balance_hash=balance_proof.balance_hash,
)
assert state_change_record.data == state_change
def handle_contract_send_channelsettle(
raiden: "RaidenService", channel_settle_event: ContractSendChannelSettle
):
assert raiden.wal, "The Raiden Service must be initialize to handle events"
canonical_identifier = CanonicalIdentifier(
chain_identifier=raiden.chain.network_id,
token_network_address=channel_settle_event.token_network_identifier,
channel_identifier=channel_settle_event.channel_identifier,
)
triggered_by_block_hash = channel_settle_event.triggered_by_block_hash
payment_channel: PaymentChannel = raiden.chain.payment_channel(
canonical_identifier=canonical_identifier
)
token_network_proxy: TokenNetwork = payment_channel.token_network
if not token_network_proxy.client.can_query_state_for_block(triggered_by_block_hash):
# The only time this can happen is during restarts after a long time
# when the triggered block ends up getting pruned
# In that case it's safe to just use the latest view of the chain to
# query the on-chain participant/channel details
triggered_by_block_hash = token_network_proxy.client.blockhash_from_blocknumber(
"latest"
)
participants_details = token_network_proxy.detail_participants(
participant1=payment_channel.participant1,
participant2=payment_channel.participant2,
block_identifier=triggered_by_block_hash,
channel_identifier=channel_settle_event.channel_identifier,
)
our_details = participants_details.our_details
partner_details = participants_details.partner_details
log_details = {
"chain_id": canonical_identifier.chain_identifier,
"token_network_identifier": canonical_identifier.token_network_address,
"channel_identifier": canonical_identifier.channel_identifier,
"node": pex(raiden.address),
"partner": to_checksum_address(partner_details.address),
"our_deposit": our_details.deposit,
"our_withdrawn": our_details.withdrawn,
"our_is_closer": our_details.is_closer,
"our_balance_hash": to_hex(our_details.balance_hash),
"our_nonce": our_details.nonce,
"our_locksroot": to_hex(our_details.locksroot),
"our_locked_amount": our_details.locked_amount,
"partner_deposit": partner_details.deposit,
"partner_withdrawn": partner_details.withdrawn,
"partner_is_closer": partner_details.is_closer,
"partner_balance_hash": to_hex(partner_details.balance_hash),
"partner_nonce": partner_details.nonce,
"partner_locksroot": to_hex(partner_details.locksroot),
"partner_locked_amount": partner_details.locked_amount,
}
if our_details.balance_hash != EMPTY_HASH:
event_record = get_event_with_balance_proof_by_balance_hash(
storage=raiden.wal.storage,
canonical_identifier=canonical_identifier,
balance_hash=our_details.balance_hash,
)
if event_record.data is None:
log.critical("our balance proof not found", **log_details)
raise RaidenUnrecoverableError(
"Our balance proof could not be found in the database"
)
our_balance_proof = event_record.data.balance_proof
our_transferred_amount = our_balance_proof.transferred_amount
our_locked_amount = our_balance_proof.locked_amount
our_locksroot = our_balance_proof.locksroot
else:
our_transferred_amount = 0
our_locked_amount = 0
our_locksroot = EMPTY_HASH
if partner_details.balance_hash != EMPTY_HASH:
state_change_record = get_state_change_with_balance_proof_by_balance_hash(
storage=raiden.wal.storage,
canonical_identifier=canonical_identifier,
balance_hash=partner_details.balance_hash,
sender=participants_details.partner_details.address,
)
if state_change_record.data is None:
log.critical("partner balance proof not found", **log_details)
raise RaidenUnrecoverableError(
"Partner balance proof could not be found in the database"
)
partner_balance_proof = state_change_record.data.balance_proof
partner_transferred_amount = partner_balance_proof.transferred_amount
partner_locked_amount = partner_balance_proof.locked_amount
partner_locksroot = partner_balance_proof.locksroot
else:
partner_transferred_amount = 0
partner_locked_amount = 0
partner_locksroot = EMPTY_HASH
payment_channel.settle(
transferred_amount=our_transferred_amount,
locked_amount=our_locked_amount,
locksroot=our_locksroot,
partner_transferred_amount=partner_transferred_amount,
partner_locked_amount=partner_locked_amount,
partner_locksroot=partner_locksroot,
block_identifier=triggered_by_block_hash,
)
def handle_contract_send_channelsettle(
self,
raiden: RaidenService,
channel_settle_event: ContractSendChannelSettle,
):
chain_id = raiden.chain.network_id
token_network_identifier = channel_settle_event.token_network_identifier
channel_identifier = channel_settle_event.channel_identifier
payment_channel: PaymentChannel = raiden.chain.payment_channel(
token_network_address=channel_settle_event.token_network_identifier,
channel_id=channel_settle_event.channel_identifier,
)
token_network_proxy: TokenNetwork = payment_channel.token_network
participants_details = token_network_proxy.detail_participants(
participant1=payment_channel.participant1,
participant2=payment_channel.participant2,
channel_identifier=channel_settle_event.channel_identifier,
)
our_details = participants_details.our_details
partner_details = participants_details.partner_details
log_details = {
'chain_id': chain_id,
'token_network_identifier': token_network_identifier,
'channel_identifier': channel_identifier,
'node': pex(raiden.address),
'partner': to_checksum_address(partner_details.address),
'our_deposit': our_details.deposit,
'our_withdrawn': our_details.withdrawn,
'our_is_closer': our_details.is_closer,
'our_balance_hash': to_hex(our_details.balance_hash),
'our_nonce': our_details.nonce,
'our_locksroot': to_hex(our_details.locksroot),
'our_locked_amount': our_details.locked_amount,
'partner_deposit': partner_details.deposit,
'partner_withdrawn': partner_details.withdrawn,
'partner_is_closer': partner_details.is_closer,
'partner_balance_hash': to_hex(partner_details.balance_hash),
'partner_nonce': partner_details.nonce,
'partner_locksroot': to_hex(partner_details.locksroot),
'partner_locked_amount': partner_details.locked_amount,
}
if our_details.balance_hash != EMPTY_HASH:
event_record = get_event_with_balance_proof_by_balance_hash(
storage=raiden.wal.storage,
chain_id=chain_id,
token_network_identifier=token_network_identifier,
channel_identifier=channel_identifier,
balance_hash=our_details.balance_hash,
)
if event_record.data is None:
log.critical(
'our balance proof not found',
**log_details,
)
raise RaidenUnrecoverableError(
'Our balance proof could not be found in the database',
)
our_balance_proof = event_record.data.balance_proof
our_transferred_amount = our_balance_proof.transferred_amount
our_locked_amount = our_balance_proof.locked_amount
our_locksroot = our_balance_proof.locksroot
else:
our_transferred_amount = 0
our_locked_amount = 0
our_locksroot = EMPTY_HASH
if partner_details.balance_hash != EMPTY_HASH:
state_change_record = get_state_change_with_balance_proof_by_balance_hash(
storage=raiden.wal.storage,
chain_id=chain_id,
token_network_identifier=token_network_identifier,
channel_identifier=channel_identifier,
balance_hash=partner_details.balance_hash,
sender=participants_details.partner_details.address,
)
if state_change_record.data is None:
log.critical(
'partner balance proof not found',
**log_details,
)
raise RaidenUnrecoverableError(
'Partner balance proof could not be found in the database',
)
partner_balance_proof = state_change_record.data.balance_proof
partner_transferred_amount = partner_balance_proof.transferred_amount
partner_locked_amount = partner_balance_proof.locked_amount
partner_locksroot = partner_balance_proof.locksroot
else:
partner_transferred_amount = 0
partner_locked_amount = 0
partner_locksroot = EMPTY_HASH
payment_channel.settle(
our_transferred_amount,
our_locked_amount,
our_locksroot,
partner_transferred_amount,
partner_locked_amount,
partner_locksroot,
)
def handle_contract_send_channelsettle(
self,
raiden: RaidenService,
channel_settle_event: ContractSendChannelSettle,
):
chain_id = raiden.chain.network_id
token_network_identifier = channel_settle_event.token_network_identifier
channel_identifier = channel_settle_event.channel_identifier
payment_channel: PaymentChannel = raiden.chain.payment_channel(
token_network_address=channel_settle_event.
token_network_identifier,
channel_id=channel_settle_event.channel_identifier,
)
token_network_proxy: TokenNetwork = payment_channel.token_network
participants_details = token_network_proxy.detail_participants(
participant1=payment_channel.participant1,
participant2=payment_channel.participant2,
block_identifier='latest',
channel_identifier=channel_settle_event.channel_identifier,
)
our_details = participants_details.our_details
partner_details = participants_details.partner_details
log_details = {
'chain_id': chain_id,
'token_network_identifier': token_network_identifier,
'channel_identifier': channel_identifier,
'node': pex(raiden.address),
'partner': to_checksum_address(partner_details.address),
'our_deposit': our_details.deposit,
'our_withdrawn': our_details.withdrawn,
'our_is_closer': our_details.is_closer,
'our_balance_hash': to_hex(our_details.balance_hash),
'our_nonce': our_details.nonce,
'our_locksroot': to_hex(our_details.locksroot),
'our_locked_amount': our_details.locked_amount,
'partner_deposit': partner_details.deposit,
'partner_withdrawn': partner_details.withdrawn,
'partner_is_closer': partner_details.is_closer,
'partner_balance_hash': to_hex(partner_details.balance_hash),
'partner_nonce': partner_details.nonce,
'partner_locksroot': to_hex(partner_details.locksroot),
'partner_locked_amount': partner_details.locked_amount,
}
if our_details.balance_hash != EMPTY_HASH:
event_record = get_event_with_balance_proof_by_balance_hash(
storage=raiden.wal.storage,
chain_id=chain_id,
token_network_identifier=token_network_identifier,
channel_identifier=channel_identifier,
balance_hash=our_details.balance_hash,
)
if event_record.data is None:
log.critical(
'our balance proof not found',
**log_details,
)
raise RaidenUnrecoverableError(
'Our balance proof could not be found in the database', )
our_balance_proof = event_record.data.balance_proof
our_transferred_amount = our_balance_proof.transferred_amount
our_locked_amount = our_balance_proof.locked_amount
our_locksroot = our_balance_proof.locksroot
else:
our_transferred_amount = 0
our_locked_amount = 0
our_locksroot = EMPTY_HASH
if partner_details.balance_hash != EMPTY_HASH:
state_change_record = get_state_change_with_balance_proof_by_balance_hash(
storage=raiden.wal.storage,
chain_id=chain_id,
token_network_identifier=token_network_identifier,
channel_identifier=channel_identifier,
balance_hash=partner_details.balance_hash,
sender=participants_details.partner_details.address,
)
if state_change_record.data is None:
log.critical(
'partner balance proof not found',
**log_details,
)
raise RaidenUnrecoverableError(
'Partner balance proof could not be found in the database',
)
partner_balance_proof = state_change_record.data.balance_proof
partner_transferred_amount = partner_balance_proof.transferred_amount
partner_locked_amount = partner_balance_proof.locked_amount
partner_locksroot = partner_balance_proof.locksroot
else:
partner_transferred_amount = 0
partner_locked_amount = 0
partner_locksroot = EMPTY_HASH
# LEFTODO: Supply a proper block id
payment_channel.settle(
transferred_amount=our_transferred_amount,
locked_amount=our_locked_amount,
locksroot=our_locksroot,
partner_transferred_amount=partner_transferred_amount,
partner_locked_amount=partner_locked_amount,
partner_locksroot=partner_locksroot,
block_identifier='latest',
)
def test_get_state_change_with_balance_proof():
""" All state changes which contain a balance proof must be found by when
querying the database.
"""
serializer = JSONSerializer
storage = SQLiteStorage(':memory:', serializer)
counter = itertools.count()
lock_expired = ReceiveLockExpired(
balance_proof=make_signed_balance_proof_from_counter(counter),
secrethash=sha3(factories.make_secret(next(counter))),
message_identifier=next(counter),
)
unlock = ReceiveUnlock(
message_identifier=next(counter),
secret=sha3(factories.make_secret(next(counter))),
balance_proof=make_signed_balance_proof_from_counter(counter),
)
transfer_refund = ReceiveTransferRefund(
transfer=make_signed_transfer_from_counter(counter),
routes=list(),
)
transfer_refund_cancel_route = ReceiveTransferRefundCancelRoute(
routes=list(),
transfer=make_signed_transfer_from_counter(counter),
secret=sha3(factories.make_secret(next(counter))),
)
mediator_from_route, mediator_signed_transfer = make_from_route_from_counter(counter)
action_init_mediator = ActionInitMediator(
routes=list(),
from_route=mediator_from_route,
from_transfer=mediator_signed_transfer,
)
target_from_route, target_signed_transfer = make_from_route_from_counter(counter)
action_init_target = ActionInitTarget(
route=target_from_route,
transfer=target_signed_transfer,
)
statechanges_balanceproofs = [
(lock_expired, lock_expired.balance_proof),
(unlock, unlock.balance_proof),
(transfer_refund, transfer_refund.transfer.balance_proof),
(transfer_refund_cancel_route, transfer_refund_cancel_route.transfer.balance_proof),
(action_init_mediator, action_init_mediator.from_transfer.balance_proof),
(action_init_target, action_init_target.transfer.balance_proof),
]
timestamp = datetime.utcnow().isoformat(timespec='milliseconds')
for state_change, _ in statechanges_balanceproofs:
storage.write_state_change(state_change, timestamp)
# Make sure state changes are returned in the correct order in which they were stored
stored_statechanges = storage.get_statechanges_by_identifier(1, 'latest')
assert isinstance(stored_statechanges[0], ReceiveLockExpired)
assert isinstance(stored_statechanges[1], ReceiveUnlock)
assert isinstance(stored_statechanges[2], ReceiveTransferRefund)
assert isinstance(stored_statechanges[3], ReceiveTransferRefundCancelRoute)
assert isinstance(stored_statechanges[4], ActionInitMediator)
assert isinstance(stored_statechanges[5], ActionInitTarget)
# Make sure state changes are returned in the correct order in which they were stored
stored_statechanges = storage.get_statechanges_by_identifier(1, 2)
assert isinstance(stored_statechanges[0], ReceiveLockExpired)
assert isinstance(stored_statechanges[1], ReceiveUnlock)
for state_change, balance_proof in statechanges_balanceproofs:
state_change_record = get_state_change_with_balance_proof_by_balance_hash(
storage=storage,
chain_id=balance_proof.chain_id,
token_network_identifier=balance_proof.token_network_identifier,
channel_identifier=balance_proof.channel_identifier,
sender=balance_proof.sender,
balance_hash=balance_proof.balance_hash,
)
assert state_change_record.data == state_change