add VersionToFloat utility function

Author: k1LoW

helpers/version.go

@@ -0,0 +1,15 @@
+package helpers
+
+import (
+	"strings"
+)
+
+// VersionToFloat converts a version string to a float32 for easier comparison.
+func VersionToFloat(version string) float32 {
+	switch {
+	case strings.HasPrefix(version, "3.0"):
+		return 3.0
+	default:
+		return 3.1
+	}
+}

helpers/version_test.go

@@ -0,0 +1,63 @@
+package helpers
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestVersionToFloat(t *testing.T) {
+	tests := []struct {
+		name     string
+		version  string
+		expected float32
+	}{
+		{
+			name:     "OpenAPI 3.0",
+			version:  "3.0",
+			expected: 3.0,
+		},
+		{
+			name:     "OpenAPI 3.0.0",
+			version:  "3.0.0",
+			expected: 3.0,
+		},
+		{
+			name:     "OpenAPI 3.0.3",
+			version:  "3.0.3",
+			expected: 3.0,
+		},
+		{
+			name:     "OpenAPI 3.1",
+			version:  "3.1",
+			expected: 3.1,
+		},
+		{
+			name:     "OpenAPI 3.1.0",
+			version:  "3.1.0",
+			expected: 3.1,
+		},
+		{
+			name:     "OpenAPI 3.1.1",
+			version:  "3.1.1",
+			expected: 3.1,
+		},
+		{
+			name:     "default to 3.1 for unknown version",
+			version:  "4.0",
+			expected: 3.1,
+		},
+		{
+			name:     "default to 3.1 for empty string",
+			version:  "",
+			expected: 3.1,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := VersionToFloat(tt.version)
+			assert.Equal(t, tt.expected, result)
+		})
+	}
+}