test: pytest for transactions

app/model/transaction.py

@@ -31,12 +31,9 @@ class Transaction(Base, BaseModel):
 
     @classmethod
     async def bulk_create(cls: "type[Transaction]", db: AsyncSession, transactions: List[TransactionCreate]) -> None:
-        transactions_list = [cls(transaction_date=transaction.transaction_date,
-                                 category=transaction.category,
-                                 type=transaction.type,
-                                 amount= transaction.amount,
-                                 name_description=transaction.name_description) for transaction in transactions]
-        db.add(transactions_list)
+        values = [transaction.model_dump() for transaction in transactions]
+        query = sql.insert(cls).values(values)
+        await db.execute(query)
         await db.commit()
 
     @classmethod

tests/conftest.py

@@ -37,7 +37,7 @@ def client(app):
         yield c
 
 
-@pytest.mark.asyncio(scope="session")
+@pytest.fixture(scope="session")
 def event_loop(request):
     loop = asyncio.get_event_loop_policy().new_event_loop()
     yield loop
@@ -52,9 +52,9 @@ async def connection_test(test_db, event_loop):
     pg_db = test_db.dbname
     pg_password = test_db.password
 
-    with DatabaseJanitor(pg_user, pg_host, pg_port, pg_db, test_db.version, pg_password):
-        connection_str = f"postgresql+asyncpg://{pg_user}:@{pg_host}:{pg_port}/{pg_db}"
-        sessionmanager.init(connection_str)
+    with DatabaseJanitor(user=pg_user, host=pg_host, port=pg_port, dbname=pg_db, version=test_db.version, password=pg_password):
+        connection_str = f"postgresql+psycopg://{pg_user}:@{pg_host}:{pg_port}/{pg_db}"
+        sessionmanager.init(connection_str, {"echo": True, "future": True})
         yield
         await sessionmanager.close()
 
@@ -73,3 +73,9 @@ async def session_override(app, connection_test):
             yield session
 
     app.dependency_overrides[get_db_session] = get_db_session_override
+
+
+@pytest.fixture(scope="function", autouse=True)
+async def get_db_session_fixture():
+    async with sessionmanager.session() as session:
+        yield session

tests/test_transactions.py

@@ -0,0 +1,109 @@
+from datetime import datetime
+from typing import List
+from fastapi import Depends
+from fastapi.testclient import TestClient
+import pytest
+
+from app.model.transaction import Transaction
+from app.schema.index import TransactionType, TransactionCreate
+
+from sqlalchemy.ext.asyncio import AsyncSession
+from app.engine.postgresdb import get_db_session
+from app.model.user import User
+
+def get_fake_transactions(user_id: int) -> List[TransactionCreate]:
+    return [
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 1),
+            category="category",
+            name_description="name_description",
+            amount=1.0,
+            type=TransactionType.EXPENSE,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 2),
+            category="category",
+            name_description="name_description",
+            amount=2.0,
+            type=TransactionType.EXPENSE,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 3),
+            category="category",
+            name_description="name_description",
+            amount=3.0,
+            type=TransactionType.INCOME,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 4),
+            category="category",
+            name_description="name_description",
+            amount=4.0,
+            type=TransactionType.INCOME,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 5),
+            category="category",
+            name_description="name_description",
+            amount=5.0,
+            type=TransactionType.EXPENSE,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 6),
+            category="category",
+            name_description="name_description",
+            amount=6.0,
+            type=TransactionType.EXPENSE,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 7),
+            category="category",
+            name_description="name_description",
+            amount=7.0,
+            type=TransactionType.INCOME,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 8),
+            category="category",
+            name_description="name_description",
+            amount=8.0,
+            type=TransactionType.INCOME,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 9),
+            category="category",
+            name_description="name_description",
+            amount=9.0,
+            type=TransactionType.EXPENSE,
+        ),
+        TransactionCreate(
+            user_id=user_id,
+            transaction_date=datetime(2022, 1, 10),
+            category="category",
+            name_description="name_description",
+            amount=10.0,
+            type=TransactionType.EXPENSE,
+        ),
+    ]
+
+@pytest.mark.asyncio
+async def test_transactions(client: TestClient, get_db_session_fixture: AsyncSession) -> None:
+
+    session_override = get_db_session_fixture
+    user = await User.create(session_override, name="user", email="email", hashed_password="password")
+
+    fake_transactions = get_fake_transactions(user.id)
+    await Transaction.bulk_create(session_override, fake_transactions)
+
+    response = client.get("/api/v1/transactions/1")
+    assert response.status_code == 200
+    assert len(response.json()) == 10