add tests for charge

Author: surajit

crud/charge.py

@@ -2,6 +2,7 @@
 from sqlalchemy.orm import Session
 
 from models import payment
+from models.payment import Organization
 
 
 class ChargeExistException(Exception):

tests/conftest.py

@@ -7,12 +7,12 @@
 from sqlalchemy.orm import Session
 
 import schemas.schema
+from crud.charge import add_charge
 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
 
@@ -52,20 +52,36 @@ def client(db):
 
 @pytest.fixture
 def organization(db):
-    create_company(db, schemas.schema.CreateCompany(name="Test org", org_id=12345))
+    yield create_company(
+        db, schemas.schema.CreateCompany(name="Test org", org_id=12345)
+    )
+
+
+@pytest.fixture
+@pytest.mark.asyncio
+async def charge_onboarding(db, organization):
+    await add_charge(
+        db,
+        schemas.schema.CreateCharge(
+            company_name="Test org",
+            currency="USD",
+            charge_type="ONBOARDING",
+            amount=200,
+        ),
+        organization.id,
+    )
 
 
 @pytest.fixture
-def message(db):
+@pytest.mark.asyncio
+async def 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()
+    await save_message(db, message_data)
 
 
 @pytest.fixture

tests/routes/test_charge.py

@@ -0,0 +1,38 @@
+from unittest import mock
+
+import pytest
+
+
+@mock.patch("logger.log.logger.debug")
+@pytest.mark.asyncio
+async def test_create_charge(mock_logger_debug, organization, client):
+    mock_logger_debug.return_value = "Ok"
+    response = client.post(
+        "/api/v1/charges/",
+        json={
+            "currency": "USD",
+            "charge_type": "MAINTENANCE",
+            "amount": 200,
+            "company_name": "Test org",
+        },
+    )
+    assert response.status_code == 201
+    assert response.json()["charge_type"] == "maintenance"
+
+
+@mock.patch("logger.log.logger.debug")
+@pytest.mark.asyncio
+async def test_create_charge_for_duplicate_charge(
+    mock_logger_debug, charge_onboarding, client
+):
+    mock_logger_debug.return_value = "Ok"
+    response = client.post(
+        "/api/v1/charges/",
+        json={
+            "currency": "USD",
+            "charge_type": "ONBOARDING",
+            "amount": 200,
+            "company_name": "Test org",
+        },
+    )
+    assert response.status_code == 200