def test_get_client_id_project_id_exist(self):
sess = mock.Mock()
resp = mock.Mock()
sess.get.return_value = resp
sot = message.Message(**FAKE1)
sot._translate_response = mock.Mock()
res = sot.get(sess)
url = 'queues/%(queue)s/messages/%(message)s' % {
'queue': FAKE2['queue_name'],
'message': FAKE2['id']
}
sot = message.Message(**FAKE2)
sot._translate_response = mock.Mock()
res = sot.get(sess)
headers = {
'Client-ID': 'OLD_CLIENT_ID',
'X-PROJECT-ID': 'OLD_PROJECT_ID'
}
sess.get.assert_called_with(url,
endpoint_filter=sot.service,
headers=headers)
sot._translate_response.assert_called_once_with(resp)
self.assertEqual(sot, res)
def test_post_client_id_project_id_exist(self):
sess = mock.Mock()
resp = mock.Mock()
sess.post.return_value = resp
resources = [
'/v2/queues/queue1/messages/578ee000508f153f256f717d'
'/v2/queues/queue1/messages/579edd6c368cb61de9a7e233'
]
resp.json.return_value = {'resources': resources}
messages = [{
'body': {
'key': 'value1'
},
'ttl': 3600
}, {
'body': {
'key': 'value2'
},
'ttl': 1800
}]
sot = message.Message(**FAKE2)
res = sot.post(sess, messages)
url = '/queues/%(queue)s/messages' % {'queue': FAKE2['queue_name']}
headers = {
'Client-ID': 'OLD_CLIENT_ID',
'X-PROJECT-ID': 'OLD_PROJECT_ID'
}
sess.post.assert_called_once_with(url,
endpoint_filter=sot.service,
headers=headers,
json={'messages': messages})
resp.json.assert_called_once_with()
self.assertEqual(resources, res)
def test_basic(self):
sot = message.Message()
self.assertEqual('messages', sot.resources_key)
self.assertEqual('/queues/%(queue_name)s/messages', sot.base_path)
self.assertTrue(sot.allow_create)
self.assertTrue(sot.allow_fetch)
self.assertTrue(sot.allow_delete)
self.assertTrue(sot.allow_list)
def test_message_post(self, mock_get_resource):
message_obj = message.Message(queue_name="test_queue")
mock_get_resource.return_value = message_obj
self._verify("openstack.message.v2.message.Message.post",
self.proxy.post_message,
method_args=["test_queue", ["msg1", "msg2"]],
expected_args=[["msg1", "msg2"]])
mock_get_resource.assert_called_once_with(message.Message, None,
queue_name="test_queue")
def test_delete_client_id_project_id_exist(self):
sess = mock.Mock()
resp = mock.Mock()
sess.delete.return_value = resp
sot = message.Message(**FAKE2)
sot.claim_id = None
sot._translate_response = mock.Mock()
sot.delete(sess)
url = 'queues/%(queue)s/messages/%(message)s' % {
'queue': FAKE2['queue_name'], 'message': FAKE2['id']}
headers = {'Client-ID': 'OLD_CLIENT_ID',
'X-PROJECT-ID': 'OLD_PROJECT_ID'}
sess.delete.assert_called_with(url, endpoint_filter=sot.service,
headers=headers)
sot._translate_response.assert_called_once_with(resp, has_body=False)
def test_delete_unclaimed(self, mock_uuid):
sess = mock.Mock()
resp = mock.Mock()
sess.delete.return_value = resp
sess.get_project_id.return_value = 'NEW_PROJECT_ID'
mock_uuid.return_value = 'NEW_CLIENT_ID'
sot = message.Message(**FAKE1)
sot.claim_id = None
sot._translate_response = mock.Mock()
sot.delete(sess)
url = 'queues/%(queue)s/messages/%(message)s' % {
'queue': FAKE1['queue_name'], 'message': FAKE1['id']}
headers = {'Client-ID': 'NEW_CLIENT_ID',
'X-PROJECT-ID': 'NEW_PROJECT_ID'}
sess.delete.assert_called_with(url,
headers=headers)
sess.get_project_id.assert_called_once_with()
sot._translate_response.assert_called_once_with(resp, has_body=False)
def test_get(self, mock_uuid):
sess = mock.Mock()
resp = mock.Mock()
sess.get.return_value = resp
sess.get_project_id.return_value = 'NEW_PROJECT_ID'
mock_uuid.return_value = 'NEW_CLIENT_ID'
sot = message.Message(**FAKE1)
sot._translate_response = mock.Mock()
res = sot.get(sess)
url = 'queues/%(queue)s/messages/%(message)s' % {
'queue': FAKE1['queue_name'], 'message': FAKE1['id']}
headers = {'Client-ID': 'NEW_CLIENT_ID',
'X-PROJECT-ID': 'NEW_PROJECT_ID'}
sess.get.assert_called_with(url,
headers=headers)
sess.get_project_id.assert_called_once_with()
sot._translate_response.assert_called_once_with(resp)
self.assertEqual(sot, res)
