fixes

Author: andreaangiolillo

tools/cli/internal/openapi/filter/bump.go

@@ -18,7 +18,7 @@ import (
 	"github.com/getkin/kin-openapi/openapi3"
 )
 
-// BumpFilter modifies includes the fields "x-state" and "x-beta" to the "preview" and "upcoming" APIs Paths.
+// BumpFilter modifies includes the fields "x-state" and "x-beta" to the "preview" and "upcoming" APIs Operations.
 // The "x-state" and "x-beta" fields are bump.sh custom fields to include budges
 // Bump.sh feature: https://docs.bump.sh/help/specification-support/doc-badges/
 type BumpFilter struct {
@@ -51,22 +51,26 @@ func (f *BumpFilter) Apply() error {
 
 func (f *BumpFilter) includeBumpFieldForUpcoming() error {
 	for _, p := range f.oas.Paths.Map() {
-		if p.Extensions == nil {
-			p.Extensions = map[string]any{}
+		for _, op := range p.Operations() {
+			if op.Extensions == nil {
+				op.Extensions = map[string]any{}
+			}
+			op.Extensions[stateFieldName] = stateFieldValueUpcoming
 		}
-		p.Extensions[stateFieldName] = stateFieldValueUpcoming
 	}
 
 	return nil
 }
 
 func (f *BumpFilter) includeBumpFieldForPreview() error {
 	for _, p := range f.oas.Paths.Map() {
-		if p.Extensions == nil {
-			p.Extensions = map[string]any{}
+		for _, op := range p.Operations() {
+			if op.Extensions == nil {
+				op.Extensions = map[string]any{}
+			}
+			op.Extensions[stateFieldName] = stateFieldValuePreview
+			op.Extensions[betaFieldName] = true
 		}
-		p.Extensions[stateFieldName] = stateFieldValuePreview
-		p.Extensions[betaFieldName] = true
 	}
 	return nil
 }

tools/cli/internal/openapi/filter/bump_test.go

@@ -63,10 +63,13 @@ func TestBumpFilter_Apply_Preview(t *testing.T) {
 	assert.Contains(t, oas.Paths.Map(), "test")
 
 	testPath := oas.Paths.Map()["test"]
-	assert.Contains(t, testPath.Extensions, "x-state")
-	assert.Equal(t, "Preview", testPath.Extensions["x-state"])
-	assert.Contains(t, testPath.Extensions, "x-beta")
-	assert.Equal(t, true, testPath.Extensions["x-beta"])
+	assert.NotNil(t, testPath.Get)
+
+	op := testPath.Get
+	assert.Contains(t, op.Extensions, "x-state")
+	assert.Equal(t, "Preview", op.Extensions["x-state"])
+	assert.Contains(t, op.Extensions, "x-beta")
+	assert.Equal(t, true, op.Extensions["x-beta"])
 }
 
 func TestBumpFilter_Apply_Upcoming(t *testing.T) {
@@ -106,9 +109,12 @@ func TestBumpFilter_Apply_Upcoming(t *testing.T) {
 	assert.Contains(t, oas.Paths.Map(), "test")
 
 	testPath := oas.Paths.Map()["test"]
-	assert.Contains(t, testPath.Extensions, "x-state")
-	assert.Equal(t, "Upcoming", testPath.Extensions["x-state"])
-	assert.NotContains(t, testPath.Extensions, "x-beta")
+	assert.NotNil(t, testPath.Get)
+	op := testPath.Get
+
+	assert.Contains(t, op.Extensions, "x-state")
+	assert.Equal(t, "Upcoming", op.Extensions["x-state"])
+	assert.NotContains(t, op.Extensions, "x-beta")
 }
 
 func TestBumpFilter_Apply_Stable(t *testing.T) {
@@ -148,6 +154,9 @@ func TestBumpFilter_Apply_Stable(t *testing.T) {
 	assert.Contains(t, oas.Paths.Map(), "test")
 
 	testPath := oas.Paths.Map()["test"]
-	assert.NotContains(t, testPath.Extensions, "x-state")
-	assert.NotContains(t, testPath.Extensions, "x-beta")
+	assert.NotNil(t, testPath.Get)
+	op := testPath.Get
+
+	assert.NotContains(t, op.Extensions, "x-state")
+	assert.NotContains(t, op.Extensions, "x-beta")
 }