From ce79d95abb4911663d02c3b363ccd59ef932020f Mon Sep 17 00:00:00 2001
From: Jatin Naik <jatin.naik@snyk.io>
Date: Fri, 18 Oct 2024 17:50:24 +0100
Subject: [PATCH] fix: openapi walker walks into additional properties

---
 internal/openapiwalker/walker.go      |  8 ++++++
 internal/openapiwalker/walker_test.go | 35 +++++++++++++++++++++++++++
 2 files changed, 43 insertions(+)

diff --git a/internal/openapiwalker/walker.go b/internal/openapiwalker/walker.go
index fefe9a23..1b7f9364 100644
--- a/internal/openapiwalker/walker.go
+++ b/internal/openapiwalker/walker.go
@@ -250,6 +250,14 @@ func ProcessRefs(data any, p RefProcessor) error {
 		if err := ProcessRefs(v.Not, p); err != nil {
 			return err
 		}
+		if err := ProcessRefs(v.AdditionalProperties, p); err != nil {
+			return err
+		}
+
+	case openapi3.AdditionalProperties:
+		if err := ProcessRefs(v.Schema, p); err != nil {
+			return err
+		}
 
 	case *openapi3.PathItem:
 		if v != nil {
diff --git a/internal/openapiwalker/walker_test.go b/internal/openapiwalker/walker_test.go
index 5b06b7fc..c94a2bad 100644
--- a/internal/openapiwalker/walker_test.go
+++ b/internal/openapiwalker/walker_test.go
@@ -290,6 +290,41 @@ func TestProcessRefs(t *testing.T) {
 			},
 		})
 	})
+
+	c.Run("processes schema refs inside additional properties", func(c *qt.C) {
+		doc := &openapi3.T{}
+		doc.Components = &openapi3.Components{
+			Schemas: openapi3.Schemas{
+				"key-1": {
+					Value: &openapi3.Schema{
+						AdditionalProperties: openapi3.AdditionalProperties{
+							Schema: &openapi3.SchemaRef{
+								Ref: "value-1",
+							},
+						},
+					},
+				},
+			},
+		}
+
+		recorder := &recordRefProcessor{}
+		err := ProcessRefs(doc, recorder)
+		c.Assert(err, qt.IsNil)
+		c.Assert(toJson(c, recorder.SchemaRefs), qt.JSONEquals, []*openapi3.SchemaRef{
+			{
+				Value: &openapi3.Schema{
+					AdditionalProperties: openapi3.AdditionalProperties{
+						Schema: &openapi3.SchemaRef{
+							Ref: "value-1",
+						},
+					},
+				},
+			},
+			{
+				Ref: "value-1",
+			},
+		})
+	})
 }
 
 func toJson(c *qt.C, obj any) []byte {