fix: add job_run_as_user to integ test (#57)

Signed-off-by: Jericho Tolentino <68654047+jericht@users.noreply.github.com>
aws-deadline · Oct 13, 2023 · 962808f · 962808f
1 parent d8ac9f4
commit 962808f
Showing 1 changed file with 8 additions and 14 deletions.
diff --git a/test/integ/test_job_submissions.py b/test/integ/test_job_submissions.py
@@ -23,7 +23,8 @@
     Farm,
     Fleet,
     Job,
-    PosixUser,
+    JobRunAsUser,
+    PosixSessionUser,
     Queue,
     QueueFleetAssociation,
     TaskStatus,
@@ -48,8 +49,8 @@ def fleet(deadline_resources: DeadlineResources) -> Fleet:
 
 
 @pytest.fixture(scope="session")
-def job_run_as_user() -> PosixUser:
-    return PosixUser(
+def job_run_as_user() -> PosixSessionUser:
+    return PosixSessionUser(
         user="job-run-as-user",
         group="job-run-as-user-group",
     )
@@ -58,7 +59,7 @@ def job_run_as_user() -> PosixUser:
 @pytest.fixture(scope="session")
 def worker_config(
     worker_config: DeadlineWorkerConfiguration,
-    job_run_as_user: PosixUser,
+    job_run_as_user: PosixSessionUser,
 ) -> DeadlineWorkerConfiguration:
     return dataclasses.replace(
         worker_config,
@@ -74,20 +75,13 @@ def queue_with_job_run_as_user(
     farm: Farm,
     fleet: Fleet,
     deadline_client: DeadlineClient,
-    job_run_as_user: PosixUser,
+    job_run_as_user: PosixSessionUser,
 ) -> Generator[Queue, None, None]:
     queue = Queue.create(
         client=deadline_client,
         display_name=f"Queue with jobsRunAsUser {job_run_as_user.user}",
         farm=farm,
-        raw_kwargs={
-            "jobRunAsUser": {
-                "posix": {
-                    "user": job_run_as_user.user,
-                    "group": job_run_as_user.group,
-                },
-            },
-        },
+        job_run_as_user=JobRunAsUser(posix=job_run_as_user),
     )
 
     qfa = QueueFleetAssociation.create(
@@ -144,7 +138,7 @@ def test_jobs_run_as_user(
         deadline_client: DeadlineClient,
         farm: Farm,
         queue_with_job_run_as_user: Queue,
-        job_run_as_user: PosixUser,
+        job_run_as_user: PosixSessionUser,
     ) -> None:
         # WHEN
         job = Job.submit(