Files
libopenapi/datamodel/low/model_builder_test.go
quobix 7afb1e9d11 Addressed #159
Cleaned up all interesting static checks. All that is left is cosmetics.

Signed-off-by: quobix <dave@quobix.com>
2024-02-06 22:13:41 -05:00

424 lines
9.4 KiB
Go

package low
import (
"sync"
"testing"
"github.com/pb33f/libopenapi/orderedmap"
"github.com/pb33f/libopenapi/utils"
"github.com/stretchr/testify/assert"
"gopkg.in/yaml.v3"
)
type hotdog struct {
Name NodeReference[string]
ValueName ValueReference[string]
Fat NodeReference[int]
Ketchup NodeReference[float32]
Mustard NodeReference[float64]
Grilled NodeReference[bool]
MaxTemp NodeReference[int]
MaxTempHigh NodeReference[int64]
MaxTempAlt []NodeReference[int]
Drinks []NodeReference[string]
Sides []NodeReference[float32]
BigSides []NodeReference[float64]
Temps []NodeReference[int]
HighTemps []NodeReference[int64]
Buns []NodeReference[bool]
UnknownElements NodeReference[*yaml.Node]
LotsOfUnknowns []NodeReference[*yaml.Node]
Where *orderedmap.Map[string, NodeReference[*yaml.Node]]
There *orderedmap.Map[string, NodeReference[string]]
AllTheThings NodeReference[*orderedmap.Map[KeyReference[string], ValueReference[string]]]
}
func TestBuildModel_Mismatch(t *testing.T) {
yml := `crisps: are tasty`
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
hd := hotdog{}
cErr := BuildModel(&rootNode, &hd)
assert.NoError(t, cErr)
assert.Empty(t, hd.Name)
}
func TestBuildModel(t *testing.T) {
yml := `name: yummy
valueName: yammy
beef: true
fat: 200
ketchup: 200.45
mustard: 324938249028.98234892374892374923874823974
grilled: true
maxTemp: 250
maxTempAlt: [1,2,3,4,5]
maxTempHigh: 7392837462032342
drinks:
- nice
- rice
- spice
sides:
- 0.23
- 22.23
- 99.45
- 22311.2234
bigSides:
- 98237498.9872349872349872349872347982734927342983479234234234234234234
- 9827347234234.982374982734987234987
- 234234234.234982374982347982374982374982347
- 987234987234987234982734.987234987234987234987234987234987234987234982734982734982734987234987234987234987
temps:
- 1
- 2
highTemps:
- 827349283744710
- 11732849090192923
buns:
- true
- false
unknownElements:
well:
whoKnows: not me?
doYou:
love: beerToo?
lotsOfUnknowns:
- wow:
what: aTrip
- amazing:
french: fries
- amazing:
french: fries
where:
things:
are:
wild: out here
howMany:
bears: 200
there:
oh: yeah
care: bear
allTheThings:
beer: isGood
cake: isNice`
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
hd := hotdog{}
cErr := BuildModel(rootNode.Content[0], &hd)
assert.Equal(t, 200, hd.Fat.Value)
assert.Equal(t, 4, hd.Fat.ValueNode.Line)
assert.Equal(t, true, hd.Grilled.Value)
assert.Equal(t, "yummy", hd.Name.Value)
assert.Equal(t, "yammy", hd.ValueName.Value)
assert.Equal(t, float32(200.45), hd.Ketchup.Value)
assert.Len(t, hd.Drinks, 3)
assert.Len(t, hd.Sides, 4)
assert.Len(t, hd.BigSides, 4)
assert.Len(t, hd.Temps, 2)
assert.Len(t, hd.HighTemps, 2)
assert.Equal(t, int64(11732849090192923), hd.HighTemps[1].Value)
assert.Len(t, hd.MaxTempAlt, 5)
assert.Equal(t, int64(7392837462032342), hd.MaxTempHigh.Value)
assert.Equal(t, 2, hd.Temps[1].Value)
assert.Equal(t, 27, hd.Temps[1].ValueNode.Line)
var unknownElements map[string]any
_ = hd.UnknownElements.Value.Decode(&unknownElements)
assert.Len(t, unknownElements, 2)
assert.Len(t, hd.LotsOfUnknowns, 3)
assert.Equal(t, 2, orderedmap.Len(hd.Where))
assert.Equal(t, 2, orderedmap.Len(hd.There))
assert.Equal(t, "bear", hd.There.GetOrZero("care").Value)
assert.Equal(t, 324938249028.98234892374892374923874823974, hd.Mustard.Value)
allTheThings := hd.AllTheThings.Value
for pair := orderedmap.First(allTheThings); pair != nil; pair = pair.Next() {
if pair.Key().Value == "beer" {
assert.Equal(t, "isGood", pair.Value().Value)
}
if pair.Key().Value == "cake" {
assert.Equal(t, "isNice", pair.Value().Value)
}
}
assert.NoError(t, cErr)
}
func TestBuildModel_UseCopyNotRef(t *testing.T) {
yml := `cake: -99999`
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
hd := hotdog{}
cErr := BuildModel(&rootNode, hd)
assert.Error(t, cErr)
assert.Empty(t, hd.Name)
}
func TestBuildModel_UseUnsupportedPrimitive(t *testing.T) {
type notSupported struct {
cake string
}
ns := notSupported{}
yml := `cake: party`
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
cErr := BuildModel(rootNode.Content[0], &ns)
assert.Error(t, cErr)
assert.Empty(t, ns.cake)
}
func TestBuildModel_UsingInternalConstructs(t *testing.T) {
type internal struct {
Extensions NodeReference[string]
PathItems NodeReference[string]
Thing NodeReference[string]
}
yml := `extensions: one
pathItems: two
thing: yeah`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
// try a null build
try := BuildModel(nil, ins)
assert.NoError(t, try)
cErr := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, cErr)
assert.Empty(t, ins.PathItems.Value)
assert.Empty(t, ins.Extensions.Value)
assert.Equal(t, "yeah", ins.Thing.Value)
}
func TestSetField_MapHelperWrapped(t *testing.T) {
type internal struct {
Thing KeyReference[*orderedmap.Map[KeyReference[string], ValueReference[string]]]
}
yml := `thing:
what: not
chip: chop
lip: lop`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, try)
assert.Equal(t, 3, orderedmap.Len(ins.Thing.Value))
}
func TestSetField_MapHelper(t *testing.T) {
type internal struct {
Thing *orderedmap.Map[KeyReference[string], ValueReference[string]]
}
yml := `thing:
what: not
chip: chop
lip: lop`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, try)
assert.Equal(t, 3, orderedmap.Len(ins.Thing))
}
func TestSetField_ArrayHelper(t *testing.T) {
type internal struct {
Thing NodeReference[[]ValueReference[string]]
}
yml := `thing:
- nice
- rice
- slice`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, try)
assert.Len(t, ins.Thing.Value, 3)
}
func TestSetField_Enum_Helper(t *testing.T) {
type internal struct {
Thing NodeReference[[]ValueReference[*yaml.Node]]
}
yml := `thing:
- nice
- rice
- slice`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, try)
assert.Len(t, ins.Thing.Value, 3)
}
func TestSetField_Default_Helper(t *testing.T) {
type cake struct {
thing int
}
// this should be ignored, no custom objects in here my friend.
type internal struct {
Thing cake
}
yml := `thing:
type: cake`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, try)
assert.Equal(t, 0, ins.Thing.thing)
}
func TestHandleSlicesOfInts(t *testing.T) {
type internal struct {
Thing NodeReference[[]ValueReference[*yaml.Node]]
}
yml := `thing:
- 5
- 1.234`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
assert.NoError(t, try)
var thing0 int64
_ = ins.Thing.GetValue()[0].Value.Decode(&thing0)
var thing1 float64
_ = ins.Thing.GetValue()[1].Value.Decode(&thing1)
assert.Equal(t, int64(5), thing0)
assert.Equal(t, 1.234, thing1)
}
func TestHandleSlicesOfBools(t *testing.T) {
type internal struct {
Thing NodeReference[[]ValueReference[*yaml.Node]]
}
yml := `thing:
- true
- false`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(rootNode.Content[0], ins)
var thing0 bool
_ = ins.Thing.GetValue()[0].Value.Decode(&thing0)
var thing1 bool
_ = ins.Thing.GetValue()[1].Value.Decode(&thing1)
assert.NoError(t, try)
assert.Equal(t, true, thing0)
assert.Equal(t, false, thing1)
}
func TestSetField_Ignore(t *testing.T) {
type Complex struct {
}
type internal struct {
Thing *Complex
}
yml := `thing:
- nice
- rice
- slice`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
try := BuildModel(&rootNode, ins)
assert.NoError(t, try)
assert.Nil(t, ins.Thing)
}
func TestBuildModelAsync(t *testing.T) {
type internal struct {
Thing KeyReference[*orderedmap.Map[KeyReference[string], ValueReference[string]]]
}
yml := `thing:
what: not
chip: chop
lip: lop`
ins := new(internal)
var rootNode yaml.Node
mErr := yaml.Unmarshal([]byte(yml), &rootNode)
assert.NoError(t, mErr)
var wg sync.WaitGroup
var errors []error
wg.Add(1)
BuildModelAsync(rootNode.Content[0], ins, &wg, &errors)
wg.Wait()
assert.Equal(t, 3, orderedmap.Len(ins.Thing.Value))
}
func TestSetField_NilValueNode(t *testing.T) {
assert.NotPanics(t, func() {
SetField(nil, nil, nil)
})
}
func TestBuildModelAsync_HandlesError(t *testing.T) {
errs := []error{}
wg := sync.WaitGroup{}
wg.Add(1)
BuildModelAsync(utils.CreateStringNode("cake"), "cake", &wg, &errs)
assert.NotEmpty(t, errs)
}