Optional second parameter when using object in swaggertype (#844)

swaggo · Nov 29, 2020 · 031a42d · 031a42d
1 parent ee29e2b
commit 031a42d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/schema.go b/schema.go
@@ -155,12 +155,12 @@ func BuildCustomSchema(types []string) (*spec.Schema, error) {
 	}
 
 	switch types[0] {
-	case "primitive":
+	case PRIMITIVE:
 		if len(types) == 1 {
 			return nil, errors.New("need primitive type after primitive")
 		}
 		return BuildCustomSchema(types[1:])
-	case "array":
+	case ARRAY:
 		if len(types) == 1 {
 			return nil, errors.New("need array item type after array")
 		}
@@ -169,9 +169,9 @@ func BuildCustomSchema(types []string) (*spec.Schema, error) {
 			return nil, err
 		}
 		return spec.ArrayProperty(schema), nil
-	case "object":
+	case OBJECT:
 		if len(types) == 1 {
-			return nil, errors.New("need object item type after object")
+			return PrimitiveSchema(types[0]), nil
 		}
 		schema, err := BuildCustomSchema(types[1:])
 		if err != nil {

diff --git a/schema_test.go b/schema_test.go
@@ -106,8 +106,8 @@ func TestBuildCustomSchema(t *testing.T) {
 	assert.Equal(t, schema.SchemaProps.Items.Schema.SchemaProps.Type, spec.StringOrArray{"string"})
 
 	schema, err = BuildCustomSchema([]string{"object"})
-	assert.Error(t, err)
-	assert.Nil(t, schema)
+	assert.NoError(t, err)
+	assert.Equal(t, schema.SchemaProps.Type, spec.StringOrArray{"object"})
 
 	schema, err = BuildCustomSchema([]string{"object", "oops"})
 	assert.Error(t, err)