ODSC-47592. Fix test_jobs_python_runtime.py

- fix test_jobs_cli.py - fix test_jobs_runs.py
oracle · Sep 20, 2023 · 5b1dd56 · 5b1dd56
1 parent 73ac062
commit 5b1dd56
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 16 deletions.
diff --git a/tests/integration/jobs/test_jobs_cli.py b/tests/integration/jobs/test_jobs_cli.py
@@ -19,13 +19,9 @@ def test_create_watch_delete_job(self):
             run,
             args=[
                 "-f",
-                os.path.join(
-                    curr_dir,
-                    "../yamls",
-                    "sample_job.yaml",
-                    "--auth",
-                    os.environ.get("OCI_IAM_TYPE", AuthType.SECURITY_TOKEN),
-                ),
+                os.path.join(curr_dir, "../yamls", "sample_job.yaml"),
+                "--auth",
+                os.environ.get("OCI_IAM_TYPE", AuthType.SECURITY_TOKEN),
             ],
         )
         assert res.exit_code == 0, res.output
@@ -43,13 +39,9 @@ def test_create_watch_delete_dataflow(self):
             run,
             args=[
                 "-f",
-                os.path.join(
-                    curr_dir,
-                    "../yamls",
-                    "sample_dataflow.yaml",
-                    "--auth",
-                    os.environ.get("OCI_IAM_TYPE", AuthType.SECURITY_TOKEN),
-                ),
+                os.path.join(curr_dir, "../yamls", "sample_dataflow.yaml"),
+                "--auth",
+                os.environ.get("OCI_IAM_TYPE", AuthType.SECURITY_TOKEN),
             ],
         )
         assert res.exit_code == 0, res.output

diff --git a/tests/integration/jobs/test_jobs_runs.py b/tests/integration/jobs/test_jobs_runs.py
@@ -117,7 +117,7 @@ def job_run_test_infra(self):
     @staticmethod
     def list_objects(uri: str) -> list:
         """Lists objects on OCI object storage."""
-        oci_os = fsspec.filesystem("oci", config=oci.config.from_file())
+        oci_os = fsspec.filesystem("oci", **default_signer())
         if uri.startswith("oci://"):
             uri = uri[len("oci://") :]
         items = oci_os.ls(uri, detail=False, refresh=True)
@@ -126,7 +126,7 @@ def list_objects(uri: str) -> list:
     @staticmethod
     def remove_objects(uri: str):
         """Removes objects from OCI object storage."""
-        oci_os = fsspec.filesystem("oci", config=oci.config.from_file())
+        oci_os = fsspec.filesystem("oci", **default_signer())
         try:
             oci_os.rm(uri, recursive=True)
         except FileNotFoundError: