test: load checks for all frameworks

Signed-off-by: Nikita Pivkin <nikita.pivkin@smartforce.io>

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)