fix test

aiidateam · Dec 5, 2023 · 9a0ea2e · 9a0ea2e
1 parent 0b4c400
commit 9a0ea2e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/aiida_worktree/engine/worktree.py b/aiida_worktree/engine/worktree.py
@@ -251,7 +251,6 @@ def _update_process_status(self) -> None:
 
     @override
     def run(self) -> t.Any:
-        self.report("Run: ")
         self.setup()
         return self._do_step()
 
@@ -381,7 +380,6 @@ def _on_awaitable_finished(self, awaitable: Awaitable) -> None:
             print(e)
 
     def setup(self):
-
         # track if the awaitable callback is added to the runner
         self.ctx._awaitable_actions = []
         self.ctx.new_data = dict()

diff --git a/tests/test_worktree.py b/tests/test_worktree.py
@@ -54,13 +54,13 @@ def test_pause(wt_engine):
 
 def test_reset_message(wt_calcjob):
     """Modify a node and save the worktree.
-    This will add a message to the worktree_msg extra field."""
+    This will add a message to the worktree_queue extra field."""
     wt = wt_calcjob
     wt.submit()
     wt = WorkTree.load(wt.process.pk)
     wt.nodes["add2"].set({"y": orm.Int(10).store()})
     wt.save()
-    msgs = wt.process.base.extras.get("worktree_msg", [])
+    msgs = wt.process.base.extras.get("worktree_queue", [])
     assert len(msgs) == 1