Fix

truenas · Dec 12, 2024 · 5d4f8e9 · 5d4f8e9
1 parent 05ed97b
commit 5d4f8e9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/api2/test_job_logs.py b/tests/api2/test_job_logs.py
@@ -50,6 +50,8 @@ def mock(self, *args):
             return job.id
     """):
         with mock("test.test2", """
+            from middlewared.service import job
+
             @job(logs=True)
             def mock(self, job, *args):
                 job.logs_fd.write(b'Job logs')
@@ -64,12 +66,14 @@ def mock(self, job, *args):
 
 def test_job_download_logs_unprivileged_downloads_internal_logs_with_read_role(c_replication_read):
     with mock("test.test1", """
-        from middlewared.service import accepts, job
+        from middlewared.service import job
 
         @job(logs=True, read_roles=["REPLICATION_TASK_READ"])
         def mock(self, job, *args):
             job.logs_fd.write(b'Job logs')
     """):
+        c = c_replication_read
+
         jid = call("test.test1")
 
         c.call("core.job_wait", jid, job=True)