def test_lists_only_owned_budgets(self):
"""Test that a user only sees their own budgets."""
self.c.login(username=self.user.username, password='******')
own_budget = BudgetFactory(user=self.user)
other_budget = BudgetFactory()
res = self.c.get('/budgets/budget')
self.assertIn(own_budget.name.encode(), res.content)
self.assertNotIn(other_budget.name.encode(), res.content)
def test_list_correct_information(self):
"""Test to ensure that the logged in user only sees their personal information."""
self.c.login(
username=self.user.username,
password='******'
)
own_budget = BudgetFactory(user=self.user)
other_budget = BudgetFactory()
res = self.c.get('/budgets/budget')
self.assertIn(own_budget.name.encode(), res.content)
self.assertNotIn(other_budget.name.encode(), res.content)
def setUp(self):
"""Create instances for testing."""
self.user = UserFactory()
self.user.set_password('secret')
self.user.save()
self.budget = BudgetFactory(user=self.user)
self.c = Client()
self.transaction = TransactionFactory(budget=self.budget)
def test_transactions_listed_in_view(self):
"""Test that budget view also lists transactions."""
self.c.login(username=self.user.username, password='******')
budget = BudgetFactory(user=self.user)
transaction = TransactionFactory(budget=budget)
res = self.c.get('/budgets/budget')
self.assertIn(transaction.description.encode(), res.content)
def test_view_list_when_logged_in(self):
"""Test the view list route when logged in."""
self.c.login(username=self.user.username, password='******')
budget = BudgetFactory(user=self.user)
res = self.c.get('/budgets/budget')
self.assertIn(budget.name.encode(), res.content)
def setUp(self):
self.user = UserFactory()
self.user.set_password('secret')
self.user.save()
self.budget = BudgetFactory(user=self.user)
self.c = Client()
def setUp(self):
"""Create instances for testing."""
self.budget = BudgetFactory(
name='test',
total_budget='1000.0',
)
def setUp(self):
"""Create the test factory."""
self.budget = BudgetFactory(
name='budget',
total_budget='200.0',
)