From cfec1399ad7a0c0d25b880d68d58a0ad1814a68b Mon Sep 17 00:00:00 2001 From: Yutong Li <52769999+YutongLi291@users.noreply.github.com> Date: Tue, 13 Aug 2024 17:57:48 -0700 Subject: [PATCH] fix: append export AWS_ENDPOINT_URL_DEADLINE at end of worker SSM commands (#141) Signed-off-by: Yutong Li <52769999+YutongLi291@users.noreply.github.com> --- src/deadline_test_fixtures/deadline/worker.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/deadline_test_fixtures/deadline/worker.py b/src/deadline_test_fixtures/deadline/worker.py index 1ad38ad..e6117ef 100644 --- a/src/deadline_test_fixtures/deadline/worker.py +++ b/src/deadline_test_fixtures/deadline/worker.py @@ -485,8 +485,7 @@ def configure_worker_command(self, *, config: DeadlineWorkerConfiguration) -> st LOG.info( f"Using AWS_ENDPOINT_URL_DEADLINE: {os.environ.get('AWS_ENDPOINT_URL_DEADLINE')}" ) - cmds.insert( - 0, + cmds.append( f"[System.Environment]::SetEnvironmentVariable('AWS_ENDPOINT_URL_DEADLINE', '{os.environ.get('AWS_ENDPOINT_URL_DEADLINE')}', [System.EnvironmentVariableTarget]::Machine); " "$env:AWS_ENDPOINT_URL_DEADLINE = [System.Environment]::GetEnvironmentVariable('AWS_ENDPOINT_URL_DEADLINE','Machine')", ) @@ -670,8 +669,7 @@ def configure_worker_command( LOG.info( f"Using AWS_ENDPOINT_URL_DEADLINE: {os.environ.get('AWS_ENDPOINT_URL_DEADLINE')}" ) - cmds.insert( - 0, + cmds.append( f"runuser -l {config.agent_user} -s /bin/bash -c 'echo export AWS_ENDPOINT_URL_DEADLINE={os.environ.get('AWS_ENDPOINT_URL_DEADLINE')} >> ~/.bashrc'", )