def return_transfer_list(): t1 = Transfer() t2 = Transfer() t3 = Transfer() t4 = Transfer() t2.state = State.SENT t3.state = State.RESPONDED t4.state = State.FINALIZED return [t1, t2, t3, t4]
async def test_interledger_run_with_restore_sent(): base = "sofie_asset_transfer" package = base + ".interledger.StateInterledger" # Create a transfer already half processed # for protocol PoW, it should have state (TransferOut | NotHere) t = Transfer() t.data = {} t.data["assetId"] = 123 t.state = State.RESPONDED t.future = asyncio.Future() t.future.set_result({"status": True}) t.result = {"status": True} with patch(package + ".restore_pending_transfers_ready", return_value=[]) as mock_ready: with patch(package + ".restore_pending_transfers_sent", return_value=[t]) as mock_sent: interledger = StateInterledger(StateInitiator(), StateResponder()) init = base + ".state_interfaces.StateInitiator" resp = base + ".state_interfaces.StateResponder" get_future = asyncio.Future() receive_future = asyncio.Future() commit_future = asyncio.Future() get_future.set_result([]) receive_future.set_result({"status": True}) commit_future.set_result(True) with patch(init + ".get_transfers", return_value=get_future) as mock_get_transfers: with patch( resp + ".receive_transfer", return_value=receive_future) as mock_receive_transfer: with patch(init + ".commit_transfer", return_value=commit_future) as mock_commit: with patch(base + ".interledger.Interledger.cleanup" ) as mock_cleanup: # mock cleanup to check the transfer reaches the end time = 1 task = asyncio.ensure_future(interledger.run()) await asyncio.sleep(time) interledger.stop() await task assert len(interledger.transfers_sent) == 1 assert len(interledger.results_commit) == 1 assert interledger.pending == 0
async def test_interledger_process_result_abort(): i = Interledger(MockInitiator([]), None) t = Transfer() t.state = State.RESPONDED t.result = {"status": False} i.transfers_sent = [t] i.pending = 1 task = asyncio.ensure_future(i.process_result()) assert task.done() == False await task tr = i.transfers_sent[0] assert tr.state == State.FINALIZED assert len(i.results_commit) == 0 assert len(i.results_abort) == 1 assert i.pending == 0
async def test_interledger_transfer_result(): async def foo(): return 42 i = Interledger(MockInitiator([]), MockResponder()) t = Transfer() t.state = State.SENT t.future = asyncio.ensure_future(foo()) i.transfers_sent = [t] task = asyncio.ensure_future(i.transfer_result()) assert task.done() == False await task tr = i.transfers_sent[0] assert tr.state == State.RESPONDED assert tr.result == 42