Tear down in custom policies integration test

bridgecrewio · Aug 18, 2024 · 013f7e8 · 013f7e8
1 parent 0a6a599
commit 013f7e8
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/common/integration_features/test_custom_policies_integration.py b/tests/common/integration_features/test_custom_policies_integration.py
@@ -7,7 +7,8 @@
     CustomPoliciesIntegration
 from checkov.common.bridgecrew.platform_integration import BcPlatformIntegration
 from checkov.common.checks_infra.checks_parser import GraphCheckParser
-from checkov.common.checks_infra.registry import Registry, get_all_graph_checks_registries, get_graph_checks_registry
+from checkov.common.checks_infra.registry import Registry, get_all_graph_checks_registries, get_graph_checks_registry, \
+    GraphSupportedIACFrameworks
 from checkov.common.models.enums import CheckResult
 from checkov.common.output.record import Record
 from checkov.common.output.report import Report
@@ -19,10 +20,8 @@
 
 class TestCustomPoliciesIntegration(unittest.TestCase):
     def tearDown(self) -> None:
-        get_graph_checks_registry("cloudformation").checks = []
-        get_graph_checks_registry("terraform").checks = []
-        get_graph_checks_registry("kubernetes").checks = []
-        get_graph_checks_registry("bicep").checks = []
+        for framework in GraphSupportedIACFrameworks:
+            get_graph_checks_registry(framework.value.lower()).checks = []
 
     def test_integration_valid(self):
         instance = BcPlatformIntegration()