Working version of Configuration Governance examples test

Author: Mike Kistler

examples/test_configuration_governance_v1_examples.py

@@ -74,6 +74,8 @@ def setup_class(cls):
             enterprise_scope_id = config['ENTERPRISE_SCOPE_ID']
             subacct_scope_id = config['SUBACCT_SCOPE_ID']
 
+            cls.clean_rules()
+
         print('Setup complete.')
 
     needscredentials = pytest.mark.skipif(
@@ -139,16 +141,16 @@ def test_create_attachments_example(self):
         try:
             # begin-create_attachments
 
-            rule_scope_model = {
-                'note': 'My enterprise',
-                'scope_id': enterprise_scope_id,
-                'scope_type': 'enterprise'
+            excluded_scope_model = {
+                'note': 'Development account',
+                'scope_id': subacct_scope_id,
+                'scope_type': 'enterprise.account'
             }
 
             attachment_request_model = {
                 'account_id': account_id,
                 'included_scope': {'note':'My enterprise','scope_id':enterprise_scope_id,'scope_type':'enterprise'},
-                'excluded_scopes': [rule_scope_model]
+                'excluded_scopes': [excluded_scope_model]
             }
 
             create_attachments_response = configuration_governance_service.create_attachments(
@@ -183,7 +185,8 @@ def test_get_attachment_example(self):
             # end-get_attachment
 
             global attachment_etag_link
-            attachment_etag_link = attachment[''];
+            attachment_etag_link = configuration_governance_service.get_attachment(
+                rule_id=rule_id_link, attachment_id=attachment_id_link).get_headers().get('Etag')
         except ApiException as e:
             pytest.fail(str(e))
 
@@ -204,7 +207,7 @@ def test_get_rule_example(self):
             # end-get_rule
 
             global rule_etag_link
-            rule_etag_link = rule[''];
+            rule_etag_link = configuration_governance_service.get_rule(rule_id=rule_id_link).get_headers().get('etag')
         except ApiException as e:
             pytest.fail(str(e))
 
@@ -257,7 +260,7 @@ def test_update_rule_example(self):
 
             rule = configuration_governance_service.update_rule(
                 rule_id=rule_id_link,
-                if_match='testString',
+                if_match=rule_etag_link,
                 name='Disable public access',
                 description='Ensure that public access to account resources is disabled.',
                 target={'service_name':service_name,'resource_kind':'service','additional_target_attributes':[]},
@@ -303,19 +306,19 @@ def test_update_attachment_example(self):
         try:
             # begin-update_attachment
 
-            rule_scope_model = {
-                'note': 'My enterprise',
-                'scope_id': enterprise_scope_id,
-                'scope_type': 'enterprise'
+            excluded_scope_model = {
+                'note': 'Development account',
+                'scope_id': subacct_scope_id,
+                'scope_type': 'enterprise.account'
             }
 
             attachment = configuration_governance_service.update_attachment(
                 rule_id=rule_id_link,
                 attachment_id=attachment_id_link,
-                if_match='testString',
+                if_match=attachment_etag_link,
                 account_id=account_id,
                 included_scope={'note':'My enterprise','scope_id':enterprise_scope_id,'scope_type':'enterprise'},
-                excluded_scopes=[rule_scope_model]
+                excluded_scopes=[excluded_scope_model]
             ).get_result()
 
             print(json.dumps(attachment, indent=2))
@@ -364,6 +367,25 @@ def test_delete_rule_example(self):
         except ApiException as e:
             pytest.fail(str(e))
 
+    @classmethod
+    def clean_rules(cls):
+        """
+        Clean up rules from prior test runs
+        """
+        try:
+            rule_list = configuration_governance_service.list_rules(
+                account_id=account_id,
+                labels=test_label
+            ).get_result()
+
+            for rule in rule_list['rules']:
+                rule_id = rule['rule_id']
+                print(f'deleting rule {rule_id}')
+                configuration_governance_service.delete_rule(rule_id)
+
+        except ApiException as e:
+            print(str(e))
+
 # endregion
 ##############################################################################
 # End of Examples for Service: ConfigurationGovernanceV1