add test for duplicate message id

Author: surajit

tests/conftest.py

@@ -9,6 +9,10 @@
 import schemas.schema
 from crud.company import create_company
 from crud.contact import create_contact
+from crud.message import save_message
+from models.message import MessageType
+from models.message import Carrier
+from models.message import Message
 from models.payment import Organization
 from tests.test_database import SQLALCHEMY_DATABASE_URL
 
@@ -51,6 +55,19 @@ def organization(db):
     create_company(db, schemas.schema.CreateCompany(name="Test org", org_id=12345))
 
 
+@pytest.fixture
+def sendgrid_message(db):
+    message_data = dict(
+        message_id="123657ab",
+        status_code=202,
+        message_type=MessageType.EMAIL.value,
+        carrier=Carrier.SENDGRID.value,
+    )
+    db_item = Message(**message_data)
+    db.add(db_item)
+    db.commit()
+
+
 @pytest.fixture
 def contact(db):
     organization = db.query(Organization).filter_by(name="Test org").all()[0]

tests/routes/test_email.py

@@ -47,3 +47,20 @@ def test_email_send_with_sendgrid_bad_request_exception(mock_send_email, client)
     response = client.post("/api/v1/email/send/", json=read_json("email_data.json"))
     assert response.status_code == 200
     assert response.json() == {"detail": "Bad request"}
+
+
+@mock.patch("routes.v1.email.email.send_email")
+def test_email_send_with_db_duplicate_message_id(
+    mock_send_email, client, sendgrid_message
+):
+    sendgrid_response_mock = SendGridResponseMock(
+        headers={"x-message-id": "123657ab"}, status_code=202
+    )
+    mock_send_email.return_value = sendgrid_response_mock
+    response = client.post("/api/v1/email/send/", json=read_json("email_data.json"))
+    assert response.status_code == 200
+    assert response.json() == {
+        "detail": "duplicate key value violates unique constraint "
+        '"ix_message_message_id"\nDETAIL:  Key (message_id)=('
+        "123657ab) already exists.\n"
+    }