327 lines
9.5 KiB
Go
327 lines
9.5 KiB
Go
/*
|
|
Copyright 2015 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 fields
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestSplitTerms(t *testing.T) {
|
|
testcases := map[string][]string{
|
|
// Simple selectors
|
|
`a`: {`a`},
|
|
`a=avalue`: {`a=avalue`},
|
|
`a=avalue,b=bvalue`: {`a=avalue`, `b=bvalue`},
|
|
`a=avalue,b==bvalue,c!=cvalue`: {`a=avalue`, `b==bvalue`, `c!=cvalue`},
|
|
|
|
// Empty terms
|
|
``: nil,
|
|
`a=a,`: {`a=a`, ``},
|
|
`,a=a`: {``, `a=a`},
|
|
|
|
// Escaped values
|
|
`k=\,,k2=v2`: {`k=\,`, `k2=v2`}, // escaped comma in value
|
|
`k=\\,k2=v2`: {`k=\\`, `k2=v2`}, // escaped backslash, unescaped comma
|
|
`k=\\\,,k2=v2`: {`k=\\\,`, `k2=v2`}, // escaped backslash and comma
|
|
`k=\a\b\`: {`k=\a\b\`}, // non-escape sequences
|
|
`k=\`: {`k=\`}, // orphan backslash
|
|
|
|
// Multi-byte
|
|
`함=수,목=록`: {`함=수`, `목=록`},
|
|
}
|
|
|
|
for selector, expectedTerms := range testcases {
|
|
if terms := splitTerms(selector); !reflect.DeepEqual(terms, expectedTerms) {
|
|
t.Errorf("splitSelectors(`%s`): Expected\n%#v\ngot\n%#v", selector, expectedTerms, terms)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSplitTerm(t *testing.T) {
|
|
testcases := map[string]struct {
|
|
lhs string
|
|
op string
|
|
rhs string
|
|
ok bool
|
|
}{
|
|
// Simple terms
|
|
`a=value`: {lhs: `a`, op: `=`, rhs: `value`, ok: true},
|
|
`b==value`: {lhs: `b`, op: `==`, rhs: `value`, ok: true},
|
|
`c!=value`: {lhs: `c`, op: `!=`, rhs: `value`, ok: true},
|
|
|
|
// Empty or invalid terms
|
|
``: {lhs: ``, op: ``, rhs: ``, ok: false},
|
|
`a`: {lhs: ``, op: ``, rhs: ``, ok: false},
|
|
|
|
// Escaped values
|
|
`k=\,`: {lhs: `k`, op: `=`, rhs: `\,`, ok: true},
|
|
`k=\=`: {lhs: `k`, op: `=`, rhs: `\=`, ok: true},
|
|
`k=\\\a\b\=\,\`: {lhs: `k`, op: `=`, rhs: `\\\a\b\=\,\`, ok: true},
|
|
|
|
// Multi-byte
|
|
`함=수`: {lhs: `함`, op: `=`, rhs: `수`, ok: true},
|
|
}
|
|
|
|
for term, expected := range testcases {
|
|
lhs, op, rhs, ok := splitTerm(term)
|
|
if lhs != expected.lhs || op != expected.op || rhs != expected.rhs || ok != expected.ok {
|
|
t.Errorf(
|
|
"splitTerm(`%s`): Expected\n%s,%s,%s,%v\nGot\n%s,%s,%s,%v",
|
|
term,
|
|
expected.lhs, expected.op, expected.rhs, expected.ok,
|
|
lhs, op, rhs, ok,
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEscapeValue(t *testing.T) {
|
|
// map values to their normalized escaped values
|
|
testcases := map[string]string{
|
|
``: ``,
|
|
`a`: `a`,
|
|
`=`: `\=`,
|
|
`,`: `\,`,
|
|
`\`: `\\`,
|
|
`\=\,\`: `\\\=\\\,\\`,
|
|
}
|
|
|
|
for unescapedValue, escapedValue := range testcases {
|
|
actualEscaped := EscapeValue(unescapedValue)
|
|
if actualEscaped != escapedValue {
|
|
t.Errorf("EscapeValue(%s): expected %s, got %s", unescapedValue, escapedValue, actualEscaped)
|
|
}
|
|
|
|
actualUnescaped, err := UnescapeValue(escapedValue)
|
|
if err != nil {
|
|
t.Errorf("UnescapeValue(%s): unexpected error %v", escapedValue, err)
|
|
}
|
|
if actualUnescaped != unescapedValue {
|
|
t.Errorf("UnescapeValue(%s): expected %s, got %s", escapedValue, unescapedValue, actualUnescaped)
|
|
}
|
|
}
|
|
|
|
// test invalid escape sequences
|
|
invalidTestcases := []string{
|
|
`\`, // orphan slash is invalid
|
|
`\\\`, // orphan slash is invalid
|
|
`\a`, // unrecognized escape sequence is invalid
|
|
}
|
|
for _, invalidValue := range invalidTestcases {
|
|
_, err := UnescapeValue(invalidValue)
|
|
if _, ok := err.(InvalidEscapeSequence); !ok || err == nil {
|
|
t.Errorf("UnescapeValue(%s): expected invalid escape sequence error, got %#v", invalidValue, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSelectorParse(t *testing.T) {
|
|
testGoodStrings := []string{
|
|
"x=a,y=b,z=c",
|
|
"",
|
|
"x!=a,y=b",
|
|
`x=a||y\=b`,
|
|
`x=a\=\=b`,
|
|
}
|
|
testBadStrings := []string{
|
|
"x=a||y=b",
|
|
"x==a==b",
|
|
"x=a,b",
|
|
"x in (a)",
|
|
"x in (a,b,c)",
|
|
"x",
|
|
}
|
|
for _, test := range testGoodStrings {
|
|
lq, err := ParseSelector(test)
|
|
if err != nil {
|
|
t.Errorf("%v: error %v (%#v)\n", test, err, err)
|
|
}
|
|
if test != lq.String() {
|
|
t.Errorf("%v restring gave: %v\n", test, lq.String())
|
|
}
|
|
}
|
|
for _, test := range testBadStrings {
|
|
_, err := ParseSelector(test)
|
|
if err == nil {
|
|
t.Errorf("%v: did not get expected error\n", test)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDeterministicParse(t *testing.T) {
|
|
s1, err := ParseSelector("x=a,a=x")
|
|
s2, err2 := ParseSelector("a=x,x=a")
|
|
if err != nil || err2 != nil {
|
|
t.Errorf("Unexpected parse error")
|
|
}
|
|
if s1.String() != s2.String() {
|
|
t.Errorf("Non-deterministic parse")
|
|
}
|
|
}
|
|
|
|
func expectMatch(t *testing.T, selector string, ls Set) {
|
|
lq, err := ParseSelector(selector)
|
|
if err != nil {
|
|
t.Errorf("Unable to parse %v as a selector\n", selector)
|
|
return
|
|
}
|
|
if !lq.Matches(ls) {
|
|
t.Errorf("Wanted %s to match '%s', but it did not.\n", selector, ls)
|
|
}
|
|
}
|
|
|
|
func expectNoMatch(t *testing.T, selector string, ls Set) {
|
|
lq, err := ParseSelector(selector)
|
|
if err != nil {
|
|
t.Errorf("Unable to parse %v as a selector\n", selector)
|
|
return
|
|
}
|
|
if lq.Matches(ls) {
|
|
t.Errorf("Wanted '%s' to not match '%s', but it did.", selector, ls)
|
|
}
|
|
}
|
|
|
|
func TestEverything(t *testing.T) {
|
|
if !Everything().Matches(Set{"x": "y"}) {
|
|
t.Errorf("Nil selector didn't match")
|
|
}
|
|
if !Everything().Empty() {
|
|
t.Errorf("Everything was not empty")
|
|
}
|
|
}
|
|
|
|
func TestSelectorMatches(t *testing.T) {
|
|
expectMatch(t, "", Set{"x": "y"})
|
|
expectMatch(t, "x=y", Set{"x": "y"})
|
|
expectMatch(t, "x=y,z=w", Set{"x": "y", "z": "w"})
|
|
expectMatch(t, "x!=y,z!=w", Set{"x": "z", "z": "a"})
|
|
expectMatch(t, "notin=in", Set{"notin": "in"}) // in and notin in exactMatch
|
|
expectNoMatch(t, "x=y", Set{"x": "z"})
|
|
expectNoMatch(t, "x=y,z=w", Set{"x": "w", "z": "w"})
|
|
expectNoMatch(t, "x!=y,z!=w", Set{"x": "z", "z": "w"})
|
|
|
|
fieldset := Set{
|
|
"foo": "bar",
|
|
"baz": "blah",
|
|
"complex": `=value\,\`,
|
|
}
|
|
expectMatch(t, "foo=bar", fieldset)
|
|
expectMatch(t, "baz=blah", fieldset)
|
|
expectMatch(t, "foo=bar,baz=blah", fieldset)
|
|
expectMatch(t, `foo=bar,baz=blah,complex=\=value\\\,\\`, fieldset)
|
|
expectNoMatch(t, "foo=blah", fieldset)
|
|
expectNoMatch(t, "baz=bar", fieldset)
|
|
expectNoMatch(t, "foo=bar,foobar=bar,baz=blah", fieldset)
|
|
}
|
|
|
|
func TestOneTermEqualSelector(t *testing.T) {
|
|
if !OneTermEqualSelector("x", "y").Matches(Set{"x": "y"}) {
|
|
t.Errorf("No match when match expected.")
|
|
}
|
|
if OneTermEqualSelector("x", "y").Matches(Set{"x": "z"}) {
|
|
t.Errorf("Match when none expected.")
|
|
}
|
|
}
|
|
|
|
func expectMatchDirect(t *testing.T, selector, ls Set) {
|
|
if !SelectorFromSet(selector).Matches(ls) {
|
|
t.Errorf("Wanted %s to match '%s', but it did not.\n", selector, ls)
|
|
}
|
|
}
|
|
|
|
func expectNoMatchDirect(t *testing.T, selector, ls Set) {
|
|
if SelectorFromSet(selector).Matches(ls) {
|
|
t.Errorf("Wanted '%s' to not match '%s', but it did.", selector, ls)
|
|
}
|
|
}
|
|
|
|
func TestSetMatches(t *testing.T) {
|
|
labelset := Set{
|
|
"foo": "bar",
|
|
"baz": "blah",
|
|
}
|
|
expectMatchDirect(t, Set{}, labelset)
|
|
expectMatchDirect(t, Set{"foo": "bar"}, labelset)
|
|
expectMatchDirect(t, Set{"baz": "blah"}, labelset)
|
|
expectMatchDirect(t, Set{"foo": "bar", "baz": "blah"}, labelset)
|
|
expectNoMatchDirect(t, Set{"foo": "=blah"}, labelset)
|
|
expectNoMatchDirect(t, Set{"baz": "=bar"}, labelset)
|
|
expectNoMatchDirect(t, Set{"foo": "=bar", "foobar": "bar", "baz": "blah"}, labelset)
|
|
}
|
|
|
|
func TestNilMapIsValid(t *testing.T) {
|
|
selector := Set(nil).AsSelector()
|
|
if selector == nil {
|
|
t.Errorf("Selector for nil set should be Everything")
|
|
}
|
|
if !selector.Empty() {
|
|
t.Errorf("Selector for nil set should be Empty")
|
|
}
|
|
}
|
|
|
|
func TestSetIsEmpty(t *testing.T) {
|
|
if !(Set{}).AsSelector().Empty() {
|
|
t.Errorf("Empty set should be empty")
|
|
}
|
|
if !(andTerm(nil)).Empty() {
|
|
t.Errorf("Nil andTerm should be empty")
|
|
}
|
|
if (&hasTerm{}).Empty() {
|
|
t.Errorf("hasTerm should not be empty")
|
|
}
|
|
if (¬HasTerm{}).Empty() {
|
|
t.Errorf("notHasTerm should not be empty")
|
|
}
|
|
if !(andTerm{andTerm{}}).Empty() {
|
|
t.Errorf("Nested andTerm should be empty")
|
|
}
|
|
if (andTerm{&hasTerm{"a", "b"}}).Empty() {
|
|
t.Errorf("Nested andTerm should not be empty")
|
|
}
|
|
}
|
|
|
|
func TestRequiresExactMatch(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
S Selector
|
|
Label string
|
|
Value string
|
|
Found bool
|
|
}{
|
|
"empty set": {Set{}.AsSelector(), "test", "", false},
|
|
"empty hasTerm": {&hasTerm{}, "test", "", false},
|
|
"skipped hasTerm": {&hasTerm{"a", "b"}, "test", "", false},
|
|
"valid hasTerm": {&hasTerm{"test", "b"}, "test", "b", true},
|
|
"valid hasTerm no value": {&hasTerm{"test", ""}, "test", "", true},
|
|
"valid notHasTerm": {¬HasTerm{"test", "b"}, "test", "", false},
|
|
"valid notHasTerm no value": {¬HasTerm{"test", ""}, "test", "", false},
|
|
"nil andTerm": {andTerm(nil), "test", "", false},
|
|
"empty andTerm": {andTerm{}, "test", "", false},
|
|
"nested andTerm": {andTerm{andTerm{}}, "test", "", false},
|
|
"nested andTerm matches": {andTerm{&hasTerm{"test", "b"}}, "test", "b", true},
|
|
"andTerm with non-match": {andTerm{&hasTerm{}, &hasTerm{"test", "b"}}, "test", "b", true},
|
|
}
|
|
for k, v := range testCases {
|
|
value, found := v.S.RequiresExactMatch(v.Label)
|
|
if value != v.Value {
|
|
t.Errorf("%s: expected value %s, got %s", k, v.Value, value)
|
|
}
|
|
if found != v.Found {
|
|
t.Errorf("%s: expected found %t, got %t", k, v.Found, found)
|
|
}
|
|
}
|
|
}
|