fix unit test mock

aws-deadline · Dec 4, 2024 · 0782530 · 0782530
1 parent b6d9839
commit 0782530
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/unit/sessions/actions/test_run_attachment_download.py b/test/unit/sessions/actions/test_run_attachment_download.py
@@ -91,6 +91,7 @@ class TestStart:
     def session(
         self,
         session_id: str,
+        session_dir: str,
         job_details: JobDetails,
         job_user: SessionUser,
         job_attachment_details: JobAttachmentDetails,
@@ -102,6 +103,7 @@ def session(
         session._job_attachment_details = job_attachment_details
         session._os_user = job_user
         session.openjd_session = mock_openjd_session_cls
+        session.working_directory = session_dir
         session._queue_id = TestStart.QUEUE_ID
         session._queue._job_id = TestStart.JOB_ID
         return session