Refactor saving and caching tests, treatment of None IDs.

[plomtask] / tests / todos.py

diff --git a/tests/todos.py b/tests/todos.py
index 56aaf48ccc1faa917b23fd89443678efa049c396..7632f39cf3b75674036452b609c7e90b3f379e34 100644 (file)
--- a/tests/todos.py
+++ b/tests/todos.py
@@ -206,6 +206,7 @@ class TestsWithDB(TestCaseWithDB, TestCaseSansDB):
         """Test removal."""
         todo_1 = Todo(None, self.proc, False, self.date1)
         todo_1.save(self.db_conn)
+        assert todo_1.id_ is not None
         todo_0 = Todo(None, self.proc, False, self.date1)
         todo_0.save(self.db_conn)
         todo_0.add_child(todo_1)
@@ -233,6 +234,7 @@ class TestsWithDB(TestCaseWithDB, TestCaseSansDB):
         todo_1.comment = 'foo'
         todo_1.effort = -0.1
         todo_1.save(self.db_conn)
+        assert todo_1.id_ is not None
         Todo.by_id(self.db_conn, todo_1.id_)
         todo_1.comment = ''
         todo_1_id = todo_1.id_