|
| 1 | +package codegen |
| 2 | + |
| 3 | +import ( |
| 4 | + "testing" |
| 5 | + |
| 6 | + "github.com/stretchr/testify/assert" |
| 7 | +) |
| 8 | + |
| 9 | +func TestFormatMapping_Resolve(t *testing.T) { |
| 10 | + fm := FormatMapping{ |
| 11 | + Default: SimpleTypeSpec{Type: "int"}, |
| 12 | + Formats: map[string]SimpleTypeSpec{ |
| 13 | + "int32": {Type: "int32"}, |
| 14 | + "int64": {Type: "int64"}, |
| 15 | + }, |
| 16 | + } |
| 17 | + |
| 18 | + assert.Equal(t, "int", fm.Resolve("").Type) |
| 19 | + assert.Equal(t, "int32", fm.Resolve("int32").Type) |
| 20 | + assert.Equal(t, "int64", fm.Resolve("int64").Type) |
| 21 | + assert.Equal(t, "int", fm.Resolve("unknown-format").Type) |
| 22 | +} |
| 23 | + |
| 24 | +func TestTypeMapping_Merge(t *testing.T) { |
| 25 | + base := DefaultTypeMapping |
| 26 | + |
| 27 | + user := TypeMapping{ |
| 28 | + Integer: FormatMapping{ |
| 29 | + Default: SimpleTypeSpec{Type: "int64"}, |
| 30 | + }, |
| 31 | + String: FormatMapping{ |
| 32 | + Formats: map[string]SimpleTypeSpec{ |
| 33 | + "date-time": {Type: "civil.DateTime", Import: "cloud.google.com/go/civil"}, |
| 34 | + }, |
| 35 | + }, |
| 36 | + } |
| 37 | + |
| 38 | + merged := base.Merge(user) |
| 39 | + |
| 40 | + // Integer default overridden |
| 41 | + assert.Equal(t, "int64", merged.Integer.Default.Type) |
| 42 | + // Integer formats still inherited from base |
| 43 | + assert.Equal(t, "int32", merged.Integer.Formats["int32"].Type) |
| 44 | + |
| 45 | + // String date-time overridden |
| 46 | + assert.Equal(t, "civil.DateTime", merged.String.Formats["date-time"].Type) |
| 47 | + assert.Equal(t, "cloud.google.com/go/civil", merged.String.Formats["date-time"].Import) |
| 48 | + // String default still inherited from base |
| 49 | + assert.Equal(t, "string", merged.String.Default.Type) |
| 50 | + // Other string formats still inherited |
| 51 | + assert.Equal(t, "openapi_types.UUID", merged.String.Formats["uuid"].Type) |
| 52 | + |
| 53 | + // Number and Boolean unchanged |
| 54 | + assert.Equal(t, "float32", merged.Number.Default.Type) |
| 55 | + assert.Equal(t, "bool", merged.Boolean.Default.Type) |
| 56 | +} |
| 57 | + |
| 58 | +func TestDefaultTypeMapping_Completeness(t *testing.T) { |
| 59 | + // Verify all the default mappings match what was previously hardcoded |
| 60 | + dm := DefaultTypeMapping |
| 61 | + |
| 62 | + // Integer |
| 63 | + assert.Equal(t, "int", dm.Integer.Resolve("").Type) |
| 64 | + assert.Equal(t, "int32", dm.Integer.Resolve("int32").Type) |
| 65 | + assert.Equal(t, "int64", dm.Integer.Resolve("int64").Type) |
| 66 | + assert.Equal(t, "uint32", dm.Integer.Resolve("uint32").Type) |
| 67 | + assert.Equal(t, "int", dm.Integer.Resolve("unknown").Type) |
| 68 | + |
| 69 | + // Number |
| 70 | + assert.Equal(t, "float32", dm.Number.Resolve("").Type) |
| 71 | + assert.Equal(t, "float32", dm.Number.Resolve("float").Type) |
| 72 | + assert.Equal(t, "float64", dm.Number.Resolve("double").Type) |
| 73 | + assert.Equal(t, "float32", dm.Number.Resolve("unknown").Type) |
| 74 | + |
| 75 | + // Boolean |
| 76 | + assert.Equal(t, "bool", dm.Boolean.Resolve("").Type) |
| 77 | + |
| 78 | + // String |
| 79 | + assert.Equal(t, "string", dm.String.Resolve("").Type) |
| 80 | + assert.Equal(t, "[]byte", dm.String.Resolve("byte").Type) |
| 81 | + assert.Equal(t, "openapi_types.Email", dm.String.Resolve("email").Type) |
| 82 | + assert.Equal(t, "openapi_types.Date", dm.String.Resolve("date").Type) |
| 83 | + assert.Equal(t, "time.Time", dm.String.Resolve("date-time").Type) |
| 84 | + assert.Equal(t, "json.RawMessage", dm.String.Resolve("json").Type) |
| 85 | + assert.Equal(t, "openapi_types.UUID", dm.String.Resolve("uuid").Type) |
| 86 | + assert.Equal(t, "openapi_types.File", dm.String.Resolve("binary").Type) |
| 87 | + assert.Equal(t, "string", dm.String.Resolve("unknown").Type) |
| 88 | +} |
0 commit comments