def test_process_start():
status_queue = Queue()
under_test = InThreadStatusTracker(status_queue)
job_id = "some_job_id"
service_id = "some_service_id"
start = RunStartInfo(
job_id=job_id,
filename="file name",
start_time=datetime.now(),
stop_time=datetime.now(),
run_name="run_name",
nexus_structure="nxs structure",
service_id=service_id,
instrument_name="instrument name",
broker="broker",
metadata="{}",
)
assert status_queue.empty()
now = datetime.now()
under_test.process_start(start)
under_test.send_status_if_updated(now)
assert not status_queue.empty()
job_status = status_queue.get()
command_status = status_queue.get()
assert status_queue.empty()
assert command_status.state == CommandState.WAITING_RESPONSE
assert job_status.state == JobState.WAITING
def test_process_msg_start():
start_msg = serialise_start("job id", "file name")
status_queue = Queue()
under_test = InThreadStatusTracker(status_queue)
under_test.process_start = Mock()
under_test.process_message(start_msg)
under_test.process_start.assert_called_once_with(
deserialise_start(start_msg))