def setUp(self):
self.endpoint = 'http://softlayer.com/v1/account/queues/queue_name'
self.queue_name = 'queue_name'
self.message_id = 'message_id'
self.base_href = "%s/messages/%s" % (self.endpoint, self.message_id)
self.auth = Mock()
self.message = Message(self.endpoint, self.queue_name, self.message_id, self.auth)
class TestMessage(unittest.TestCase):
def setUp(self):
self.endpoint = 'http://softlayer.com/v1/account/queues/queue_name'
self.queue_name = 'queue_name'
self.message_id = 'message_id'
self.base_href = "%s/messages/%s" % (self.endpoint, self.message_id)
self.auth = Mock()
self.message = Message(self.endpoint, self.queue_name, self.message_id, self.auth)
@patch('requests.request')
def test_request(self, res):
header_resp = Mock()
res.return_value.error = None
res.return_value.status_code = 200
res.return_value.content = '{"some": "json"}'
res.return_value.headers = header_resp
self.message.headers = {'test': Mock()}
result = self.message.request('DELETE')
self.assertEqual(result.json, {'some': 'json'})
self.assertEqual(result.headers, header_resp)
res.assert_called_with(
'DELETE', self.base_href,
headers=self.message.headers, auth=self.auth)
def test_client_init(self):
self.assertEquals(self.message.url, self.base_href)
self.assertEquals(self.message.queue_name, self.queue_name)
self.assertEquals(self.message.id, self.message_id)
self.assertEquals(self.message.auth, self.auth)
@patch('softlayer_messaging.queue.Message.request')
def test_delete_message(self, res):
result = self.message.delete()
res.assert_called_with('DELETE')
self.assertEqual(result, True)
def test_repr(self):
self.message.url = "URL"
value = repr(self.message)
self.assertEqual(value, "<Message [URL]>")