Complete unit tests

Author: harishkannarao

tests_unit/routers/test_customer.py

@@ -3,9 +3,9 @@
 
 import pytest
 from assertpy import assert_that
-from fastapi import Response
 from fastapi.encoders import jsonable_encoder
 from fastapi.testclient import TestClient
+from httpx import Response
 from pytest_mock import MockerFixture
 
 from app.model.customer import Customer
@@ -19,12 +19,16 @@ def mock_execute(mocker: MockerFixture) -> AsyncMock:
 
 @pytest.fixture
 def mock_execute_many(mocker: MockerFixture) -> AsyncMock:
-    mock_execute_many: AsyncMock = mocker.patch(
-        "app.db.database_config.database.execute_many"
-    )
+    mock_execute_many: AsyncMock = mocker.patch("app.db.database_config.database.execute_many")
     return mock_execute_many
 
 
+@pytest.fixture
+def mock_fetch_all(mocker: MockerFixture) -> AsyncMock:
+    mock_fetch_all: AsyncMock = mocker.patch("app.db.database_config.database.fetch_all")
+    return mock_fetch_all
+
+
 def test_customers_delete(mock_execute: AsyncMock, test_client: TestClient):
     initial_delete: Response = test_client.delete("/context/customers")
     assert_that(initial_delete.status_code).is_equal_to(204)
@@ -48,3 +52,24 @@ def test_customers_insert(mock_execute_many: AsyncMock, test_client: TestClient)
     assert_that(inserted_rows).is_length(2)
     assert_that(inserted_rows).contains(jsonable_encoder(customer1))
     assert_that(inserted_rows).contains(jsonable_encoder(customer2))
+
+
+def test_customers_read(mock_fetch_all: AsyncMock, test_client: TestClient):
+    customer1 = Customer(first_name="fname1", last_name="lname1")
+    customer2 = Customer(first_name="fname2", last_name="lname2")
+
+    mock_fetch_all.return_value = jsonable_encoder([customer1, customer2])
+
+    response: Response = test_client.get("/context/customers")
+
+    assert_that(response.status_code).is_equal_to(200)
+    response_json: list[dict[str, Any]] = response.json()
+
+    assert_that(response_json).is_length(2)
+    assert_that(response_json).contains(jsonable_encoder(customer1))
+    assert_that(response_json).contains(jsonable_encoder(customer2))
+
+    assert len(mock_fetch_all.call_args_list) == 1
+    assert_that(mock_fetch_all.call_args.kwargs["query"]).is_equal_to(
+        "SELECT * FROM CUSTOMERS"
+    )