mirror of
https://github.com/LukeHagar/libopenapi.git
synced 2025-12-06 04:20:11 +00:00
c3cf5f1e38
Also ran `gofmt` across the entire project. Things need cleaning up. Signed-off-by: Dave Shanley <dave@quobix.com>
131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
// Copyright 2022 Princess B33f Heavy Industries / Dave Shanley
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package high
|
|
|
|
import (
|
|
"github.com/pb33f/libopenapi/datamodel/low"
|
|
"github.com/stretchr/testify/assert"
|
|
"gopkg.in/yaml.v3"
|
|
"testing"
|
|
)
|
|
|
|
func TestExtractExtensions(t *testing.T) {
|
|
n := make(map[low.KeyReference[string]]low.ValueReference[any])
|
|
n[low.KeyReference[string]{
|
|
Value: "pb33f",
|
|
}] = low.ValueReference[any]{
|
|
Value: "new cowboy in town",
|
|
}
|
|
ext := ExtractExtensions(n)
|
|
assert.Equal(t, "new cowboy in town", ext["pb33f"])
|
|
}
|
|
|
|
type textExtension struct {
|
|
Cowboy string
|
|
Power int
|
|
}
|
|
|
|
type parent struct {
|
|
low *child
|
|
}
|
|
|
|
func (p *parent) GoLow() *child {
|
|
return p.low
|
|
}
|
|
|
|
type child struct {
|
|
Extensions map[low.KeyReference[string]]low.ValueReference[any]
|
|
}
|
|
|
|
func (c *child) GetExtensions() map[low.KeyReference[string]]low.ValueReference[any] {
|
|
return c.Extensions
|
|
}
|
|
|
|
func TestUnpackExtensions(t *testing.T) {
|
|
|
|
var resultA, resultB yaml.Node
|
|
|
|
ymlA := `
|
|
cowboy: buckaroo
|
|
power: 100`
|
|
|
|
ymlB := `
|
|
cowboy: frogman
|
|
power: 2`
|
|
|
|
err := yaml.Unmarshal([]byte(ymlA), &resultA)
|
|
assert.NoError(t, err)
|
|
err = yaml.Unmarshal([]byte(ymlB), &resultB)
|
|
assert.NoError(t, err)
|
|
|
|
n := make(map[low.KeyReference[string]]low.ValueReference[any])
|
|
n[low.KeyReference[string]{
|
|
Value: "x-rancher-a",
|
|
}] = low.ValueReference[any]{
|
|
ValueNode: resultA.Content[0],
|
|
}
|
|
|
|
n[low.KeyReference[string]{
|
|
Value: "x-rancher-b",
|
|
}] = low.ValueReference[any]{
|
|
ValueNode: resultB.Content[0],
|
|
}
|
|
|
|
c := new(child)
|
|
c.Extensions = n
|
|
|
|
p := new(parent)
|
|
p.low = c
|
|
|
|
res, err := UnpackExtensions[textExtension, *child](p)
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, res)
|
|
assert.Equal(t, "buckaroo", res["x-rancher-a"].Cowboy)
|
|
assert.Equal(t, 100, res["x-rancher-a"].Power)
|
|
assert.Equal(t, "frogman", res["x-rancher-b"].Cowboy)
|
|
assert.Equal(t, 2, res["x-rancher-b"].Power)
|
|
}
|
|
|
|
func TestUnpackExtensions_Fail(t *testing.T) {
|
|
|
|
var resultA, resultB yaml.Node
|
|
|
|
ymlA := `
|
|
cowboy: buckaroo
|
|
power: 100`
|
|
|
|
// this is incorrect types, unpacking will fail.
|
|
ymlB := `
|
|
cowboy: 0
|
|
power: hello`
|
|
|
|
err := yaml.Unmarshal([]byte(ymlA), &resultA)
|
|
assert.NoError(t, err)
|
|
err = yaml.Unmarshal([]byte(ymlB), &resultB)
|
|
assert.NoError(t, err)
|
|
|
|
n := make(map[low.KeyReference[string]]low.ValueReference[any])
|
|
n[low.KeyReference[string]{
|
|
Value: "x-rancher-a",
|
|
}] = low.ValueReference[any]{
|
|
ValueNode: resultA.Content[0],
|
|
}
|
|
|
|
n[low.KeyReference[string]{
|
|
Value: "x-rancher-b",
|
|
}] = low.ValueReference[any]{
|
|
ValueNode: resultB.Content[0],
|
|
}
|
|
|
|
c := new(child)
|
|
c.Extensions = n
|
|
|
|
p := new(parent)
|
|
p.low = c
|
|
|
|
res, er := UnpackExtensions[textExtension, *child](p)
|
|
assert.Error(t, er)
|
|
assert.Empty(t, res)
|
|
}
|