mirror of
https://github.com/LukeHagar/libopenapi.git
synced 2025-12-07 04:20:14 +00:00
77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
// Copyright 2023 Princess B33f Heavy Industries / Dave Shanley
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package base
|
|
|
|
import (
|
|
"context"
|
|
"github.com/pb33f/libopenapi/datamodel/low"
|
|
lowbase "github.com/pb33f/libopenapi/datamodel/low/base"
|
|
"github.com/pb33f/libopenapi/index"
|
|
"github.com/stretchr/testify/assert"
|
|
"gopkg.in/yaml.v3"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestSchemaProxy_MarshalYAML(t *testing.T) {
|
|
const ymlComponents = `components:
|
|
schemas:
|
|
rice:
|
|
type: string
|
|
nice:
|
|
properties:
|
|
rice:
|
|
$ref: '#/components/schemas/rice'
|
|
ice:
|
|
properties:
|
|
rice:
|
|
$ref: '#/components/schemas/rice'`
|
|
|
|
idx := func() *index.SpecIndex {
|
|
var idxNode yaml.Node
|
|
err := yaml.Unmarshal([]byte(ymlComponents), &idxNode)
|
|
assert.NoError(t, err)
|
|
return index.NewSpecIndexWithConfig(&idxNode, index.CreateOpenAPIIndexConfig())
|
|
}()
|
|
|
|
const ref = "#/components/schemas/nice"
|
|
const ymlSchema = `$ref: '` + ref + `'`
|
|
var node yaml.Node
|
|
_ = yaml.Unmarshal([]byte(ymlSchema), &node)
|
|
|
|
lowProxy := new(lowbase.SchemaProxy)
|
|
err := lowProxy.Build(context.Background(), nil, node.Content[0], idx)
|
|
assert.NoError(t, err)
|
|
|
|
lowRef := low.NodeReference[*lowbase.SchemaProxy]{
|
|
Value: lowProxy,
|
|
}
|
|
|
|
sp := NewSchemaProxy(&lowRef)
|
|
|
|
origin := sp.GetReferenceOrigin()
|
|
assert.Nil(t, origin)
|
|
|
|
rend, _ := sp.Render()
|
|
assert.Equal(t, "$ref: '#/components/schemas/nice'", strings.TrimSpace(string(rend)))
|
|
|
|
}
|
|
|
|
func TestCreateSchemaProxy(t *testing.T) {
|
|
sp := CreateSchemaProxy(&Schema{Description: "iAmASchema"})
|
|
assert.Equal(t, "iAmASchema", sp.rendered.Description)
|
|
assert.False(t, sp.IsReference())
|
|
}
|
|
|
|
func TestCreateSchemaProxyRef(t *testing.T) {
|
|
sp := CreateSchemaProxyRef("#/components/schemas/MySchema")
|
|
assert.Equal(t, "#/components/schemas/MySchema", sp.GetReference())
|
|
assert.True(t, sp.IsReference())
|
|
}
|
|
|
|
func TestSchemaProxy_NoSchema_GetOrigin(t *testing.T) {
|
|
sp := &SchemaProxy{}
|
|
assert.Nil(t, sp.GetReferenceOrigin())
|
|
}
|