blob: d565b323047e58a12fcbd295ac591a2c1a872efc [file] [log] [blame] [edit]
/*
Copyright 2018 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package typed_test
import (
"fmt"
"testing"
"sigs.k8s.io/structured-merge-diff/fieldpath"
"sigs.k8s.io/structured-merge-diff/typed"
"sigs.k8s.io/structured-merge-diff/value"
)
type objSetPair struct {
object typed.YAMLObject
set *fieldpath.Set
}
type fieldsetTestCase struct {
name string
rootTypeName string
schema typed.YAMLObject
pairs []objSetPair
}
var (
// Short names for readable test cases.
_NS = fieldpath.NewSet
_P = fieldpath.MakePathOrDie
_KBF = fieldpath.KeyByFields
_SV = value.StringValue
_BV = value.BooleanValue
_IV = value.IntValue
_FV = value.FloatValue
)
var fieldsetCases = []fieldsetTestCase{{
name: "simple pair",
rootTypeName: "stringPair",
schema: `types:
- name: stringPair
map:
fields:
- name: key
type:
scalar: string
- name: value
type:
namedType: __untyped_atomic_
- name: __untyped_atomic_
scalar: untyped
list:
elementType:
namedType: __untyped_atomic_
elementRelationship: atomic
map:
elementType:
namedType: __untyped_atomic_
elementRelationship: atomic
`,
pairs: []objSetPair{
{`{"key":"foo","value":1}`, _NS(_P("key"), _P("value"))},
{`{"key":"foo","value":{"a": "b"}}`, _NS(_P("key"), _P("value"))},
{`{"key":"foo","value":null}`, _NS(_P("key"), _P("value"))},
{`{"key":"foo"}`, _NS(_P("key"))},
{`{"key":"foo","value":true}`, _NS(_P("key"), _P("value"))},
},
}, {
name: "struct grab bag",
rootTypeName: "myStruct",
schema: `types:
- name: myStruct
map:
fields:
- name: numeric
type:
scalar: numeric
- name: string
type:
scalar: string
- name: bool
type:
scalar: boolean
- name: setStr
type:
list:
elementType:
scalar: string
elementRelationship: associative
- name: setBool
type:
list:
elementType:
scalar: boolean
elementRelationship: associative
- name: setNumeric
type:
list:
elementType:
scalar: numeric
elementRelationship: associative
- name: color
type:
map:
fields:
- name: R
type:
scalar: numeric
- name: G
type:
scalar: numeric
- name: B
type:
scalar: numeric
elementRelationship: atomic
- name: arbitraryWavelengthColor
type:
map:
elementType:
scalar: numeric
elementRelationship: atomic
- name: args
type:
list:
elementType:
map:
fields:
- name: key
type:
scalar: string
- name: value
type:
scalar: string
elementRelationship: atomic
`,
pairs: []objSetPair{
{`{"numeric":1}`, _NS(_P("numeric"))},
{`{"numeric":3.14159}`, _NS(_P("numeric"))},
{`{"string":"aoeu"}`, _NS(_P("string"))},
{`{"bool":true}`, _NS(_P("bool"))},
{`{"bool":false}`, _NS(_P("bool"))},
{`{"setStr":["a","b","c"]}`, _NS(
_P("setStr", _SV("a")),
_P("setStr", _SV("b")),
_P("setStr", _SV("c")),
)},
{`{"setBool":[true,false]}`, _NS(
_P("setBool", _BV(true)),
_P("setBool", _BV(false)),
)},
{`{"setNumeric":[1,2,3,3.14159]}`, _NS(
_P("setNumeric", _IV(1)),
_P("setNumeric", _IV(2)),
_P("setNumeric", _IV(3)),
_P("setNumeric", _FV(3.14159)),
)},
{`{"color":{}}`, _NS(_P("color"))},
{`{"color":null}`, _NS(_P("color"))},
{`{"color":{"R":255,"G":0,"B":0}}`, _NS(_P("color"))},
{`{"arbitraryWavelengthColor":{}}`, _NS(_P("arbitraryWavelengthColor"))},
{`{"arbitraryWavelengthColor":null}`, _NS(_P("arbitraryWavelengthColor"))},
{`{"arbitraryWavelengthColor":{"IR":255}}`, _NS(_P("arbitraryWavelengthColor"))},
{`{"args":[]}`, _NS(_P("args"))},
{`{"args":null}`, _NS(_P("args"))},
{`{"args":[null]}`, _NS(_P("args"))},
{`{"args":[{"key":"a","value":"b"},{"key":"c","value":"d"}]}`, _NS(_P("args"))},
},
}, {
name: "associative list",
rootTypeName: "myRoot",
schema: `types:
- name: myRoot
map:
fields:
- name: list
type:
namedType: myList
- name: atomicList
type:
namedType: mySequence
- name: myList
list:
elementType:
namedType: myElement
elementRelationship: associative
keys:
- key
- id
- name: mySequence
list:
elementType:
scalar: string
elementRelationship: atomic
- name: myElement
map:
fields:
- name: key
type:
scalar: string
- name: id
type:
scalar: numeric
- name: value
type:
namedType: myValue
- name: bv
type:
scalar: boolean
- name: nv
type:
scalar: numeric
- name: myValue
map:
elementType:
scalar: string
`,
pairs: []objSetPair{
{`{"list":[]}`, _NS()},
{`{"list":[{"key":"a","id":1,"value":{"a":"a"}}]}`, _NS(
_P("list", _KBF("key", _SV("a"), "id", _IV(1))),
_P("list", _KBF("key", _SV("a"), "id", _IV(1)), "key"),
_P("list", _KBF("key", _SV("a"), "id", _IV(1)), "id"),
_P("list", _KBF("key", _SV("a"), "id", _IV(1)), "value", "a"),
)},
{`{"list":[{"key":"a","id":1},{"key":"a","id":2},{"key":"b","id":1}]}`, _NS(
_P("list", _KBF("key", _SV("a"), "id", _IV(1))),
_P("list", _KBF("key", _SV("a"), "id", _IV(2))),
_P("list", _KBF("key", _SV("b"), "id", _IV(1))),
_P("list", _KBF("key", _SV("a"), "id", _IV(1)), "key"),
_P("list", _KBF("key", _SV("a"), "id", _IV(1)), "id"),
_P("list", _KBF("key", _SV("a"), "id", _IV(2)), "key"),
_P("list", _KBF("key", _SV("a"), "id", _IV(2)), "id"),
_P("list", _KBF("key", _SV("b"), "id", _IV(1)), "key"),
_P("list", _KBF("key", _SV("b"), "id", _IV(1)), "id"),
)},
{`{"atomicList":["a","a","a"]}`, _NS(_P("atomicList"))},
},
}}
func (tt fieldsetTestCase) test(t *testing.T) {
parser, err := typed.NewParser(tt.schema)
if err != nil {
t.Fatalf("failed to create schema: %v", err)
}
for i, v := range tt.pairs {
v := v
t.Run(fmt.Sprintf("%v-%v", tt.name, i), func(t *testing.T) {
t.Parallel()
tv, err := parser.Type(tt.rootTypeName).FromYAML(v.object)
if err != nil {
t.Errorf("failed to parse object: %v", err)
}
fs, err := tv.ToFieldSet()
if err != nil {
t.Fatalf("got validation errors: %v", err)
}
if !fs.Equals(v.set) {
t.Errorf("wanted\n%s\ngot\n%s\n", v.set, fs)
}
})
}
}
func TestToFieldSet(t *testing.T) {
for _, tt := range fieldsetCases {
tt := tt
t.Run(tt.name, func(t *testing.T) {
t.Parallel()
tt.test(t)
})
}
}