tests/utils.py

from datetime import datetime
from typing import List
from app.schema.index import TransactionCreate, TransactionType


def get_fake_transactions(user_id: int) -> List[TransactionCreate]:
    return [
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 1),
            category="category1",
            name_description="name_description",
            amount=1.0,
            type=TransactionType.EXPENSE,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 2),
            category="category2",
            name_description="name_description",
            amount=2.0,
            type=TransactionType.EXPENSE,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 3),
            category="category3",
            name_description="name_description",
            amount=3.0,
            type=TransactionType.INCOME,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 4),
            category="category1",
            name_description="name_description",
            amount=4.0,
            type=TransactionType.INCOME,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 5),
            category="category2",
            name_description="name_description",
            amount=5.0,
            type=TransactionType.EXPENSE,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 6),
            category="category3",
            name_description="name_description",
            amount=6.0,
            type=TransactionType.EXPENSE,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 7),
            category="category1",
            name_description="name_description",
            amount=7.0,
            type=TransactionType.INCOME,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 8),
            category="category2",
            name_description="name_description",
            amount=8.0,
            type=TransactionType.INCOME,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 9),
            category="category3",
            name_description="name_description",
            amount=9.0,
            type=TransactionType.EXPENSE,
        ),
        TransactionCreate(
            user_id=user_id,
            transaction_date=datetime(2022, 1, 10),
            category="category1",
            name_description="name_description",
            amount=10.0,
            type=TransactionType.EXPENSE,
        ),
    ]