fixing tests

shared/workspaces/test/test_workflow_service.py

@@ -189,8 +189,8 @@ class TestWorkflowService:
         mock_workflow_service._submit_dag = MagicMock()
 
         # Test DAG submission
-        mock_workflow_service._execute_prepared(fake_folder)
-        mock_workflow_service._submit_dag.assert_called_with(fake_folder, fake_dagfile)
+        mock_workflow_service._execute_prepared(fake_folder, False)
+        mock_workflow_service._submit_dag.assert_called_with(fake_folder, fake_dagfile, False)
 
         # DAG cleanup
         mock_workflow_service._submit_dag = orig_submit_dag
@@ -204,8 +204,8 @@ class TestWorkflowService:
         mock_workflow_service._submit_job = MagicMock()
 
         # Test job submission
-        mock_workflow_service._execute_prepared(fake_folder)
-        mock_workflow_service._submit_job.assert_called_with(fake_folder, fake_jobfile)
+        mock_workflow_service._execute_prepared(fake_folder, False)
+        mock_workflow_service._submit_job.assert_called_with(fake_folder, fake_jobfile, False)
 
         # Cleanup
         mock_workflow_service._submit_job = orig_submit_job
@@ -216,11 +216,11 @@ class TestWorkflowService:
         fake_jobfile = pathlib.Path("fake.submit")
         fake_jobfile.touch()
         with patch("workspaces.workflow.services.workflow_service.subprocess.run") as mock_sp_run:
-            log_path = mock_workflow_service._submit_job(fake_folder, fake_jobfile)
+            log_path = mock_workflow_service._submit_job(fake_folder, fake_jobfile, False)
             mock_sp_run.assert_called_with(
                 ["condor_submit", fake_jobfile.name],
                 cwd=str(fake_folder.absolute()),
-                preexec_fn=mock_workflow_service._switch_to_submituser,
+                preexec_fn=mock_workflow_service._switch_to_submituser(False),
             )
             assert log_path.name == "fake.log"
 
@@ -230,11 +230,11 @@ class TestWorkflowService:
         fake_folder = pathlib.Path("fake_folder")
         fake_dagfile = pathlib.Path("fake.dag")
         with patch("workspaces.workflow.services.workflow_service.subprocess.run") as mock_sp_run:
-            log_path = mock_workflow_service._submit_dag(fake_folder, fake_dagfile)
+            log_path = mock_workflow_service._submit_dag(fake_folder, fake_dagfile, False)
             mock_sp_run.assert_called_with(
                 ["condor_submit_dag", "-f", fake_dagfile.name],
                 cwd=str(fake_folder.absolute()),
-                preexec_fn=mock_workflow_service._switch_to_submituser,
+                preexec_fn=mock_workflow_service._switch_to_submituser(False),
             )
             assert log_path.name == "fake.dag.dagman.log"