update MockWFI and ignore archive_workflow to fix unit tests

lanl · Oct 22, 2024 · 04c253f · 04c253f
1 parent 90dce1e
commit 04c253f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/beeflow/tests/mocks.py b/beeflow/tests/mocks.py
@@ -105,6 +105,10 @@ def execute_workflow(self):
         """Fake executing a workflow."""
         pass # noqa 
 
+    def export_graphml(self):
+        """Fake exporting a graphml of a workflow."""
+        pass
+
 
 class MockGDBDriver:
     """A mock GDB driver.

diff --git a/beeflow/tests/test_wf_manager.py b/beeflow/tests/test_wf_manager.py
@@ -190,6 +190,7 @@ def test_cancel_workflow(client, mocker, setup_teardown_workflow, temp_db):
                  return_value=MockWFI())
     mocker.patch('beeflow.wf_manager.resources.wf_utils.get_db_path', temp_db.db_file)
     mocker.patch('beeflow.wf_manager.resources.wf_actions.db_path', temp_db.db_file)
+    mocker.patch('beeflow.wf_manager.resources.wf_actions.archive_workflow', return_value=None)
 
     wf_name = 'wf'
     workdir = 'dir'