CLOUDP-308422: addressing follow up comments

Author: andreaangiolillo

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

@@ -15,6 +15,7 @@
 package filter
 
 import (
+	"maps"
 	"strings"
 
 	"github.com/getkin/kin-openapi/openapi3"
@@ -55,12 +56,12 @@ func applyOnOperation(op *openapi3.Operation) {
 		}
 
 		for _, content := range response.Value.Content {
-			if v, ok := content.Extensions["x-sunset"]; ok {
-				if v != sunsetToBeDecided {
-					continue
-				}
-				delete(content.Extensions, "x-sunset")
+			if content.Extensions == nil {
+				continue
 			}
+			maps.DeleteFunc(content.Extensions, func(key string, v any) bool {
+				return v == sunsetToBeDecided
+			})
 		}
 	}
 }

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

@@ -326,6 +326,7 @@ func TestSunsetFilter_Apply(t *testing.T) {
 
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
 			f := &SunsetFilter{
 				oas: tc.initSpec,
 			}