diff --git a/test/rego/rego_checks_test.go b/test/rego/rego_checks_test.go index 0b23a313..84ad572d 100644 --- a/test/rego/rego_checks_test.go +++ b/test/rego/rego_checks_test.go @@ -5,9 +5,11 @@ import ( "testing" checks "github.com/aquasecurity/trivy-checks" + "github.com/aquasecurity/trivy/pkg/iac/framework" "github.com/aquasecurity/trivy/pkg/iac/rego" "github.com/aquasecurity/trivy/pkg/iac/rules" "github.com/aquasecurity/trivy/pkg/iac/scan" + "github.com/aquasecurity/trivy/pkg/iac/scanners/options" "github.com/aquasecurity/trivy/pkg/iac/state" trivyTypes "github.com/aquasecurity/trivy/pkg/iac/types" ruleTypes "github.com/aquasecurity/trivy/pkg/iac/types/rules" @@ -31,7 +33,10 @@ func addTests(tc testCases) { } func TestRegoChecks(t *testing.T) { - regoScanner := rego.NewScanner(trivyTypes.SourceCloud) + regoScanner := rego.NewScanner( + trivyTypes.SourceCloud, + options.ScannerWithFrameworks(framework.CIS_AWS_1_2, framework.CIS_AWS_1_4, framework.Default), + ) err := regoScanner.LoadPolicies(true, false, checks.EmbeddedPolicyFileSystem, []string{"."}, nil) require.NoError(t, err)