diff --git a/expenses_manager/expenses/tests/test_dashboard.py b/expenses_manager/expenses/tests/test_dashboard.py index 8e23b10..2d19d89 100644 --- a/expenses_manager/expenses/tests/test_dashboard.py +++ b/expenses_manager/expenses/tests/test_dashboard.py @@ -2,7 +2,7 @@ import pytest from datetime import date from decimal import Decimal from django.urls import reverse -from expenses.models import Expense, Category +from expenses.models import Account, Expense, Category pytestmark = pytest.mark.django_db @@ -33,23 +33,28 @@ def test_dashboard_groups_by_category(client, django_user_model): food = Category.objects.create(name='Food', owner=user) rent = Category.objects.create(name='Rent', owner=user) + general_account = Account.objects.create(name='General', owner=user, initial_balance=300, active=True) + Expense.objects.create( owner=user, category=food, amount=Decimal('10'), date=date(2024, 1, 1), + account=general_account, ) Expense.objects.create( owner=user, category=food, amount=Decimal('5'), date=date(2024, 1, 2), + account=general_account, ) Expense.objects.create( owner=user, category=rent, amount=Decimal('20'), date=date(2024, 1, 3), + account=general_account, ) response = client.get( @@ -70,12 +75,14 @@ def test_dashboard_filters_by_year(client, django_user_model): client.login(username='test', password='1234') cat = Category.objects.create(name='General', owner=user) + general_account = Account.objects.create(name='General', owner=user, initial_balance=300, active=True) Expense.objects.create( owner=user, category=cat, amount=10, date=date(2023, 5, 1), + account=general_account, ) Expense.objects.create( @@ -83,6 +90,7 @@ def test_dashboard_filters_by_year(client, django_user_model): category=cat, amount=20, date=date(2024, 5, 1), + account=general_account, ) response = client.get( diff --git a/expenses_manager/expenses/tests/test_expenses.py b/expenses_manager/expenses/tests/test_expenses.py index b84c873..9109c14 100644 --- a/expenses_manager/expenses/tests/test_expenses.py +++ b/expenses_manager/expenses/tests/test_expenses.py @@ -2,7 +2,7 @@ import pytest from datetime import date from decimal import Decimal from django.urls import reverse -from expenses.models import Expense, Category, Tag +from expenses.models import Expense, Category, Tag, Account pytestmark = pytest.mark.django_db @@ -13,11 +13,14 @@ def test_expense_can_have_tags(client, django_user_model): tag = Tag.objects.create(name='Food', owner=user) category = Category.objects.create(name='General', owner=user) + general_account = Account.objects.create(name='General', owner=user, initial_balance=300, active=True) + expense = Expense.objects.create( owner=user, category=category, amount=10, date=date.today(), + account=general_account, ) expense.tags.add(tag)