fix: add job_run_as_user to integ test

aws-deadline · Oct 13, 2023 · 6bf4dd6 · 6bf4dd6
1 parent d8ac9f4
commit 6bf4dd6
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/test/integ/test_job_submissions.py b/test/integ/test_job_submissions.py
@@ -28,6 +28,10 @@
     QueueFleetAssociation,
     TaskStatus,
 )
+from deadline_test_fixtures.models import (
+    JobRunAsUser,
+    PosixSessionUser,
+)
 
 LOG = logging.getLogger(__name__)
 
@@ -80,14 +84,12 @@ def queue_with_job_run_as_user(
         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=PosixSessionUser(
+                user=job_run_as_user.user,
+                group=job_run_as_user.group,
+            ),
+        ),
     )
 
     qfa = QueueFleetAssociation.create(