with transactions

Author: harishkannarao

app/routers/customer.py

@@ -16,19 +16,24 @@
 
 @router.get("")
 async def read_customers() -> list[Customer]:
-    rows = await database.fetch_all(query=SELECT_ALL_CUSTOMERS)
-    customers: list[Customer] = list(map(lambda row: Customer(**dict(row)), rows))
-    return customers
+    async with database.transaction():
+        rows = await database.fetch_all(query=SELECT_ALL_CUSTOMERS)
+        customers: list[Customer] = list(map(lambda row: Customer(**dict(row)), rows))
+        return customers
 
 
 @router.put("", status_code=status.HTTP_204_NO_CONTENT, tags=["users"])
 async def insert_customers(customers: list[Customer]) -> None:
-    rows: list[dict[str, Any]] = list(map(lambda customer: vars(customer), customers))
-    await database.execute_many(query=INSERT_CUSTOMER, values=rows)
-    return None
+    async with database.transaction():
+        rows: list[dict[str, Any]] = list(
+            map(lambda customer: vars(customer), customers)
+        )
+        await database.execute_many(query=INSERT_CUSTOMER, values=rows)
+        return None
 
 
 @router.delete("", status_code=status.HTTP_204_NO_CONTENT, tags=["users"])
 async def delete_all_customers() -> None:
-    await database.execute(query=DELETE_ALL_CUSTOMERS)
-    return None
+    async with database.transaction():
+        await database.execute(query=DELETE_ALL_CUSTOMERS)
+        return None

tests_unit/routers/test_customer.py

@@ -13,6 +13,14 @@
 CUSTOMERS_ENDPOINT = "/context/customers"
 
 
+@pytest.fixture
+def mock_transaction(mocker: MockerFixture) -> AsyncMock:
+    mock_transaction: AsyncMock = mocker.patch(
+        "app.db.database_config.database.transaction"
+    )
+    return mock_transaction
+
+
 @pytest.fixture
 def mock_execute(mocker: MockerFixture) -> AsyncMock:
     mock_execute: AsyncMock = mocker.patch("app.db.database_config.database.execute")
@@ -35,15 +43,19 @@ def mock_fetch_all(mocker: MockerFixture) -> AsyncMock:
     return mock_fetch_all
 
 
-def test_customers_delete(mock_execute: AsyncMock, test_client: TestClient):
+def test_customers_delete(
+    mock_transaction: AsyncMock, mock_execute: AsyncMock, test_client: TestClient
+):
     response: Response = test_client.delete(CUSTOMERS_ENDPOINT)
     assert_that(response.status_code).is_equal_to(204)
 
     assert len(mock_execute.call_args_list) == 1
     assert mock_execute.call_args.kwargs["query"] == "TRUNCATE TABLE CUSTOMERS"
 
 
-def test_customers_insert(mock_execute_many: AsyncMock, test_client: TestClient):
+def test_customers_insert(
+    mock_transaction: AsyncMock, mock_execute_many: AsyncMock, test_client: TestClient
+):
     customer1 = Customer(first_name="fname1", last_name="lname1")
     customer2 = Customer(first_name="fname2", last_name="lname2")
     input_json = jsonable_encoder([customer1, customer2])
@@ -61,7 +73,9 @@ def test_customers_insert(mock_execute_many: AsyncMock, test_client: TestClient)
     assert_that(inserted_rows).contains(jsonable_encoder(customer2))
 
 
-def test_customers_read(mock_fetch_all: AsyncMock, test_client: TestClient):
+def test_customers_read(
+    mock_transaction: AsyncMock, mock_fetch_all: AsyncMock, test_client: TestClient
+):
     customer1 = Customer(first_name="fname1", last_name="lname1")
     customer2 = Customer(first_name="fname2", last_name="lname2")