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();