def connect_and_transfer():
with create_upstream_context_with_handshake() as ctxt:
ctxt.send_msg(Message.MachineReady())
board_available = ctxt.expect_message("BoardAvailable")
board_id = board_available.data.get("BoardId")
ctxt.send_msg(Message.StartTransport(board_id))
ctxt.send_msg(Message.StopTransport(TransferState.COMPLETE, board_id))
transport_finished = ctxt.expect_message("TransportFinished")
assert transport_finished.data.get(
"BoardId") == board_id, "BoardId mismatch"
def test_complete_cycle():
with create_upstream_context() as ctxt:
ctxt.send_msg(Message.ServiceDescription("AcceptanceTestCompleteCycle", 2))
ctxt.expect_message("ServiceDescription")
msg = Message.MachineReady()
ctxt.send_msg(msg)
board_available = ctxt.expect_message("BoardAvailable")
board_id = board_available.data.get("BoardId")
msg = Message.StartTransport(board_id)
ctxt.send_msg(msg)
ctxt.expect_message("TransportFinished")
msg = Message.StopTransport(TransferState.COMPLETE, board_id)
ctxt.send_msg(msg)
ctxt.close();
def connect_board_available_disconnect_n_times():
for _ in range(10):
with create_upstream_context_with_board_available() as ctxt:
ctxt.send_msg(Message.MachineReady())
ctxt.send_msg(Message.RevokeMachineReady())
