mirror of https://github.com/k3s-io/k3s
135 lines
3.6 KiB
Go
135 lines
3.6 KiB
Go
/*
|
|
Copyright 2014 Google Inc. All rights reserved.
|
|
|
|
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 labels
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestSelectorParse(t *testing.T) {
|
|
testGoodStrings := []string{
|
|
"x=a,y=b,z=c",
|
|
"",
|
|
"x!=a,y=b",
|
|
}
|
|
testBadStrings := []string{
|
|
"x=a||y=b",
|
|
"x==a==b",
|
|
}
|
|
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")
|
|
}
|
|
}
|
|
|
|
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"})
|
|
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"})
|
|
|
|
labelset := Set{
|
|
"foo": "bar",
|
|
"baz": "blah",
|
|
}
|
|
expectMatch(t, "foo=bar", labelset)
|
|
expectMatch(t, "baz=blah", labelset)
|
|
expectMatch(t, "foo=bar,baz=blah", labelset)
|
|
expectNoMatch(t, "foo=blah", labelset)
|
|
expectNoMatch(t, "baz=bar", labelset)
|
|
expectNoMatch(t, "foo=bar,foobar=bar,baz=blah", labelset)
|
|
}
|
|
|
|
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)
|
|
}
|