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_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_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)
