async def test_interledger_receive_transfer():
t = Transfer()
t.payload = {}
init = MockInitiator([t])
resp = MockResponder()
interledger = Interledger(init, resp)
# interledger.test_interrupt = True
task = asyncio.ensure_future(interledger.receive_transfer())
assert task.done() == False
l = await task
assert init.events == []
assert len(interledger.transfers) == 1
assert l == 1
assert interledger.transfers[0].state == State.READY
async def test_interledger_confirm_transfer_abort():
async def foo():
return {'abort_status': True,
'abort_tx_hash': '0x444'}
i = Interledger(MockInitiator([]), MockResponder())
t = Transfer()
t.payload = {}
t.payload['id'] = str(uuid4().int)
t.result = {}
t.state = State.CONFIRMING
t.confirm_task = asyncio.ensure_future(foo())
i.results_aborting = [t]
task = asyncio.ensure_future(i.confirm_transfer())
assert task.done() == False
await task
res = i.results_abort[0]
assert t.state == State.FINALIZED
assert res['abort_status'] == True
assert res['abort_tx_hash'] == '0x444'
assert len(i.results_abort) == 1
assert len(i.results_commit) == 0
assert len(i.results_aborting) == 0
async def test_interledger_transfer_result():
async def foo():
return 42
i = Interledger(MockInitiator([]), MockResponder())
t = Transfer()
t.state = State.SENT
t.send_task = asyncio.ensure_future(foo())
i.transfers_sent = [t]
task = asyncio.ensure_future(i.transfer_result())
assert task.done() == False
await task
assert len(i.transfers_sent) == 0
assert len(i.transfers_responded) == 1
tr = i.transfers_responded[0]
assert tr.state == State.RESPONDED
assert tr.result == 42
async def test_interledger_send_transfer():
t = Transfer()
t.payload = {}
t.payload['nonce'] = str(uuid4().int)
t.payload['data'] = b"dummy"
i = Interledger(MockInitiator([]), MockResponder())
i.transfers = [t]
task = asyncio.ensure_future(i.send_transfer())
assert task.done() == False
await task
assert len(i.transfers_sent) == 1
tr = i.transfers[0]
assert tr.state == State.SENT
assert asyncio.isfuture(tr.send_task)
await tr.send_task
assert tr.send_task.done() == True
async def test_interledger_run_no_cleanup():
l1, l2, l3 = [], [], []
for i in range(4):
t1, t2, t3 = Transfer(), Transfer(), Transfer()
t1.payload, t2.payload, t3.payload = {}, {}, {}
t1.payload['nonce'], t1.payload['id'], t1.payload['data'] = str(uuid4().int), '1', b"dummy1"
t2.payload['nonce'], t2.payload['id'], t2.payload['data'] = str(uuid4().int), '2', b"dummy2"
t3.payload['nonce'], t3.payload['id'], t3.payload['data'] = str(uuid4().int), '3', b"dummy3"
l1.append(t1)
l2.append(t2)
l3.append(t3)
init = MockInitiator(l1)
i = Interledger(init, MockResponder())
with patch("data_transfer.interledger.Interledger.cleanup") as mock_cleanup:
task = asyncio.ensure_future(i.run())
time = 0.5
# Consume l1
await asyncio.sleep(time) # Simulate interledger running
# New events
init.events = l2
# Consume l2
await asyncio.sleep(time) # Simulate interledger running
# New events
i.responder = MockResponderAbort()
init.events = l3
# Consume l3, but with a responder returning False -> abort
await asyncio.sleep(time) # Simulate interledger running
assert len(i.transfers) == 12
assert len(i.transfers_sent) == 0
assert len(i.transfers_responded) == 0
assert len(i.results_committing) == 0
assert len(i.results_aborting) == 0
assert len(i.results_commit) == 8
assert len(i.results_abort) == 4
i.stop()
await task
async def test_interledger_process_result_abort():
i = Interledger(MockInitiator([]), MockResponder())
t = Transfer()
t.payload = {}
t.payload['id'] = str(uuid4().int)
t.state = State.RESPONDED
t.result = {"status": False}
i.transfers_responded = [t]
task = asyncio.ensure_future(i.process_result())
assert task.done() == False
await task
tr = i.results_aborting[0]
assert tr.state == State.CONFIRMING
assert tr.result['status'] == False
assert len(i.results_commit) == 0
assert len(i.results_abort) == 0
assert len(i.transfers_responded) == 0
def return_transfer_list():
t1 = Transfer()
t2 = Transfer()
t3 = Transfer()
t4 = Transfer()
t5 = Transfer()
# t1 is State.ready by default
t2.state = State.SENT
t3.state = State.RESPONDED
t4.state = State.CONFIRMING
t5.state = State.FINALIZED
return [t1, t2, t3, t4, t5]