Merge pull request #13 from neotoolkit/feat/extension

feat(extension): add hardcode extensions back

Author: sashamelentyev

extensions.go

@@ -0,0 +1,6 @@
+package openapi
+
+type Extensions struct {
+	Example string `json:"x-example,omitempty" yaml:"x-example,omitempty"`
+	Faker   string `json:"x-faker,omitempty" yaml:"x-faker,omitempty"`
+}

schema.go

@@ -2,13 +2,11 @@ package openapi
 
 import (
 	"fmt"
-
-	"github.com/goccy/go-yaml"
 )
 
 // Schema -.
 type Schema struct {
-	Extensions map[string]interface{} `json:"-" yaml:"-"`
+	Extensions
 
 	Properties Schemas     `json:"properties,omitempty" yaml:"properties,omitempty"`
 	Type       string      `json:"type,omitempty" yaml:"type,omitempty"`
@@ -20,20 +18,9 @@ type Schema struct {
 	Ref        string      `json:"$ref,omitempty" yaml:"$ref,omitempty"`
 }
 
-// GetExtensions -.
-func (s *Schema) GetExtensions() (map[string]interface{}, error) {
-	bytes, err := yaml.Marshal(s.Extensions)
-	if err != nil {
-		return nil, err
-	}
-
-	var extensions map[string]interface{}
-
-	if err := yaml.Unmarshal(bytes, &extensions); err != nil {
-		return nil, err
-	}
-
-	return extensions, nil
+// IsRef -.
+func (s Schema) IsRef() bool {
+	return s.Ref != ""
 }
 
 // Schemas -.

schema_test.go

@@ -183,18 +183,3 @@ func TestSchema_ResponseByExample(t *testing.T) {
 		})
 	}
 }
-
-func TestGetExtensions(t *testing.T) {
-	s := openapi.Schema{
-		Extensions: map[string]interface{}{
-			"x-example": "x-example",
-		},
-	}
-
-	e, err := s.GetExtensions()
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	require.IsType(t, map[string]interface{}{}, e)
-}