async def test_request_reply_success(self):
""" check JSON messages can be published and received """
service_name = "clock"
req = Requester(
amqp_url=AMQP_URL,
service_name=service_name,
serialization=CONTENT_TYPE_JSON,
compression=COMPRESSION_GZIP,
)
async def on_request(payload, message):
now = datetime.datetime.now(
tz=datetime.timezone.utc if payload["utc"] else None)
return dict(timestamp=now.timestamp())
rep = Responder(
amqp_url=AMQP_URL,
service_name=service_name,
serialization=CONTENT_TYPE_JSON,
compression=COMPRESSION_BZ2,
on_request=on_request,
)
try:
await req.start()
await rep.start()
request_msg = dict(abc=123, utc=True)
response_msg = await req.request(request_msg)
self.assertIn("timestamp", response_msg)
finally:
await req.stop()
await rep.stop()
async def test_start_stop_responder_twice(self):
""" check calling start and stop twice on responder """
async def on_request(payload, message):
return {}
rep = Responder(
amqp_url=AMQP_URL,
service_name="test",
serialization=CONTENT_TYPE_JSON,
on_request=on_request,
)
await rep.start()
await rep.start()
await rep.stop()
await rep.stop()
async def test_requester_non_default_exchange_name(self):
""" check requester using non-default exchange """
service_name = "clock"
exchange_name = "direct-test"
req = Requester(
amqp_url=AMQP_URL,
service_name=service_name,
exchange_name=exchange_name,
serialization=CONTENT_TYPE_JSON,
)
async def on_request(payload, message):
now = datetime.datetime.now(
tz=datetime.timezone.utc if payload["utc"] else None
)
return dict(timestamp=now.timestamp())
rep = Responder(
amqp_url=AMQP_URL,
service_name=service_name,
exchange_name=exchange_name,
serialization=CONTENT_TYPE_JSON,
on_request=on_request,
)
try:
await req.start()
await rep.start()
# await asyncio.sleep(20.0)
request_msg = dict(abc=123, utc=True)
response_msg = await req.request(request_msg)
self.assertIn("timestamp", response_msg)
finally:
await req.stop()
await rep.stop()
async def test_request_reply_failure_message_timeout(self):
""" check messages that are not handled in time raise a TimeoutError """
service_name = "clock"
req = Requester(
amqp_url=AMQP_URL,
service_name=service_name,
serialization=CONTENT_TYPE_JSON,
)
async def on_request(payload, message):
# Simulate a failure that causes the request message to be rejected
# by the responder thereby causing a response to timeout.
raise Exception("BOOM!")
rep = Responder(
amqp_url=AMQP_URL,
service_name=service_name,
serialization=CONTENT_TYPE_JSON,
on_request=on_request,
)
try:
await req.start()
await rep.start()
request_msg = dict(abc=123, utc=True)
with self.assertRaises(asyncio.TimeoutError):
# Absord the intentioal exception error reported when we
# generate an exception in the responder handler function.
with self.assertLogs("gestalt.amq.responder"):
response_msg = await req.request(request_msg)
finally:
await req.stop()
await rep.stop()