def setUp(self):
self.user = UserFactory()
self.account = AccountFactory(user=self.user)
self.income_category = IncomeCategoryFactory(user=self.user)
self.income = IncomeFactory(user=self.user,
account=self.account,
category=self.income_category)
def setUp(self):
self.user = UserFactory()
self.account = AccountFactory(user=self.user)
self.expense_category = ExpenseCategoryFactory(user=self.user)
self.expense = ExpenseFactory(user=self.user,
account=self.account,
category=self.expense_category)
def test_get_expenses_by_account(self):
"""
"""
COUNT = random.randint(1, 10)
account = AccountFactory()
ExpenseFactory.create_batch(COUNT, account=account)
expenses = services.get_expenses_by_account(account.id)
self.assertEqual(COUNT, expenses.count())
def test_reverse_link_from_expense_category_model(self):
"""Test reverse link from ExpenseCategory model."""
user = UserFactory()
account = AccountFactory(user=user)
expense_category = ExpenseCategoryFactory(user=user)
ExpenseFactory.create_batch(
7,
user=user,
account=account,
category=expense_category
)
self.assertEqual(expense_category.expenses.count(), 7)
def test_reverse_link_from_income_category_model(self):
"""Test reverse link from IncomeCategory model."""
user = UserFactory()
account = AccountFactory(user=user)
income_category = IncomeCategoryFactory(user=user)
IncomeFactory.create_batch(
6,
user=user,
account=account,
category=income_category
)
self.assertEqual(income_category.incomes.count(), 6)