add tests for email

Author: surajit

settings.py

@@ -1,7 +1,9 @@
 import os
-
+from pathlib import Path
 from decouple import config
 
+BASE_DIR = Path(__file__).resolve(strict=True).parent
+
 DATABASE_HOST = config("DATABASE_HOST")
 DATABASE = config("DATABASE_NAME")
 DATABASE_USER = config("DATABASE_USER")
@@ -17,3 +19,5 @@
 SENDGRID_API_KEY = config("SENDGRID_API_KEY")
 WELCOME_MESSAGE_TEMPLATE_ID = config("WELCOME_MESSAGE_TEMPLATE_ID")
 PAYMENT_CONFIRMATION_TEMPLATE_ID = config("WELCOME_MESSAGE_TEMPLATE_ID")
+
+TEST_DATA_DIR = BASE_DIR.joinpath('tests/data')

tests/data/__init_-.py

@@ -0,0 +1,18 @@
+{
+    "recipients": [
+        {
+            "recipient_email": "[email protected]",
+            "recipient_name": "DC"
+        },
+        {
+            "recipient_email": "[email protected]",
+            "recipient_name": "Flash"
+        }
+    ],
+    "template_content": {
+        "content": "Hey there DC please continue with the Flash seasons"
+    },
+    "sender": "[email protected]",
+    "subject": "Flash Fan",
+    "email_type": "welcome_email"
+}

tests/data/email_data.json

@@ -0,0 +1,31 @@
+import json
+from unittest import mock
+
+from models.message import Message
+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)
+
+
+class SendGridResponseMock:
+    def __init__(self, headers, status_code):
+        self.headers = headers
+        self.status_code = status_code
+
+
+@mock.patch("routes.v1.email.email.send_email")
+def test_email_send(mock_send_email, client, db):
+    sendgrid_response_mock = SendGridResponseMock(
+        headers={"x-message-id": "xvis10203sn"}, status_code=202
+    )
+    mock_send_email.return_value = sendgrid_response_mock
+    response = client.post("/api/v1/email/send/", json=read_json("email_data.json"))
+    message = db.query(Message).filter_by(message_id="xvis10203sn").all()[0]
+    assert response.status_code == 200
+    assert message.message_id == "xvis10203sn"
+    assert message.message_type.value == "EMAIL"
+    assert message.status_code == "202"
+    assert message.carrier.value == "SENDGRID"