From 078253099f83baa45c5f3c3a7c5da0737ab3a4c7 Mon Sep 17 00:00:00 2001
From: Godot Bian <13778003+godobyte@users.noreply.github.com>
Date: Wed, 4 Dec 2024 23:29:23 +0000
Subject: [PATCH] fix unit test mock

---
 test/unit/sessions/actions/test_run_attachment_download.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/unit/sessions/actions/test_run_attachment_download.py b/test/unit/sessions/actions/test_run_attachment_download.py
index ced95dd3..36a8b448 100644
--- 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