fix(misconf): init frameworks before updating them

Signed-off-by: nikpivkin <[email protected]>
aquasecurity · Aug 23, 2024 · 2ac74a7 · 2ac74a7
1 parent 6fe6727
commit 2ac74a7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pkg/iac/rego/metadata.go b/pkg/iac/rego/metadata.go
@@ -54,6 +54,9 @@ func NewStaticMetadata(pkgPath string, inputOpt InputOptions) *StaticMetadata {
 }
 
 func (sm *StaticMetadata) Update(meta map[string]any) error {
+	if sm.Frameworks == nil {
+		sm.Frameworks = make(map[framework.Framework][]string)
+	}
 
 	upd := func(field *string, key string) {
 		if raw, ok := meta[key]; ok {

diff --git a/pkg/iac/rego/metadata_test.go b/pkg/iac/rego/metadata_test.go
@@ -135,6 +135,14 @@ func Test_UpdateStaticMetadata(t *testing.T) {
 
 		assert.Equal(t, expected, sm)
 	})
+
+	t.Run("frameworks is not initialized", func(t *testing.T) {
+		sm := StaticMetadata{}
+		err := sm.Update(map[string]any{
+			"frameworks": map[string]any{"all": []any{"a", "b", "c"}},
+		})
+		require.NoError(t, err)
+	})
 }
 
 func Test_NewEngineMetadata(t *testing.T) {