Regression test task failure archiving

Author: arhall0

beeflow/tests/test_wf_update.py

@@ -54,3 +54,25 @@ def test_archive_workflow(tmpdir, mocker, test_function, expected_state):
         mock_update_wf_status.assert_called_once_with("wf_id_test", expected_state)
         mock_remove_wf_dir.assert_called_once_with("wf_id_test")
         mock_log.assert_called_once_with("Removing Workflow Directory")
+
+
+@pytest.mark.parametrize(
+    "wf_state, tries_to_archive", [("Archived", False), ("Running", True)]
+)
+def test_failed_task_wf_archive(mocker, wf_state, tries_to_archive):
+    """Failed task archives workflow if not archived."""
+    task = mocker.MagicMock()
+    db = mocker.MagicMock()
+    db.workflows.get_workflow_state.return_value = wf_state
+    state_update = mocker.MagicMock()
+    state_update.job_state = "FAILED"
+    wfi = mocker.MagicMock()
+    wfi.workflow_id = "123456"
+    mock_archive_fail_workflow = mocker.patch(
+        "beeflow.wf_manager.resources.wf_update.archive_fail_workflow"
+    )
+    mock_log_info = mocker.patch("logging.Logger.info")
+    wfu = wf_update.WFUpdate()
+    wfu.handle_state_change(state_update, task, wfi, db)
+    assert tries_to_archive == len(mock_archive_fail_workflow.mock_calls)
+    assert tries_to_archive == len(mock_log_info.mock_calls)