fix: add nested policy integration type prediction

opengovern · Jan 22, 2025 · 9c26ed0 · 9c26ed0
1 parent ac82b95
commit 9c26ed0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/jobs/post-install-job/job/migrations/compliance/git_parser.go b/jobs/post-install-job/job/migrations/compliance/git_parser.go
@@ -730,6 +730,18 @@ func (g *GitParser) HandleSingleFramework(framework Framework) error {
 		for _, it := range c.IntegrationType {
 			integrationTypes[it] = true
 		}
+		if c.Policy != nil {
+			for _, it := range c.Policy.IntegrationType {
+				integrationTypes[it] = true
+			}
+		}
+		if c.PolicyID != nil {
+			if policy, ok := g.controlsPolicies[*c.PolicyID]; ok {
+				for _, it := range policy.IntegrationType {
+					integrationTypes[it] = true
+				}
+			}
+		}
 	}
 	var integrationTypesList []string
 	for k, _ := range integrationTypes {