test: Set AWS_ENDPOINT_URL_DEADLINE on Worker

Signed-off-by: Charles Moore <[email protected]>
aws-deadline · Jun 6, 2024 · 14b3573 · 14b3573
1 parent 6bc4d8f
commit 14b3573
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/deadline_test_fixtures/deadline/worker.py b/src/deadline_test_fixtures/deadline/worker.py
@@ -56,6 +56,14 @@ def configure_worker_command(*, config: DeadlineWorkerConfiguration) -> str:  #
         cmds.append(
             f"runuser -l {config.user} -s /bin/bash -c 'aws configure add-model --service-model file://{config.service_model_path}'"
         )
+
+    if os.environ.get('AWS_ENDPOINT_URL_DEADLINE'):
+        LOG.info(f"Using AWS_ENDPOINT_URL_DEADLINE: {os.environ.get('AWS_ENDPOINT_URL_DEADLINE')}")
+        env = os.environ.get('AWS_ENDPOINT_URL_DEADLINE')
+        cmds.insert(
+            0,
+            f"runuser -l {config.user} -s /bin/bash -c 'echo export AWS_ENDPOINT_URL_DEADLINE={env} >> ~/.bashrc'"
+        )
 
     return " && ".join(cmds)