mirror of https://github.com/k3s-io/k3s
110 lines
2.5 KiB
Go
110 lines
2.5 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 cache
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
|
|
)
|
|
|
|
// Test public interface
|
|
func doTestStore(t *testing.T, store Store) {
|
|
mkObj := func(id string, val string) testStoreObject {
|
|
return testStoreObject{id: id, val: val}
|
|
}
|
|
|
|
store.Add(mkObj("foo", "bar"))
|
|
if item, ok, _ := store.Get(mkObj("foo", "")); !ok {
|
|
t.Errorf("didn't find inserted item")
|
|
} else {
|
|
if e, a := "bar", item.(testStoreObject).val; e != a {
|
|
t.Errorf("expected %v, got %v", e, a)
|
|
}
|
|
}
|
|
store.Update(mkObj("foo", "baz"))
|
|
if item, ok, _ := store.Get(mkObj("foo", "")); !ok {
|
|
t.Errorf("didn't find inserted item")
|
|
} else {
|
|
if e, a := "baz", item.(testStoreObject).val; e != a {
|
|
t.Errorf("expected %v, got %v", e, a)
|
|
}
|
|
}
|
|
store.Delete(mkObj("foo", ""))
|
|
if _, ok, _ := store.Get(mkObj("foo", "")); ok {
|
|
t.Errorf("found deleted item??")
|
|
}
|
|
|
|
// Test List.
|
|
store.Add(mkObj("a", "b"))
|
|
store.Add(mkObj("c", "d"))
|
|
store.Add(mkObj("e", "e"))
|
|
{
|
|
found := util.StringSet{}
|
|
for _, item := range store.List() {
|
|
found.Insert(item.(testStoreObject).val)
|
|
}
|
|
if !found.HasAll("b", "d", "e") {
|
|
t.Errorf("missing items, found: %v", found)
|
|
}
|
|
if len(found) != 3 {
|
|
t.Errorf("extra items")
|
|
}
|
|
}
|
|
|
|
// Test Replace.
|
|
store.Replace([]interface{}{
|
|
mkObj("foo", "foo"),
|
|
mkObj("bar", "bar"),
|
|
})
|
|
|
|
{
|
|
found := util.StringSet{}
|
|
for _, item := range store.List() {
|
|
found.Insert(item.(testStoreObject).val)
|
|
}
|
|
if !found.HasAll("foo", "bar") {
|
|
t.Errorf("missing items")
|
|
}
|
|
if len(found) != 2 {
|
|
t.Errorf("extra items")
|
|
}
|
|
}
|
|
}
|
|
|
|
func testStoreKeyFunc(obj interface{}) (string, error) {
|
|
return obj.(testStoreObject).id, nil
|
|
}
|
|
|
|
type testStoreObject struct {
|
|
id string
|
|
val string
|
|
}
|
|
|
|
func TestCache(t *testing.T) {
|
|
doTestStore(t, NewStore(testStoreKeyFunc))
|
|
}
|
|
|
|
func TestFIFOCache(t *testing.T) {
|
|
doTestStore(t, NewFIFO(testStoreKeyFunc))
|
|
}
|
|
|
|
func TestUndeltaStore(t *testing.T) {
|
|
nop := func([]interface{}) {}
|
|
doTestStore(t, NewUndeltaStore(nop, testStoreKeyFunc))
|
|
}
|