def test_post_runner_with_cloned_checks(self): instance = BcPlatformIntegration() instance.skip_policy_download = False instance.platform_integration_configured = True custom_policies_integration = CustomPoliciesIntegration(instance) # mock _get_policies_from_platform method custom_policies_integration._get_policies_from_platform = types.MethodType(_get_policies_from_platform, custom_policies_integration) custom_policies_integration.pre_scan() scan_reports = Report("terraform") record = Record( check_id="CKV_AWS_5", check_name="Ensure all data stored in the Elasticsearch is securely encrypted at rest", check_result={"result": CheckResult.FAILED}, code_block=[], file_path="./main.tf", file_line_range=[7, 10], resource="aws_elasticsearch_domain.enabled", evaluations=None, check_class='', file_abs_path=",.", entity_tags={"tag1": "value1"}, bc_check_id="BC_AWS_ELASTICSEARCH_3" ) scan_reports.failed_checks.append(record) custom_policies_integration.post_runner(scan_reports) self.assertEqual(2, len(scan_reports.failed_checks)) self.assertEqual('mikepolicies_cloned_AWS_1625063607541', scan_reports.failed_checks[1].check_id)
def test_integration_valid(self): instance = BcPlatformIntegration() instance.skip_policy_download = False instance.platform_integration_configured = True custom_policies_integration = CustomPoliciesIntegration(instance) self.assertTrue(custom_policies_integration.is_valid()) instance.skip_policy_download = True self.assertFalse(custom_policies_integration.is_valid()) instance.platform_integration_configured = False self.assertFalse(custom_policies_integration.is_valid()) instance.skip_policy_download = False self.assertFalse(custom_policies_integration.is_valid())
def test_pre_scan_with_cloned_checks(self): instance = BcPlatformIntegration() instance.skip_policy_download = False instance.platform_integration_configured = True custom_policies_integration = CustomPoliciesIntegration(instance) # mock _get_policies_from_platform method custom_policies_integration._get_policies_from_platform = types.MethodType(_get_policies_from_platform, custom_policies_integration) custom_policies_integration.pre_scan() self.assertEqual(1, len(custom_policies_integration.policies)) self.assertEqual(1, len(custom_policies_integration.bc_cloned_checks))