Added tests for RpcAmqpClient

Author: Relrin

tests/amqp/test_rpc_client.py

@@ -0,0 +1,71 @@
+import pytest
+from sage_utils.amqp.clients import RpcAmqpClient
+from sage_utils.amqp.extension import AmqpExtension
+from sage_utils.constants import VALIDATION_ERROR
+from sage_utils.wrappers import Response
+
+from tests.fixtures import Application, FakeConfig, FakeRegisterMicroserviceWorker
+
+
+REQUEST_QUEUE = FakeRegisterMicroserviceWorker.QUEUE_NAME
+REQUEST_EXCHANGE = FakeRegisterMicroserviceWorker.REQUEST_EXCHANGE_NAME
+RESPONSE_EXCHANGE_NAME = FakeRegisterMicroserviceWorker.RESPONSE_EXCHANGE_NAME
+VALIDATION_ERROR_DECR = FakeRegisterMicroserviceWorker.ERROR_DESCRIPTION
+
+
+@pytest.mark.asyncio
+async def test_rpc_amqp_client_returns_ok(event_loop):
+    app = Application(config=FakeConfig(), loop=event_loop)
+    register_worker = FakeRegisterMicroserviceWorker(app)
+    extension = AmqpExtension(app)
+    extension.register_worker(register_worker)
+
+    await extension.init(event_loop)
+
+    client = RpcAmqpClient(
+        app=app,
+        routing_key=REQUEST_QUEUE,
+        request_exchange=REQUEST_EXCHANGE,
+        response_queue='',
+        response_exchange=RESPONSE_EXCHANGE_NAME
+    )
+    response = await client.send(payload={'name': 'microservice', 'version': '1.0.0'})
+
+    assert Response.CONTENT_FIELD_NAME in response.keys()
+    assert response[Response.CONTENT_FIELD_NAME] == 'OK'
+
+    assert Response.EVENT_FIELD_NAME in response.keys()
+    assert response[Response.EVENT_FIELD_NAME] is None
+
+    await extension.deinit(event_loop)
+
+
+@pytest.mark.asyncio
+async def test_rpc_amqp_client_returns_an_error(event_loop):
+    app = Application(config=FakeConfig(), loop=event_loop)
+    register_worker = FakeRegisterMicroserviceWorker(app)
+    extension = AmqpExtension(app)
+    extension.register_worker(register_worker)
+
+    await extension.init(event_loop)
+
+    client = RpcAmqpClient(
+        app=app,
+        routing_key=REQUEST_QUEUE,
+        request_exchange=REQUEST_EXCHANGE,
+        response_queue='',
+        response_exchange=RESPONSE_EXCHANGE_NAME
+    )
+    response = await client.send(payload={})
+
+    assert Response.ERROR_FIELD_NAME in response.keys()
+    assert 'message' in response[Response.ERROR_FIELD_NAME].keys()
+    assert response[Response.ERROR_FIELD_NAME]['message'] == VALIDATION_ERROR_DECR
+
+    assert 'type' in response[Response.ERROR_FIELD_NAME].keys()
+    assert response[Response.ERROR_FIELD_NAME]['type'] == VALIDATION_ERROR
+
+    assert Response.EVENT_FIELD_NAME in response.keys()
+    assert response[Response.EVENT_FIELD_NAME] is None
+
+    await extension.deinit(event_loop)

tests/fixtures.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 import json
 
@@ -14,15 +15,16 @@ def get(self, key, default=None):
 
 class Application(object):
 
-    def __init__(self, config=None):
+    def __init__(self, config=None, loop=None):
         super(Application, self).__init__()
         self.config = config
+        self.loop = loop or asyncio.get_event_loop()
 
 
 class BaseWorker(AmqpWorker):
     QUEUE_NAME = 'process-request'
-    REQUEST_EXCHANGE_NAME = 'amqp.direct'
-    RESPONSE_EXCHANGE_NAME = 'amqp.direct'
+    REQUEST_EXCHANGE_NAME = 'amq.direct'
+    RESPONSE_EXCHANGE_NAME = 'amq.direct'
     CONTENT_TYPE = 'application/json'
 
     def generate_response(self, data):
@@ -67,10 +69,11 @@ async def run(self, *args, **kwargs):
 
 class FakeRegisterMicroserviceWorker(BaseWorker):
     QUEUE_NAME = 'microservice.register'
+    ERROR_DESCRIPTION = "Name and version must be specified."
 
     def generate_response(self, raw_data):
         data = json.loads(raw_data.strip())
 
         if 'name' not in data.keys() and 'version' not in data.keys():
-            return Response.from_error(VALIDATION_ERROR, "Name and version must be specified.")
+            return Response.from_error(VALIDATION_ERROR, self.ERROR_DESCRIPTION)
         return Response.with_content("OK")