def test_basic(self): sot = message.Message() self.assertEqual('messages', sot.resources_key) self.assertEqual('/queues/%(queue_name)s/messages', sot.base_path) self.assertEqual('messaging', sot.service.service_type) self.assertTrue(sot.allow_create) self.assertFalse(sot.allow_retrieve) self.assertFalse(sot.allow_update) self.assertFalse(sot.allow_delete) self.assertFalse(sot.allow_list)
def test_delete(self): sess = mock.Mock() sess.delete = mock.Mock() sess.delete.return_value = mock.Mock() sot = message.Message() sot.delete_by_id( sess, message.Message.new(client=CLIENT, queue=QUEUE, **FAKE_HREF)) url = '/queues/%s/messages/1234' % QUEUE sess.delete.assert_called_with( url, service=sot.service, accept=None, headers={'Client-ID': CLIENT})
def test_create(self): sess = mock.Mock() obj = mock.Mock() obj.json = mock.Mock(return_value={'resources': {'k': 'v'}}) sess.post = mock.Mock(return_value=obj) sot = message.Message() msg = message.Message.new(client_id=CLIENT, queue_name=QUEUE, **FAKE) sot.create_messages(sess, [msg]) url = '/queues/%s/messages' % QUEUE sess.post.assert_called_with(url, headers={'Client-ID': CLIENT}, data=mock.ANY) args, kwargs = sess.post.call_args self.assertIn("data", kwargs) self.assertDictEqual(json.loads(kwargs["data"])[0], FAKE)
def test_create(self): sess = mock.Mock() sess.post = mock.Mock() sess.post.return_value = mock.MagicMock() sot = message.Message() sot.create_messages( sess, [message.Message.new(client=CLIENT, queue=QUEUE, **FAKE)]) url = '/queues/%s/messages' % QUEUE sess.post.assert_called_with( url, service=sot.service, headers={'Client-ID': CLIENT}, data=mock.ANY) args, kwargs = sess.post.call_args self.assertIn("data", kwargs) self.assertDictEqual(json.loads(kwargs["data"])[0], FAKE)
def test_delete(self): sess = mock.Mock() sess.delete = mock.Mock() sess.delete.return_value = mock.Mock() sot = message.Message() sot.delete_by_id( sess, message.Message.new(client_id=CLIENT, queue_name=QUEUE, **FAKE_HREF)) url = '/queues/%s/messages/1234' % QUEUE sess.delete.assert_called_with(url, endpoint_filter=sot.service, headers={ 'Client-ID': CLIENT, 'Accept': '' })
def test_make_it(self): sot = message.Message(FAKE) self.assertEqual(FAKE['ttl'], sot.ttl) self.assertEqual(FAKE['body'], sot.body)