From f3b697c9f98ad244b3c3d8a6e14a54090d3d7ad4 Mon Sep 17 00:00:00 2001 From: YaqinLi Date: Wed, 13 Dec 2023 22:55:19 +0000 Subject: [PATCH 1/2] set envirornment variable as PRODUCTION if the stage type is DEFAULT --- deploy-agent/deployd/common/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy-agent/deployd/common/config.py b/deploy-agent/deployd/common/config.py index ae406cd6d0..0257463fa8 100644 --- a/deploy-agent/deployd/common/config.py +++ b/deploy-agent/deployd/common/config.py @@ -105,7 +105,7 @@ def update_variables(self, deploy_status): if deploy_status.report.stageName: self._environ['STAGE_NAME'] = deploy_status.report.stageName if deploy_status.report.stageType: - self._environ['COMPUTE_ENV_TYPE'] = deploy_status.report.stageType + self._environ['COMPUTE_ENV_TYPE'] = "PRODUCTION" if deploy_status.report.stageType == "DEFAULT" else deploy_status.report.stageType if deploy_status.first_deploy: self._environ['FIRST_DEPLOY'] = str(deploy_status.first_deploy) if deploy_status.is_docker: From 83e9dd5f4cc4bfd04152d5386713d10e048c5d24 Mon Sep 17 00:00:00 2001 From: YaqinLi Date: Wed, 13 Dec 2023 23:23:01 +0000 Subject: [PATCH 2/2] update test --- deploy-agent/tests/unit/deploy/common/test_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy-agent/tests/unit/deploy/common/test_config.py b/deploy-agent/tests/unit/deploy/common/test_config.py index 809c58acc7..826fc2b8d7 100644 --- a/deploy-agent/tests/unit/deploy/common/test_config.py +++ b/deploy-agent/tests/unit/deploy/common/test_config.py @@ -55,7 +55,7 @@ def test_get_target(self): self.assertEqual(os.environ['DEPLOY_ID'], '123') self.assertEqual(os.environ['ENV_NAME'], 'pinboard') self.assertEqual(os.environ['STAGE_NAME'], 'beta') - self.assertEqual(os.environ['COMPUTE_ENV_TYPE'], 'DEFAULT') + self.assertEqual(os.environ['COMPUTE_ENV_TYPE'], 'PRODUCTION') self.assertEqual(self.config.get_target(), '/tmp/pinboard') def test_init(self):