fixing tests

awslabs · Nov 20, 2024 · 950700d · 950700d
1 parent 728f585
commit 950700d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/servicecatalog_puppet/workflow/ssm/get_ssm_parameter_task_test.py b/servicecatalog_puppet/workflow/ssm/get_ssm_parameter_task_test.py
@@ -34,6 +34,7 @@ def test_params_for_results_display(self):
             "account_id": self.account_id,
             "region": self.region,
             "param_name": self.param_name,
+            "jmespath_location": self.jmespath_location,
         }
 
         # exercise
@@ -67,6 +68,7 @@ class GetSSMParameterByPathTaskTest(tasks_unit_tests_helper.PuppetTaskUnitTest):
     account_id = "account_id"
     path = "path"
     region = "region"
+    jmespath_location = None
 
     def setUp(self) -> None:
         from servicecatalog_puppet.workflow.ssm import get_ssm_parameter_task
@@ -78,6 +80,7 @@ def setUp(self) -> None:
             account_id=self.account_id,
             path=self.path,
             region=self.region,
+            jmespath_location=self.jmespath_location,
         )
 
         self.wire_up_mocks()
@@ -89,6 +92,7 @@ def test_params_for_results_display(self):
             "account_id": self.account_id,
             "region": self.region,
             "path": self.path,
+            "jmespath_location": self.jmespath_location,
         }
 
         # exercise