def test_get_todos(self): # if can get todo todo = Todo.create(title='test title') todo.save() self.assertEqual(todo.id, Todo.get_todos(todo.id).id) # if can not get any todos 11111 is dummyid which dose not exist self.assertEqual(None, Todo.get_todos(11111)) Todo.delete() # get all todos Todo.create(title='test title').save() Todo.create(title='test title').save() Todo.create(title='test title').save() todos = Todo.get_todos() self.assertEqual(Todo.objects.all().count(), todos.count())
def _delete(self, todo_ids): todos = Todo.get_todos(todo_ids) if todos: todos.delete()
def _update(self, data): todo = Todo.get_todos(data.get('id')) data.update({'completed': json.loads(data.get('completed'))}) form = TodoForm(data, instance=todo) if form.is_valid(): form.save()