def test_echo_node_response(token_addresses, raiden_chain, network_wait): app0, app1, app2, echo_app = raiden_chain address_to_app = {app.raiden.address: app for app in raiden_chain} token_address = token_addresses[0] echo_api = RaidenAPI(echo_app.raiden) echo_node = EchoNode(echo_api, token_address) echo_node.ready.wait(timeout=30) assert echo_node.ready.is_set() expected = list() # Create some transfers for num, app in enumerate([app0, app1, app2]): amount = 1 + num transfer_event = RaidenAPI(app.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10 ** (num + 1), ) transfer_event.wait(timeout=20) expected.append(amount) while echo_node.num_handled_transfers < len(expected): gevent.sleep(.5) # Check that all transfers were handled correctly def test_events(handled_transfer): app = address_to_app[handled_transfer.initiator] events = RaidenAPI(app.raiden).get_raiden_events_payment_history( token_address=token_address, ) received = { event.identifier: event for event in events if type(event) == EventPaymentReceivedSuccess } if len(received) != 1: return transfer = received.popitem()[1] is_not_valid = ( transfer.initiator != echo_app.raiden.address or transfer.identifier != handled_transfer.identifier + transfer.amount ) if is_not_valid: return return transfer for handled_transfer in echo_node.seen_transfers: assert wait_until(lambda: test_events(handled_transfer), network_wait) echo_node.stop()
class EchoNodeRunner(NodeRunner): def __init__(self, options: Dict[str, Any], ctx, token_address: typing.TokenAddress): super().__init__(options, ctx) self._token_address = token_address self._echo_node = None @property def welcome_string(self): return '{} [ECHO NODE]'.format(super().welcome_string) def _startup_hook(self): self._echo_node = EchoNode(self._raiden_api, self._token_address) def _shutdown_hook(self): self._echo_node.stop()
def test_echo_node_response(token_addresses, raiden_chain): app0, app1, app2, echo_app = raiden_chain address_to_app = {app.raiden.address: app for app in raiden_chain} token_address = token_addresses[0] echo_api = RaidenAPI(echo_app.raiden) echo_node = EchoNode(echo_api, token_address) echo_node.ready.wait(timeout=30) assert echo_node.ready.is_set() expected = list() # Create some transfers for num, app in enumerate([app0, app1, app2]): amount = 1 + num transfer_event = RaidenAPI(app.raiden).transfer_async( token_address, amount, echo_app.raiden.address, 10 ** (num + 1) ) transfer_event.wait(timeout=20) expected.append(amount) while echo_node.num_handled_transfers < len(expected): gevent.sleep(.5) # Check that all transfers were handled correctly for handled_transfer in echo_node.seen_transfers: app = address_to_app[handled_transfer['initiator']] events = get_channel_events_for_token(app, token_address, 0) received = {} for event in events: if event['_event_type'] == 'EventTransferReceivedSuccess': received[repr(event)] = event assert len(received) == 1 transfer = list(received.values())[0] assert transfer['initiator'] == echo_app.raiden.address assert transfer['identifier'] == ( handled_transfer['identifier'] + transfer['amount'] ) echo_node.stop()
def test_echo_node_lottery(token_addresses, raiden_chain, network_wait): app0, app1, app2, app3, echo_app, app4, app5, app6 = raiden_chain address_to_app = {app.raiden.address: app for app in raiden_chain} token_address = token_addresses[0] echo_api = RaidenAPI(echo_app.raiden) echo_node = EchoNode(echo_api, token_address) echo_node.ready.wait(timeout=30) assert echo_node.ready.is_set() expected = list() # Let 6 participants enter the pool amount = 7 for num, app in enumerate([app0, app1, app2, app3, app4, app5]): transfer_event = RaidenAPI(app.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10 ** (num + 1), ) transfer_event.wait(timeout=20) expected.append(amount) # test duplicated identifier + amount is ignored transfer_event = RaidenAPI(app5.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, # same amount as before echo_app.raiden.address, 10 ** 6, # app5 used this identifier before ).wait(timeout=20) # test pool size querying pool_query_identifier = 77 # unused identifier different from previous one transfer_event = RaidenAPI(app5.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, pool_query_identifier, ).wait(timeout=20) expected.append(amount) # fill the pool transfer_event = RaidenAPI(app6.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10 ** 7, ).wait(timeout=20) expected.append(amount) while echo_node.num_handled_transfers < len(expected): gevent.sleep(.5) def get_echoed_transfer(sent_transfer): """For a given transfer sent to echo node, get the corresponding echoed transfer""" app = address_to_app[sent_transfer.initiator] events = RaidenAPI(app.raiden).get_raiden_events_payment_history( token_address=token_address, ) def is_valid(event): return ( type(event) == EventPaymentReceivedSuccess and event.initiator == echo_app.raiden.address and event.identifier == sent_transfer.identifier + event.amount ) received = { event.identifier: event for event in events if is_valid(event) } if len(received) != 1: return return received.popitem()[1] def received_is_of_size(size): """Return transfers received from echo_node when there's size transfers""" received = {} # Check that payout was generated and pool_size_query answered for handled_transfer in echo_node.seen_transfers: event = get_echoed_transfer(handled_transfer) if not event: continue received[event.identifier] = event if len(received) == size: return received # wait for the expected echoed transfers to be handled received = wait_until(lambda: received_is_of_size(2), 2 * network_wait) assert received received = sorted(received.values(), key=lambda transfer: transfer.amount) pool_query = received[0] assert pool_query.amount == 6 assert pool_query.identifier == pool_query_identifier + 6 winning_transfer = received[1] assert winning_transfer.initiator == echo_app.raiden.address assert winning_transfer.amount == 49 assert (winning_transfer.identifier - 49) % 10 == 0 echo_node.stop()
def _startup_hook(self): self._echo_node = EchoNode(self._raiden_api, self._token_address)
def test_echo_node_lottery(token_addresses, raiden_chain): app0, app1, app2, app3, echo_app, app4, app5, app6 = raiden_chain address_to_app = {app.raiden.address: app for app in raiden_chain} token_address = token_addresses[0] echo_api = RaidenAPI(echo_app.raiden) echo_node = EchoNode(echo_api, token_address) echo_node.ready.wait(timeout=30) assert echo_node.ready.is_set() expected = list() # Let 6 participants enter the pool amount = 7 for num, app in enumerate([app0, app1, app2, app3, app4, app5]): transfer_event = RaidenAPI(app.raiden).transfer_async( token_address, amount, echo_app.raiden.address, 10 ** (num + 1) ) transfer_event.wait(timeout=20) expected.append(amount) # test duplicated identifier + amount is ignored transfer_event = RaidenAPI(app5.raiden).transfer_async( token_address, amount, # same amount as before echo_app.raiden.address, 10 ** 6 # app5 used this identifier before ).wait(timeout=20) # test pool size querying pool_query_identifier = 77 # unused identifier different from previous one transfer_event = RaidenAPI(app5.raiden).transfer_async( token_address, amount, echo_app.raiden.address, pool_query_identifier ).wait(timeout=20) expected.append(amount) # fill the pool transfer_event = RaidenAPI(app6.raiden).transfer_async( token_address, amount, echo_app.raiden.address, 10 ** 7 ).wait(timeout=20) expected.append(amount) while echo_node.num_handled_transfers < len(expected): gevent.sleep(.5) received = {} # Check that payout was generated and pool_size_query answered for handled_transfer in echo_node.seen_transfers: app = address_to_app[handled_transfer['initiator']] events = get_channel_events_for_token(app, token_address, 0) for event in events: if event['_event_type'] == 'EventTransferReceivedSuccess': received[repr(event)] = event assert len(received) == 2 received = sorted(received.values(), key=lambda transfer: transfer['amount']) pool_query = received[0] assert pool_query['amount'] == 6 assert pool_query['identifier'] == pool_query_identifier + 6 winning_transfer = received[1] assert winning_transfer['initiator'] == echo_app.raiden.address assert winning_transfer['amount'] == 49 assert (winning_transfer['identifier'] - 49) % 10 == 0 echo_node.stop()
def test_echo_node_lottery(token_addresses, raiden_chain): app0, app1, app2, app3, echo_app, app4, app5, app6 = raiden_chain address_to_app = {app.raiden.address: app for app in raiden_chain} token_address = token_addresses[0] echo_api = RaidenAPI(echo_app.raiden) echo_node = EchoNode(echo_api, token_address) echo_node.ready.wait(timeout=30) assert echo_node.ready.is_set() expected = list() # Let 6 participants enter the pool amount = 7 for num, app in enumerate([app0, app1, app2, app3, app4, app5]): transfer_event = RaidenAPI(app.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10**(num + 1), ) transfer_event.wait(timeout=20) expected.append(amount) # test duplicated identifier + amount is ignored transfer_event = RaidenAPI(app5.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, # same amount as before echo_app.raiden.address, 10**6, # app5 used this identifier before ).wait(timeout=20) # test pool size querying pool_query_identifier = 77 # unused identifier different from previous one transfer_event = RaidenAPI(app5.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, pool_query_identifier, ).wait(timeout=20) expected.append(amount) # fill the pool transfer_event = RaidenAPI(app6.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10**7, ).wait(timeout=20) expected.append(amount) while echo_node.num_handled_transfers < len(expected): gevent.sleep(.5) received = {} # Check that payout was generated and pool_size_query answered for handled_transfer in echo_node.seen_transfers: app = address_to_app[handled_transfer['initiator']] events = get_channel_events_for_token( app.raiden.default_registry.address, app, token_address, 0, ) for event in events: if event['event'] == 'EventTransferReceivedSuccess': received[repr(event)] = event assert len(received) == 2 received = sorted(received.values(), key=lambda transfer: transfer['amount']) pool_query = received[0] assert pool_query['amount'] == 6 assert pool_query['identifier'] == pool_query_identifier + 6 winning_transfer = received[1] assert winning_transfer['initiator'] == echo_app.raiden.address assert winning_transfer['amount'] == 49 assert (winning_transfer['identifier'] - 49) % 10 == 0 echo_node.stop()
def run_test_echo_node_lottery(token_addresses, raiden_chain, network_wait): app0, app1, app2, app3, echo_app, app4, app5, app6 = raiden_chain address_to_app = {app.raiden.address: app for app in raiden_chain} token_address = token_addresses[0] echo_api = RaidenAPI(echo_app.raiden) echo_node = EchoNode(echo_api, token_address) echo_node.ready.wait(timeout=30) assert echo_node.ready.is_set() expected = list() # Let 6 participants enter the pool amount = 7 for num, app in enumerate([app0, app1, app2, app3, app4, app5]): payment_status = RaidenAPI(app.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10**(num + 1), ) payment_status.payment_done.wait(timeout=20) expected.append(amount) # test duplicated identifier + amount is ignored payment_status = ( RaidenAPI(app5.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, # same amount as before echo_app.raiden.address, 10**6, # app5 used this identifier before ).payment_done.wait(timeout=20)) # test pool size querying pool_query_identifier = 77 # unused identifier different from previous one payment_status = (RaidenAPI(app5.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, pool_query_identifier, ).payment_done.wait(timeout=20)) expected.append(amount) # fill the pool payment_status = (RaidenAPI(app6.raiden).transfer_async( app.raiden.default_registry.address, token_address, amount, echo_app.raiden.address, 10**7, ).payment_done.wait(timeout=20)) expected.append(amount) while echo_node.num_handled_transfers < len(expected): gevent.sleep(0.5) def get_echoed_transfer(sent_transfer): """For a given transfer sent to echo node, get the corresponding echoed transfer""" app = address_to_app[sent_transfer.initiator] events = RaidenAPI(app.raiden).get_raiden_events_payment_history( token_address=token_address) def is_valid(event): return (type(event) == EventPaymentReceivedSuccess and event.initiator == echo_app.raiden.address and event.identifier == sent_transfer.identifier + event.amount) received = { event.identifier: event for event in events if is_valid(event) } if len(received) != 1: return None return received.popitem()[1] def received_is_of_size(size): """Return transfers received from echo_node when there's size transfers""" received = {} # Check that payout was generated and pool_size_query answered for handled_transfer in echo_node.seen_transfers: event = get_echoed_transfer(handled_transfer) if not event: continue received[event.identifier] = event if len(received) == size: return received return None # wait for the expected echoed transfers to be handled received = wait_until(lambda: received_is_of_size(2), 2 * network_wait) assert received received = sorted(received.values(), key=lambda transfer: transfer.amount) pool_query = received[0] assert pool_query.amount == 6 assert pool_query.identifier == pool_query_identifier + 6 winning_transfer = received[1] assert winning_transfer.initiator == echo_app.raiden.address assert winning_transfer.amount == 49 assert (winning_transfer.identifier - 49) % 10 == 0 echo_node.stop()