def setUp(self):
self.ams = ArgoMessagingService(endpoint="localhost",
token="s3cr3t",
project="TEST")
self.msg = AmsMessage(attributes={'foo': 'bar'}, data='baz')
self.submocks = SubMocks()
self.topicmocks = TopicMocks()
def testPublish(self):
# Mock response for POST publish to topic
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/topics/topic1:publish",
method="POST")
def publish_mock(url, request):
assert url.path == "/v1/projects/TEST/topics/topic1:publish"
# Check request produced by ams client
req_body = json.loads(request.body)
assert (req_body["messages"][0]["data"] == "Zm9vMQ==")
assert (req_body["messages"][0]["attributes"]["bar1"] == "baz1")
return '{"msgIds":["1"]}'
# Execute ams client with mocked response
with HTTMock(publish_mock):
msg = AmsMessage(data='foo1', attributes={'bar1': 'baz1'}).dict()
msg_bulk = [
AmsMessage(data='foo1', attributes={'bar1': 'baz1'}),
AmsMessage(data='foo2', attributes={'bar2': 'baz2'})
]
resp = self.ams.publish("topic1", msg)
assert resp["msgIds"][0] == "1"
resp_bulk = self.ams.publish("topic1", msg_bulk)
assert resp["msgIds"][0] == "1"
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/topics/topic1:publish",
method="POST")
def publish_bulk_mock(url, request):
assert url.path == "/v1/projects/TEST/topics/topic1:publish"
# Check request produced by ams client
req_body = json.loads(request.body)
self.assertEqual(req_body["messages"][0]["data"], "Zm9vMQ==")
self.assertEqual(req_body["messages"][0]["attributes"]["bar1"],
"baz1")
self.assertEqual(req_body["messages"][1]["data"], "Zm9vMg==")
self.assertEqual(req_body["messages"][1]["attributes"]["bar2"],
"baz2")
return '{"msgIds":["1", "2"]}'
with HTTMock(publish_bulk_mock):
msg_bulk = [
AmsMessage(data='foo1', attributes={'bar1': 'baz1'}),
AmsMessage(data='foo2', attributes={'bar2': 'baz2'})
]
resp_bulk = self.ams.publish("topic1", msg_bulk)
self.assertEqual(len(resp_bulk["msgIds"]), 2)
self.assertEqual(resp_bulk["msgIds"][0], "1")
self.assertEqual(resp_bulk["msgIds"][1], "2")
def testFailedPublish(self):
# Mock response for POST publish to topic
@urlmatch(netloc="localhost", path="/v1/projects/TEST/topics/topic1:publish",
method="POST")
def publish_mock(url, request):
assert url.path == "/v1/projects/TEST/topics/topic1:publish"
# Check request produced by ams client
req_body = json.loads(request.body)
assert req_body["messages"][0]["data"] == "Zm9vMQ=="
assert req_body["messages"][0]["attributes"]["bar1"] == "baz1"
return response(504, '<htmI><body><h1>504 Gateway Time-out</h1>\nThe server didn\'t respond in time.\n</body></html>\n', None, None, 5, request)
# Execute ams client with mocked response
with HTTMock(publish_mock):
msg = AmsMessage(data='foo1', attributes={'bar1': 'baz1'})
try:
resp = self.ams.publish("topic1", msg)
except Exception as e:
assert isinstance(e, AmsServiceException)
self.assertEqual(e.code, 504)
class TestMessage(unittest.TestCase):
def setUp(self):
m = AmsMessage()
self.message_callable = m(attributes={'foo': 'bar'}, data='baz')
self.message_callable_memoization = m(attributes={'foo1': 'bar1'})
self.message_callable_memoization_two = m(data='baz1')
self.message_send = AmsMessage(attributes={'foo': 'bar'}, data='baz')
self.message_send_no_payload = AmsMessage()
self.message_send_onlyattr = AmsMessage(attributes={'foo': 'bar'})
self.message_send_onlydata = AmsMessage(data='baz')
self.message_unicode = AmsMessage(data='ùňĭćőđĕ')
if sys.version_info >= (3, ):
self.message_unicode_py3 = AmsMessage(
data='ùňĭćőđĕḃẏṫєṡ'.encode('utf-8'))
self.message_recv = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='YmF6',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
self.message_recv_faulty = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='123456789thiswillfail',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
def test_MsgReadyToSend(self):
self.assertEqual(self.message_send.dict(), {
'attributes': {
'foo': 'bar'
},
'data': 'YmF6'
})
self.assertEqual(self.message_callable, {
'attributes': {
'foo': 'bar'
},
'data': 'YmF6'
})
self.assertEqual(self.message_callable_memoization, {
'attributes': {
'foo1': 'bar1'
},
'data': 'YmF6'
})
self.assertEqual(self.message_callable_memoization_two, {
'attributes': {
'foo1': 'bar1'
},
'data': 'YmF6MQ=='
})
self.assertEqual(self.message_send_onlyattr.dict(),
{'attributes': {
'foo': 'bar'
}})
self.assertEqual(self.message_send_onlydata.dict(), {'data': 'YmF6'})
self.assertEqual(self.message_unicode.dict(),
{'data': 'w7nFiMStxIfFkcSRxJU='})
if sys.version_info < (3, ):
self.assertEqual(self.message_unicode.get_data(), 'ùňĭćőđĕ')
else:
self.assertEqual(self.message_unicode.get_data(),
'ùňĭćőđĕ'.encode('utf-8'))
if sys.version_info >= (3, ):
self.assertEqual(self.message_unicode_py3.get_data(),
'ùňĭćőđĕḃẏṫєṡ'.encode('utf-8'))
self.message_send.set_data('baz')
if sys.version_info < (3, ):
self.assertEqual(self.message_send.get_data(), 'baz')
else:
self.assertEqual(self.message_send.get_data(), b'baz')
def test_MsgReadyToRecv(self):
if sys.version_info < (3, ):
self.assertEqual(self.message_recv.get_data(), 'baz')
else:
self.assertEqual(self.message_recv.get_data(), b'baz')
self.assertEqual(self.message_recv.get_msgid(), '1')
self.assertEqual(self.message_recv.get_publishtime(),
'2017-03-15T17:11:34.035345612Z')
self.assertEqual(self.message_recv.get_attr(), {'foo': 'bar'})
def test_MsgFaulty(self):
self.assertRaises(AmsMessageException,
self.message_recv_faulty.get_data)
self.assertRaises(AmsMessageException,
self.message_send_no_payload.get_data)
self.assertRaises(AmsMessageException,
self.message_send_no_payload.get_attr)
def setUp(self):
m = AmsMessage()
self.message_callable = m(attributes={'foo': 'bar'}, data='baz')
self.message_callable_memoization = m(attributes={'foo1': 'bar1'})
self.message_callable_memoization_two = m(data='baz1')
self.message_send = AmsMessage(attributes={'foo': 'bar'}, data='baz')
self.message_send_no_payload = AmsMessage()
self.message_send_onlyattr = AmsMessage(attributes={'foo': 'bar'})
self.message_send_onlydata = AmsMessage(data='baz')
self.message_unicode = AmsMessage(data='ùňĭćőđĕ')
if sys.version_info >= (3, ):
self.message_unicode_py3 = AmsMessage(
data='ùňĭćőđĕḃẏṫєṡ'.encode('utf-8'))
self.message_recv = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='YmF6',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
self.message_recv_faulty = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='123456789thiswillfail',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
class TestTopic(unittest.TestCase):
def setUp(self):
self.ams = ArgoMessagingService(endpoint="localhost",
token="s3cr3t",
project="TEST")
self.msg = AmsMessage(attributes={'foo': 'bar'}, data='baz')
self.submocks = SubMocks()
self.topicmocks = TopicMocks()
def testPublish(self):
# Mock response for POST publish to topic
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/topics/topic1:publish",
method="POST")
def publish_mock(url, request):
# Check request produced by ams client
req_body = json.loads(request.body)
self.assertEqual(req_body["messages"][0]["data"], 'YmF6')
self.assertEqual(req_body["messages"][0]["attributes"]["foo"],
"bar")
return '{"msgIds":["1"]}'
# Execute ams client with mocked response
with HTTMock(self.topicmocks.has_topic_mock, publish_mock):
topic = self.ams.topic('topic1')
resp = topic.publish(self.msg.dict())
# Assert that ams client handled the json response correctly
self.assertEqual(resp["msgIds"][0], "1")
self.assertEqual(topic.name, 'topic1')
self.assertEqual(topic.fullname, '/projects/TEST/topics/topic1')
def testSubscription(self):
# Execute ams client with mocked response
with HTTMock(self.topicmocks.create_topic_mock,
self.submocks.create_subscription_mock,
self.topicmocks.get_topic_mock,
self.submocks.has_subscription_mock):
topic = self.ams.topic('topic1')
sub = topic.subscription('subscription1')
# Assert that ams client handled the json response correctly
self.assertEqual(topic.name, 'topic1')
self.assertEqual(topic.fullname, '/projects/TEST/topics/topic1')
assert isinstance(sub, AmsSubscription)
self.assertEqual(sub.topic, topic)
self.assertEqual(sub.name, 'subscription1')
self.assertEqual(sub.fullname,
'/projects/TEST/subscriptions/subscription1')
def testIterSubscriptions(self):
# Mock response for GET Subscriptions request
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/subscriptions",
method="GET")
def iter_subs_mock(url, request):
assert url.path == "/v1/projects/TEST/subscriptions"
# Return two topics in json format
return response(
200,
'{"subscriptions":[{"name": "/projects/TEST/subscriptions/subscription1",\
"topic": "/projects/TEST/topics/topic1","pushConfig": \
{"pushEndpoint": "","retryPolicy": {}},"ackDeadlineSeconds": 10},\
{"name": "/projects/TEST/subscriptions/subscription2", \
"topic": "/projects/TEST/topics/topic1", \
"pushConfig": {"pushEndpoint": "","retryPolicy": {}},\
"ackDeadlineSeconds": 10}]}', None, None, 5,
request)
# Execute ams client with mocked response
with HTTMock(iter_subs_mock, self.topicmocks.create_topic_mock,
self.topicmocks.has_topic_mock):
topic = self.ams.topic('topic1')
resp = topic.iter_subs()
obj1 = next(resp)
obj2 = next(resp)
assert isinstance(obj1, AmsSubscription)
assert isinstance(obj2, AmsSubscription)
if sys.version_info < (3, ):
self.assertRaises(StopIteration, resp.next)
else:
self.assertRaises(StopIteration, resp.__next__)
self.assertEqual(obj1.name, 'subscription1')
self.assertEqual(obj2.name, 'subscription2')
def testDelete(self):
# Mock response for DELETE topic request
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/topics/topic1",
method="DELETE")
def delete_topic(url, request):
return response(200, '{}', None, None, 5, request)
# Execute ams client with mocked response
with HTTMock(delete_topic, self.topicmocks.create_topic_mock,
self.topicmocks.has_topic_mock):
topic = self.ams.topic('topic1')
self.assertTrue(topic.delete())
def testAcl(self):
# Mock response for GET topic request
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/topics/topic1:modifyAcl",
method="POST")
def modifyacl_topic_mock(url, request):
self.assertEqual(url.path,
"/v1/projects/TEST/topics/topic1:modifyAcl")
self.assertEqual(request.body,
'{"authorized_users": ["user1", "user2"]}')
# Return the details of a topic in json format
return response(200, '{}', None, None, 5, request)
# Mock response for GET topic request
@urlmatch(netloc="localhost",
path="/v1/projects/TEST/topics/topic1:acl",
method="GET")
def getacl_topic_mock(url, request):
assert url.path == "/v1/projects/TEST/topics/topic1:acl"
# Return the details of a topic in json format
return response(200, '{"authorized_users": ["user1", "user2"]}',
None, None, 5, request)
# Execute ams client with mocked response
with HTTMock(getacl_topic_mock, self.topicmocks.get_topic_mock,
modifyacl_topic_mock):
topic = self.ams.topic('topic1')
ret = topic.acl(['user1', 'user2'])
self.assertTrue(ret)
resp_users = topic.acl()
self.assertEqual(resp_users['authorized_users'],
['user1', 'user2'])
def setUp(self):
m = AmsMessage()
self.message_callable = m(attributes={'foo': 'bar'}, data='baz')
self.message_callable_memoization = m(attributes={'foo1': 'bar1'})
self.message_callable_memoization_two = m(data='baz1')
self.message_send = AmsMessage(attributes={'foo': 'bar'}, data='baz')
self.message_send_no_payload = AmsMessage()
self.message_send_onlyattr = AmsMessage(attributes={'foo': 'bar'})
self.message_send_onlydata = AmsMessage(data='baz')
self.message_recv = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='YmF6',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
self.message_recv_faulty = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='123456789thiswillfail',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
class TestMessage(unittest.TestCase):
def setUp(self):
m = AmsMessage()
self.message_callable = m(attributes={'foo': 'bar'}, data='baz')
self.message_callable_memoization = m(attributes={'foo1': 'bar1'})
self.message_callable_memoization_two = m(data='baz1')
self.message_send = AmsMessage(attributes={'foo': 'bar'}, data='baz')
self.message_send_no_payload = AmsMessage()
self.message_send_onlyattr = AmsMessage(attributes={'foo': 'bar'})
self.message_send_onlydata = AmsMessage(data='baz')
self.message_recv = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='YmF6',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
self.message_recv_faulty = AmsMessage(
b64enc=False,
attributes={'foo': 'bar'},
data='123456789thiswillfail',
messageId='1',
publishTime='2017-03-15T17:11:34.035345612Z')
def test_MsgReadyToSend(self):
self.assertEqual(self.message_send.dict(), {
'attributes': {
'foo': 'bar'
},
'data': 'YmF6'
})
self.assertEqual(self.message_callable, {
'attributes': {
'foo': 'bar'
},
'data': 'YmF6'
})
self.assertEqual(self.message_callable_memoization, {
'attributes': {
'foo1': 'bar1'
},
'data': 'YmF6'
})
self.assertEqual(self.message_callable_memoization_two, {
'attributes': {
'foo1': 'bar1'
},
'data': 'YmF6MQ=='
})
self.assertEqual(self.message_send_onlyattr.dict(),
{'attributes': {
'foo': 'bar'
}})
self.assertEqual(self.message_send_onlydata.dict(), {'data': 'YmF6'})
self.message_send.set_data('baz')
self.assertEqual(self.message_send.get_data(), 'baz')
def test_MsgReadyToRecv(self):
self.assertEqual(self.message_recv.get_data(), 'baz')
self.assertEqual(self.message_recv.get_msgid(), '1')
self.assertEqual(self.message_recv.get_publishtime(),
'2017-03-15T17:11:34.035345612Z')
self.assertEqual(self.message_recv.get_attr(), {'foo': 'bar'})
def test_MsgFaulty(self):
self.assertRaises(AmsMessageException,
self.message_recv_faulty.get_data)
self.assertRaises(AmsMessageException,
self.message_send_no_payload.get_data)
self.assertRaises(AmsMessageException,
self.message_send_no_payload.get_attr)
