def test_count_open_todos(self): self.assertEqual(self.todolist.count_open(), 1) new_todo = Todo(description='test', todolist_id=self.todolist.id, create_user=self.user) new_todo.save() self.assertEqual(self.todolist.count_open(), 2) new_todo.close() self.assertEqual(self.todolist.count_open(), 1)
def test_count_closed_todos(self): self.assertEqual(self.todolist.count_finished(), 0) new_todo = Todo(description='test', todolist_id=self.todolist.id, create_user=self.user) new_todo.close() self.todo.close() self.assertEqual(self.todolist.count_finished(), 2) self.assertIsNotNone(new_todo.deadline) self.todo.reopen() self.assertEqual(self.todolist.count_finished(), 1) self.assertIsNone(self.todo.deadline)
class ListModelTests(TestCase): def setUp(self): self.user = User.objects.create_user('test', '*****@*****.**', 'test') self.todolist = TodoList(title='title', create_user=self.user) self.todolist.save() self.todo = Todo(description='description', todolist_id=self.todolist.id, create_user=self.user) self.todo.save() def tearDown(self): self.todo.delete() self.todolist.delete() self.user.delete() def test_count_todos(self): self.assertEqual(self.todolist.count(), 1) new_todo = Todo(description='test', todolist_id=self.todolist.id, create_user=self.user) new_todo.save() self.assertEqual(self.todolist.count(), 2) def test_count_open_todos(self): self.assertEqual(self.todolist.count_open(), 1) new_todo = Todo(description='test', todolist_id=self.todolist.id, create_user=self.user) new_todo.save() self.assertEqual(self.todolist.count_open(), 2) new_todo.close() self.assertEqual(self.todolist.count_open(), 1) def test_count_closed_todos(self): self.assertEqual(self.todolist.count_finished(), 0) new_todo = Todo(description='test', todolist_id=self.todolist.id, create_user=self.user) new_todo.close() self.todo.close() self.assertEqual(self.todolist.count_finished(), 2) self.assertIsNotNone(new_todo.deadline) self.todo.reopen() self.assertEqual(self.todolist.count_finished(), 1) self.assertIsNone(self.todo.deadline)