fix test better

Signed-off-by: Paul S. Schweigert <[email protected]>
Qiskit · Sep 25, 2024 · baf5226 · baf5226
1 parent 19f19e2
commit baf5226
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/gateway/api/utils.py b/gateway/api/utils.py
@@ -123,10 +123,13 @@ def build_env_variables(token, job: Job, args: str = None) -> Dict[str, str]:
         env variables dict
     """
     extra = {}
-    arguments = "{}"
     # only set arguments if size is <1MB
-    if args and sys.getsizeof(args) < 1000000:
-        arguments = args
+    arguments = "{}"
+    if args:
+        if sys.getsizeof(args) < 1000000:
+            arguments = args
+        else:
+            logger.warn("arguments for job [%s] are > 1MB and will not be written to env var", job.id)
 
     if settings.SETTINGS_AUTH_MECHANISM != "default":
         extra = {

diff --git a/gateway/tests/api/test_utils.py b/gateway/tests/api/test_utils.py
@@ -31,6 +31,7 @@ def test_build_env_for_job(self):
                 "ENV_JOB_GATEWAY_TOKEN": "42",
                 "ENV_JOB_GATEWAY_HOST": "http://localhost:8000",
                 "ENV_JOB_ID_GATEWAY": "42",
+                "ENV_JOB_ARGUMENTS": "{}",
             },
         )