mirror of https://github.com/k3s-io/k3s
120 lines
2.6 KiB
Go
120 lines
2.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 util
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestStringSet(t *testing.T) {
|
|
s := StringSet{}
|
|
s2 := StringSet{}
|
|
if len(s) != 0 {
|
|
t.Errorf("Expected len=0: %d", len(s))
|
|
}
|
|
s.Insert("a", "b")
|
|
if len(s) != 2 {
|
|
t.Errorf("Expected len=2: %d", len(s))
|
|
}
|
|
s.Insert("c")
|
|
if s.Has("d") {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
if !s.Has("a") {
|
|
t.Errorf("Missing contents: %#v", s)
|
|
}
|
|
s.Delete("a")
|
|
if s.Has("a") {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
s.Insert("a")
|
|
if s.HasAll("a", "b", "d") {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
if !s.HasAll("a", "b") {
|
|
t.Errorf("Missing contents: %#v", s)
|
|
}
|
|
s2.Insert("a", "b", "d")
|
|
if s.IsSuperset(s2) {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
s2.Delete("d")
|
|
if !s.IsSuperset(s2) {
|
|
t.Errorf("Missing contents: %#v", s)
|
|
}
|
|
}
|
|
|
|
func TestStringSetDeleteMultiples(t *testing.T) {
|
|
s := StringSet{}
|
|
s.Insert("a", "b", "c")
|
|
if len(s) != 3 {
|
|
t.Errorf("Expected len=3: %d", len(s))
|
|
}
|
|
|
|
s.Delete("a", "c")
|
|
if len(s) != 1 {
|
|
t.Errorf("Expected len=1: %d", len(s))
|
|
}
|
|
if s.Has("a") {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
if s.Has("c") {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
if !s.Has("b") {
|
|
t.Errorf("Missing contents: %#v", s)
|
|
}
|
|
|
|
}
|
|
|
|
func TestNewStringSet(t *testing.T) {
|
|
s := NewStringSet("a", "b", "c")
|
|
if len(s) != 3 {
|
|
t.Errorf("Expected len=3: %d", len(s))
|
|
}
|
|
if !s.Has("a") || !s.Has("b") || !s.Has("c") {
|
|
t.Errorf("Unexpected contents: %#v", s)
|
|
}
|
|
}
|
|
|
|
func TestStringSetList(t *testing.T) {
|
|
s := NewStringSet("z", "y", "x", "a")
|
|
if !reflect.DeepEqual(s.List(), []string{"a", "x", "y", "z"}) {
|
|
t.Errorf("List gave unexpected result: %#v", s.List())
|
|
}
|
|
}
|
|
|
|
func TestStringSetDifference(t *testing.T) {
|
|
a := NewStringSet("1", "2", "3")
|
|
b := NewStringSet("1", "2", "4", "5")
|
|
c := a.Difference(b)
|
|
d := b.Difference(a)
|
|
if len(c) != 1 {
|
|
t.Errorf("Expected len=1: %d", len(c))
|
|
}
|
|
if !c.Has("3") {
|
|
t.Errorf("Unexpected contents: %#v", c.List())
|
|
}
|
|
if len(d) != 2 {
|
|
t.Errorf("Expected len=2: %d", len(d))
|
|
}
|
|
if !d.Has("4") || !d.Has("5") {
|
|
t.Errorf("Unexpected contents: %#v", d.List())
|
|
}
|
|
}
|