def test_subscribe(self): client = AsyncClient() id = str(uuid4()) client.subscribe(destination='jms.queue.events', ack='client', sub_id=id) req_frame = client.pop_message() self.assertEqual(len(client._subscriptions), 1) self.assertEqual(req_frame.command, Command.SUBSCRIBE) self.assertEqual(req_frame.headers['destination'], 'jms.queue.events') self.assertEqual(req_frame.headers['id'], id) self.assertEqual(req_frame.headers['ack'], 'client')
def test_unsubscribe_with_different_id(self): client = AsyncClient() client.subscribe(destination='jms.queue.events', ack='client-individual', sub_id=str(uuid4())) # ignore subscribe frame client.pop_message() client.unsubscribe( TestSubscription('jms.queue.events', 'ad052acb-a934-4e10-8ec3')) self.assertTrue(len(client._subscriptions) == 1) self.assertFalse(client.has_outgoing_messages)
def test_unsubscribe_with_different_id(self): client = AsyncClient() client.subscribe(destination='jms.queue.events', ack='client-individual', sub_id=str(uuid4())) # ignore subscribe frame client.pop_message() client.unsubscribe(TestSubscription('jms.queue.events', 'ad052acb-a934-4e10-8ec3')) self.assertEqual(len(client._subscriptions), 1) self.assertFalse(client.has_outgoing_messages)
def test_manage_subscription(self): client = AsyncClient() subscription = client.subscribe(destination='jms.queue.events', ack='client', sub_id=str(uuid4())) client.unsubscribe(subscription) self.assertTrue(len(client._subscriptions) == 0)
def test_manage_subscription(self): client = AsyncClient() subscription = client.subscribe(destination='jms.queue.events', ack='client', sub_id=str(uuid4())) client.unsubscribe(subscription) self.assertEqual(len(client._subscriptions), 0)
def test_receive_message(self): client = AsyncClient() id = 'ad052acb-a934-4e10-8ec3-00c7417ef8d1' headers = {Headers.CONTENT_LENGTH: '78', Headers.DESTINATION: 'jms.topic.vdsm_responses', Headers.CONTENT_TYPE: 'application/json', Headers.SUBSCRIPTION: id} body = ('{"jsonrpc": "2.0", "id": "e8a936a6-d886-4cfa-97b9-2d54209053f' 'f", "result": []}') frame = Frame(command=Command.MESSAGE, headers=headers, body=body) def message_handler(sub, frame): client.queue_frame(frame) client.subscribe('', 'auto', id, message_handler) # ignore subscribe frame client.pop_message() client.handle_frame(None, frame) self.assertEqual(frame, client.pop_message())
def test_restore_subcsriptions(self): client = AsyncClient() client.subscribe(destination='jms.queue.events', ack='client', sub_id=str(uuid4())) client.subscribe(destination='jms.queue.events', ack='client', sub_id=str(uuid4())) client.subscribe(destination='jms.queue.events', ack='client', sub_id=str(uuid4())) client.restore_subscriptions() self.assertEqual(len(client._subscriptions), 3)