adding unit tests

servicecatalog_puppet/workflow/tag_policies/do_execute_tag_policies_task.py

@@ -14,8 +14,6 @@
 
 
 class DoExecuteTagPoliciesTask(tasks.TaskWithReference):
-    puppet_account_id = luigi.Parameter()
-
     tag_policy_name = luigi.Parameter()
 
     region = luigi.Parameter()

servicecatalog_puppet/workflow/tag_policies/do_execute_tag_policies_task_test.py

@@ -0,0 +1,72 @@
+from unittest import skip
+from servicecatalog_puppet.workflow import tasks_unit_tests_helper
+
+
+class DoExecuteTagPoliciesTaskTest(tasks_unit_tests_helper.PuppetTaskUnitTest):
+    tag_policy_name = "tag_policy_name"
+    region = "region"
+    account_id = "account_id"
+    ou_name = "ou_name"
+    content = {}
+    description = "description"
+    manifest_file_path = "manifest_file_path"
+    requested_priority= 1
+
+    def setUp(self) -> None:
+        from servicecatalog_puppet.workflow.tag_policies import do_execute_tag_policies_task
+        self.module = do_execute_tag_policies_task
+
+        self.sut = self.module.DoExecuteTagPoliciesTask(
+            **self.get_common_args(),
+            tag_policy_name=self.tag_policy_name, region=self.region, account_id=self.account_id, ou_name=self.ou_name, content=self.content, description=self.description, manifest_file_path=self.manifest_file_path, requested_priority=self.requested_priority
+        )
+
+        self.wire_up_mocks()    
+
+    def test_params_for_results_display(self):
+        # setup
+        expected_result = {
+            "puppet_account_id": self.puppet_account_id,
+            "tag_policy_name": self.tag_policy_name,
+            "region": self.region,
+            "account_id": self.account_id,
+            "ou_name": self.ou_name,
+            "cache_invalidator": self.cache_invalidator,
+        }        
+
+        # exercise
+        actual_result = self.sut.params_for_results_display()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    @skip
+    def test_requires(self):
+        # setup
+        # exercise
+        actual_result = self.sut.requires()
+
+        # verify
+        raise NotImplementedError()
+
+    def test_api_calls_used(self):
+        # setup
+        expected_result = [
+            f"organizations.attach_policy_{self.region}",
+        ]        
+
+        # exercise
+        actual_result = self.sut.api_calls_used()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    @skip
+    def test_run(self):
+        # setup
+        # exercise
+        actual_result = self.sut.run()
+
+        # verify
+        raise NotImplementedError()
+

servicecatalog_puppet/workflow/workspaces/prepare_account_for_workspace_task.py

@@ -13,7 +13,6 @@
 
 class PrepareAccountForWorkspaceTask(tasks.TaskWithReference):
     account_id = luigi.Parameter()
-    puppet_account_id = luigi.Parameter()
 
     def params_for_results_display(self):
         return {

servicecatalog_puppet/workflow/workspaces/prepare_account_for_workspace_task_test.py

@@ -0,0 +1,51 @@
+from unittest import skip
+from servicecatalog_puppet.workflow import tasks_unit_tests_helper
+
+
+class PrepareAccountForWorkspaceTaskTest(tasks_unit_tests_helper.PuppetTaskUnitTest):
+    account_id = "account_id"
+
+    def setUp(self) -> None:
+        from servicecatalog_puppet.workflow.workspaces import prepare_account_for_workspace_task
+        self.module = prepare_account_for_workspace_task
+
+        self.sut = self.module.PrepareAccountForWorkspaceTask(
+            **self.get_common_args(),
+            account_id=self.account_id        
+        )
+
+        self.wire_up_mocks()    
+
+    def test_params_for_results_display(self):
+        # setup
+        expected_result = {
+            "account_id": self.account_id,
+        }        
+
+        # exercise
+        actual_result = self.sut.params_for_results_display()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    def test_api_calls_used(self):
+        # setup
+        expected_result = {
+            f"cloudformation.create_or_update_{self.account_id}": 1,
+        }        
+
+        # exercise
+        actual_result = self.sut.api_calls_used()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    @skip
+    def test_run(self):
+        # setup
+        # exercise
+        actual_result = self.sut.run()
+
+        # verify
+        raise NotImplementedError()
+

servicecatalog_puppet/workflow/workspaces/provision_dry_run_workspace_task.py

@@ -21,8 +21,6 @@ class ProvisionDryRunWorkspaceTask(tasks.TaskWithParameters):
     key = luigi.Parameter()
     version_id = luigi.Parameter()
 
-    puppet_account_id = luigi.Parameter()
-
     ssm_param_inputs = luigi.ListParameter(default=[], significant=False)
 
     launch_parameters = luigi.DictParameter(default={}, significant=False)

servicecatalog_puppet/workflow/workspaces/provision_dry_run_workspace_task_test.py

@@ -0,0 +1,54 @@
+from unittest import skip
+from servicecatalog_puppet.workflow import tasks_unit_tests_helper
+
+
+class ProvisionDryRunWorkspaceTaskTest(tasks_unit_tests_helper.PuppetTaskUnitTest):
+    workspace_name = "workspace_name"
+    region = "region"
+    account_id = "account_id"
+    bucket = "bucket"
+    key = "key"
+    version_id = "version_id"
+    ssm_param_inputs = []
+    launch_parameters = {}
+    manifest_parameters = {}
+    account_parameters = {}
+    ssm_param_outputs = []
+    execution = "execution"
+
+    def setUp(self) -> None:
+        from servicecatalog_puppet.workflow.workspaces import provision_dry_run_workspace_task
+        self.module = provision_dry_run_workspace_task
+
+        self.sut = self.module.ProvisionDryRunWorkspaceTask(
+            **self.get_common_args(),
+            workspace_name=self.workspace_name, region=self.region, account_id=self.account_id, bucket=self.bucket, key=self.key, version_id=self.version_id, ssm_param_inputs=self.ssm_param_inputs, launch_parameters=self.launch_parameters, manifest_parameters=self.manifest_parameters, account_parameters=self.account_parameters, ssm_param_outputs=self.ssm_param_outputs, execution=self.execution        
+        )
+
+        self.wire_up_mocks()    
+
+    def test_params_for_results_display(self):
+        # setup
+        expected_result = {
+            "puppet_account_id": self.puppet_account_id,
+            "workspace_name": self.workspace_name,
+            "region": self.region,
+            "account_id": self.account_id,
+            "cache_invalidator": self.cache_invalidator,
+        }        
+
+        # exercise
+        actual_result = self.sut.params_for_results_display()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    @skip
+    def test_run(self):
+        # setup
+        # exercise
+        actual_result = self.sut.run()
+
+        # verify
+        raise NotImplementedError()
+

servicecatalog_puppet/workflow/workspaces/provision_workspace_task.py

@@ -21,8 +21,6 @@ class ProvisionWorkspaceTask(tasks.TaskWithParameters):
     key = luigi.Parameter()
     version_id = luigi.Parameter()
 
-    puppet_account_id = luigi.Parameter()
-
     ssm_param_inputs = luigi.ListParameter(default=[], significant=False)
 
     launch_parameters = luigi.DictParameter(default={}, significant=False)

servicecatalog_puppet/workflow/workspaces/provision_workspace_task_test.py

@@ -0,0 +1,55 @@
+from unittest import skip
+from servicecatalog_puppet.workflow import tasks_unit_tests_helper
+
+
+class ProvisionWorkspaceTaskTest(tasks_unit_tests_helper.PuppetTaskUnitTest):
+    workspace_name = "workspace_name"
+    region = "region"
+    account_id = "account_id"
+    bucket = "bucket"
+    key = "key"
+    version_id = "version_id"
+    ssm_param_inputs = []
+    launch_parameters = {}
+    manifest_parameters = {}
+    account_parameters = {}
+    ssm_param_outputs = []
+    execution = "execution"
+    manifest_file_path = "manifest_file_path"
+
+    def setUp(self) -> None:
+        from servicecatalog_puppet.workflow.workspaces import provision_workspace_task
+        self.module = provision_workspace_task
+
+        self.sut = self.module.ProvisionWorkspaceTask(
+            **self.get_common_args(),
+            workspace_name=self.workspace_name, region=self.region, account_id=self.account_id, bucket=self.bucket, key=self.key, version_id=self.version_id, ssm_param_inputs=self.ssm_param_inputs, launch_parameters=self.launch_parameters, manifest_parameters=self.manifest_parameters, account_parameters=self.account_parameters, ssm_param_outputs=self.ssm_param_outputs, execution=self.execution, manifest_file_path=self.manifest_file_path        
+        )
+
+        self.wire_up_mocks()    
+
+    def test_params_for_results_display(self):
+        # setup
+        expected_result = {
+            "puppet_account_id": self.puppet_account_id,
+            "workspace_name": self.workspace_name,
+            "region": self.region,
+            "account_id": self.account_id,
+            "cache_invalidator": self.cache_invalidator,
+        }        
+
+        # exercise
+        actual_result = self.sut.params_for_results_display()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    @skip
+    def test_run(self):
+        # setup
+        # exercise
+        actual_result = self.sut.run()
+
+        # verify
+        raise NotImplementedError()
+

servicecatalog_puppet/workflow/workspaces/terminate_dry_run_workspace_task.py

@@ -21,8 +21,6 @@ class TerminateDryRunWorkspaceTask(tasks.TaskWithParameters):
     key = luigi.Parameter()
     version_id = luigi.Parameter()
 
-    puppet_account_id = luigi.Parameter()
-
     ssm_param_inputs = luigi.ListParameter(default=[], significant=False)
 
     launch_parameters = luigi.DictParameter(default={}, significant=False)

servicecatalog_puppet/workflow/workspaces/terminate_dry_run_workspace_task_test.py

@@ -0,0 +1,54 @@
+from unittest import skip
+from servicecatalog_puppet.workflow import tasks_unit_tests_helper
+
+
+class TerminateDryRunWorkspaceTaskTest(tasks_unit_tests_helper.PuppetTaskUnitTest):
+    workspace_name = "workspace_name"
+    region = "region"
+    account_id = "account_id"
+    bucket = "bucket"
+    key = "key"
+    version_id = "version_id"
+    ssm_param_inputs = []
+    launch_parameters = {}
+    manifest_parameters = {}
+    account_parameters = {}
+    ssm_param_outputs = []
+    execution = "execution"
+
+    def setUp(self) -> None:
+        from servicecatalog_puppet.workflow.workspaces import terminate_dry_run_workspace_task
+        self.module = terminate_dry_run_workspace_task
+
+        self.sut = self.module.TerminateDryRunWorkspaceTask(
+            **self.get_common_args(),
+            workspace_name=self.workspace_name, region=self.region, account_id=self.account_id, bucket=self.bucket, key=self.key, version_id=self.version_id, ssm_param_inputs=self.ssm_param_inputs, launch_parameters=self.launch_parameters, manifest_parameters=self.manifest_parameters, account_parameters=self.account_parameters, ssm_param_outputs=self.ssm_param_outputs, execution=self.execution        
+        )
+
+        self.wire_up_mocks()    
+
+    def test_params_for_results_display(self):
+        # setup
+        expected_result = {
+            "puppet_account_id": self.puppet_account_id,
+            "workspace_name": self.workspace_name,
+            "region": self.region,
+            "account_id": self.account_id,
+            "cache_invalidator": self.cache_invalidator,
+        }        
+
+        # exercise
+        actual_result = self.sut.params_for_results_display()
+
+        # verify
+        self.assertEqual(expected_result, actual_result)
+
+    @skip
+    def test_run(self):
+        # setup
+        # exercise
+        actual_result = self.sut.run()
+
+        # verify
+        raise NotImplementedError()
+

servicecatalog_puppet/workflow/workspaces/terminate_workspace_task.py

@@ -21,8 +21,6 @@ class TerminateWorkspaceTask(tasks.TaskWithParameters):
     key = luigi.Parameter()
     version_id = luigi.Parameter()
 
-    puppet_account_id = luigi.Parameter()
-
     ssm_param_inputs = luigi.ListParameter(default=[], significant=False)
 
     launch_parameters = luigi.DictParameter(default={}, significant=False)