add test for log dynamo db

Author: surajit

aws/services/dynamo_db/logs.py

@@ -25,6 +25,6 @@ async def create_log(data):
             "request_ip": data["request_ip"],
             "message": json.dumps(data["message"]),
         }
-        return table.put_item(Item=trans)
+        return table.put_item(trans)
     except botocore.exceptions.ClientError as e:
         raise BotoClientException(e)

logger/log.py

@@ -59,7 +59,8 @@ async def wrapper(*args, **kwargs):
                 request_ip=client,
                 message=message
             )
-            await create_log(data)
+            res = await create_log(data)
+            print('res',res)
         except Exception as exc:
             logger.exception(f"Exception occurred {exc}")
             pass

tests/data/dynamo_db_response.json

@@ -0,0 +1,16 @@
+{
+  "ResponseMetadata": {
+      "RequestId": "PFK7OPBR2L9FLV3N9OFCHO3VJNVV4KQNSO5A",
+      "HTTPStatusCode": 200,
+      "HTTPHeaders": {
+        "server": "Server",
+        "date": "Sat, 28 May 2022 18:32:28 GMT",
+        "content-type": "application/x-amz-json-1.0",
+        "content-length": "2",
+        "connection": "keep-alive",
+        "x-amzn-requestid": "PFK7OPBR2L9FLV3N9OFCHO3V",
+        "x-amz-crc32": "274561"
+      },
+      "RetryAttempts": 0
+    }
+  }

tests/integrations/common.py

@@ -0,0 +1,8 @@
+import json
+
+from settings import TEST_DATA_DIR
+
+
+def read_json(filename):
+    with open(TEST_DATA_DIR.joinpath(filename)) as resp_file:
+        return json.load(resp_file)

tests/integrations/test_dynamo_db.py

@@ -0,0 +1,53 @@
+import pytest
+from unittest import mock
+
+from aws.services.dynamo_db.logs import create_log, BotoClientException
+from tests.integrations.common import read_json
+
+
+class MockDynamoDbTable:
+    def put_item(self, item):
+        response = read_json("dynamo_db_response.json")
+        return response
+
+
+@mock.patch("aws.services.dynamo_db.logs.dynamodb.Table")
+@mock.patch("aws.services.dynamo_db.logs.boto3.resource")
+@pytest.mark.asyncio
+async def test_dynamo_db_log(mock_boto3_connect, mock_dynamo_db_table):
+    mock_boto3_connect.return_value = "success"
+    mock_dynamo_db_table.return_value = MockDynamoDbTable()
+    data = {
+        "log_id": 12345,
+        "request_ip": "localhost",
+        "message": {
+            "url": "http://127.0.0.1:8000/api/v1/companies/",
+            "method": "POST",
+            "data": [{"org_id": 1234, "name": "New test"}],
+        },
+    }
+    response = await create_log(data)
+    assert response == read_json("dynamo_db_response.json")
+
+
+@mock.patch("aws.services.dynamo_db.logs.dynamodb.Table")
+@mock.patch("aws.services.dynamo_db.logs.boto3.resource")
+@pytest.mark.asyncio
+async def test_dynamo_db_log_raises_client_error(
+    mock_boto3_connect, mock_dynamo_db_table
+):
+    mock_boto3_connect.return_value = "success"
+    mock_dynamo_db_table.side_effect = BotoClientException("Something went wrong")
+    data = {
+        "log_id": 12345,
+        "request_ip": "localhost",
+        "message": {
+            "url": "http://127.0.0.1:8000/api/v1/companies/",
+            "method": "POST",
+            "data": [{"org_id": 1234, "name": "New test"}],
+        },
+    }
+    with pytest.raises(BotoClientException) as exc:
+        await create_log(data)
+    assert isinstance(exc.value, BotoClientException)
+    assert exc.value.args[0] == "Something went wrong"

tests/integrations/test_sendgrid.py

@@ -1,16 +1,10 @@
-import json
 import pytest
 from unittest import mock
 
 from email_api.email import send_email
 from email_api.email import UnauthorizedException
 from email_api.email import BadRequestException
-from settings import TEST_DATA_DIR
-
-
-def read_json(filename):
-    with open(TEST_DATA_DIR.joinpath(filename)) as resp_file:
-        return json.load(resp_file)
+from tests.integrations.common import read_json
 
 
 @mock.patch("email_api.email.SendGridAPIClient.send")