def test_instantiate_link_and_connect_to_broker(mockstomp): """Test the Stomp connection routine.""" stomp = StompTransport() mockconn = mockstomp.Connection.return_value assert not stomp.is_connected() stomp.connect() mockstomp.Connection.assert_called_once() mockconn.connect.assert_called_once() assert stomp.is_connected() stomp.connect() mockstomp.Connection.assert_called_once() mockconn.connect.assert_called_once() assert stomp.is_connected() stomp.disconnect() mockstomp.Connection.assert_called_once() mockconn.connect.assert_called_once() mockconn.disconnect.assert_called_once() assert not stomp.is_connected() stomp.disconnect() mockstomp.Connection.assert_called_once() mockconn.connect.assert_called_once() mockconn.disconnect.assert_called_once() assert not stomp.is_connected()
# print(message) # logger.info(f"Submitted {len(objects)} images.") q: "Queue[Union[Dict, List]]" = Queue() # Now wait for the reply messages logger.info(f"Waiting for image-analysis messages on {PIA_UNIQUE_QUEUE}") workflows.recipe.wrap_subscribe( stomp, PIA_UNIQUE_QUEUE, lambda _, _2, message: q.put(message), acknowledgement=False, ) with tqdm(total=len(objects)) as progress: # def receive(rw, header, message): # """Process a hitfinding message""" # q.put(message) # Wait for results to filter in result_count = 0 while result_count < len(objects): message = q.get() result_count += 1 progress.write( f"Got results for image {message['file-number']} = {message['n_spots_total']}" ) progress.update() finally: stomp.disconnect()