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)