add unit test for stage type

pinterest · Nov 17, 2023 · 45941a2 · 45941a2
1 parent 2dcc254
commit 45941a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deploy-agent/tests/unit/deploy/common/test_config.py b/deploy-agent/tests/unit/deploy/common/test_config.py
@@ -46,6 +46,7 @@ def test_get_target(self):
         deploy_goal['deployId'] = '123'
         deploy_goal['stageName'] = 'beta'
         deploy_goal['envName'] = 'pinboard'
+        deploy_goal['stageType'] = 'DEFAULT'
         deploy_goal['deployStage'] = DeployStage.SERVING_BUILD
         ping_response = {'deployGoal': deploy_goal, 'opCode': OpCode.NOOP}
 
@@ -54,6 +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_STAGE_TYPE'], 'DEFAULT')
         self.assertEqual(self.config.get_target(), '/tmp/pinboard')