mirror of https://github.com/k3s-io/k3s
335 lines
8.4 KiB
Go
335 lines
8.4 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors 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 storage
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/errors"
|
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
|
"k8s.io/kubernetes/pkg/client/cache"
|
|
"k8s.io/kubernetes/pkg/runtime"
|
|
"k8s.io/kubernetes/pkg/util"
|
|
"k8s.io/kubernetes/pkg/util/sets"
|
|
"k8s.io/kubernetes/pkg/util/wait"
|
|
"k8s.io/kubernetes/pkg/watch"
|
|
)
|
|
|
|
func makeTestPod(name string, resourceVersion uint64) *api.Pod {
|
|
return &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Namespace: "ns",
|
|
Name: name,
|
|
ResourceVersion: strconv.FormatUint(resourceVersion, 10),
|
|
},
|
|
}
|
|
}
|
|
|
|
// newTestWatchCache just adds a fake clock.
|
|
func newTestWatchCache(capacity int) *watchCache {
|
|
wc := newWatchCache(capacity)
|
|
wc.clock = util.NewFakeClock(time.Now())
|
|
return wc
|
|
}
|
|
|
|
func TestWatchCacheBasic(t *testing.T) {
|
|
store := newTestWatchCache(2)
|
|
|
|
// Test Add/Update/Delete.
|
|
pod1 := makeTestPod("pod", 1)
|
|
if err := store.Add(pod1); err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if item, ok, _ := store.Get(pod1); !ok {
|
|
t.Errorf("didn't find pod")
|
|
} else {
|
|
if !api.Semantic.DeepEqual(pod1, item) {
|
|
t.Errorf("expected %v, got %v", pod1, item)
|
|
}
|
|
}
|
|
pod2 := makeTestPod("pod", 2)
|
|
if err := store.Update(pod2); err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if item, ok, _ := store.Get(pod2); !ok {
|
|
t.Errorf("didn't find pod")
|
|
} else {
|
|
if !api.Semantic.DeepEqual(pod2, item) {
|
|
t.Errorf("expected %v, got %v", pod1, item)
|
|
}
|
|
}
|
|
pod3 := makeTestPod("pod", 3)
|
|
if err := store.Delete(pod3); err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if _, ok, _ := store.Get(pod3); ok {
|
|
t.Errorf("found pod")
|
|
}
|
|
|
|
// Test List.
|
|
store.Add(makeTestPod("pod1", 4))
|
|
store.Add(makeTestPod("pod2", 5))
|
|
store.Add(makeTestPod("pod3", 6))
|
|
{
|
|
podNames := sets.String{}
|
|
for _, item := range store.List() {
|
|
podNames.Insert(item.(*api.Pod).ObjectMeta.Name)
|
|
}
|
|
if !podNames.HasAll("pod1", "pod2", "pod3") {
|
|
t.Errorf("missing pods, found %v", podNames)
|
|
}
|
|
if len(podNames) != 3 {
|
|
t.Errorf("found missing/extra items")
|
|
}
|
|
}
|
|
|
|
// Test Replace.
|
|
store.Replace([]interface{}{
|
|
makeTestPod("pod4", 7),
|
|
makeTestPod("pod5", 8),
|
|
}, "8")
|
|
{
|
|
podNames := sets.String{}
|
|
for _, item := range store.List() {
|
|
podNames.Insert(item.(*api.Pod).ObjectMeta.Name)
|
|
}
|
|
if !podNames.HasAll("pod4", "pod5") {
|
|
t.Errorf("missing pods, found %v", podNames)
|
|
}
|
|
if len(podNames) != 2 {
|
|
t.Errorf("found missing/extra items")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEvents(t *testing.T) {
|
|
store := newTestWatchCache(5)
|
|
|
|
store.Add(makeTestPod("pod", 2))
|
|
|
|
// Test for Added event.
|
|
{
|
|
_, err := store.GetAllEventsSince(1)
|
|
if err == nil {
|
|
t.Errorf("expected error too old")
|
|
}
|
|
if _, ok := err.(*errors.StatusError); !ok {
|
|
t.Errorf("expected error to be of type StatusError")
|
|
}
|
|
}
|
|
{
|
|
result, err := store.GetAllEventsSince(2)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if len(result) != 1 {
|
|
t.Fatalf("unexpected events: %v", result)
|
|
}
|
|
if result[0].Type != watch.Added {
|
|
t.Errorf("unexpected event type: %v", result[0].Type)
|
|
}
|
|
pod := makeTestPod("pod", uint64(2))
|
|
if !api.Semantic.DeepEqual(pod, result[0].Object) {
|
|
t.Errorf("unexpected item: %v, expected: %v", result[0].Object, pod)
|
|
}
|
|
if result[0].PrevObject != nil {
|
|
t.Errorf("unexpected item: %v", result[0].PrevObject)
|
|
}
|
|
}
|
|
|
|
store.Update(makeTestPod("pod", 3))
|
|
store.Update(makeTestPod("pod", 4))
|
|
|
|
// Test with not full cache.
|
|
{
|
|
_, err := store.GetAllEventsSince(1)
|
|
if err == nil {
|
|
t.Errorf("expected error too old")
|
|
}
|
|
}
|
|
{
|
|
result, err := store.GetAllEventsSince(3)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if len(result) != 2 {
|
|
t.Fatalf("unexpected events: %v", result)
|
|
}
|
|
for i := 0; i < 2; i++ {
|
|
if result[i].Type != watch.Modified {
|
|
t.Errorf("unexpected event type: %v", result[i].Type)
|
|
}
|
|
pod := makeTestPod("pod", uint64(i+3))
|
|
if !api.Semantic.DeepEqual(pod, result[i].Object) {
|
|
t.Errorf("unexpected item: %v, expected: %v", result[i].Object, pod)
|
|
}
|
|
prevPod := makeTestPod("pod", uint64(i+2))
|
|
if !api.Semantic.DeepEqual(prevPod, result[i].PrevObject) {
|
|
t.Errorf("unexpected item: %v, expected: %v", result[i].PrevObject, prevPod)
|
|
}
|
|
}
|
|
}
|
|
|
|
for i := 5; i < 9; i++ {
|
|
store.Update(makeTestPod("pod", uint64(i)))
|
|
}
|
|
|
|
// Test with full cache - there should be elements from 4 to 8.
|
|
{
|
|
_, err := store.GetAllEventsSince(3)
|
|
if err == nil {
|
|
t.Errorf("expected error too old")
|
|
}
|
|
}
|
|
{
|
|
result, err := store.GetAllEventsSince(4)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if len(result) != 5 {
|
|
t.Fatalf("unexpected events: %v", result)
|
|
}
|
|
for i := 0; i < 5; i++ {
|
|
pod := makeTestPod("pod", uint64(i+4))
|
|
if !api.Semantic.DeepEqual(pod, result[i].Object) {
|
|
t.Errorf("unexpected item: %v, expected: %v", result[i].Object, pod)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Test for delete event.
|
|
store.Delete(makeTestPod("pod", uint64(9)))
|
|
|
|
{
|
|
result, err := store.GetAllEventsSince(9)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if len(result) != 1 {
|
|
t.Fatalf("unexpected events: %v", result)
|
|
}
|
|
if result[0].Type != watch.Deleted {
|
|
t.Errorf("unexpected event type: %v", result[0].Type)
|
|
}
|
|
pod := makeTestPod("pod", uint64(9))
|
|
if !api.Semantic.DeepEqual(pod, result[0].Object) {
|
|
t.Errorf("unexpected item: %v, expected: %v", result[0].Object, pod)
|
|
}
|
|
prevPod := makeTestPod("pod", uint64(8))
|
|
if !api.Semantic.DeepEqual(prevPod, result[0].PrevObject) {
|
|
t.Errorf("unexpected item: %v, expected: %v", result[0].PrevObject, prevPod)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestWaitUntilFreshAndList(t *testing.T) {
|
|
store := newTestWatchCache(3)
|
|
|
|
// In background, update the store.
|
|
go func() {
|
|
store.Add(makeTestPod("foo", 2))
|
|
store.Add(makeTestPod("bar", 5))
|
|
}()
|
|
|
|
list, resourceVersion, err := store.WaitUntilFreshAndList(5)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if resourceVersion != 5 {
|
|
t.Errorf("unexpected resourceVersion: %v, expected: 5", resourceVersion)
|
|
}
|
|
if len(list) != 2 {
|
|
t.Errorf("unexpected list returned: %#v", list)
|
|
}
|
|
}
|
|
|
|
func TestWaitUntilFreshAndListTimeout(t *testing.T) {
|
|
store := newTestWatchCache(3)
|
|
fc := store.clock.(*util.FakeClock)
|
|
|
|
// In background, step clock after the below call starts the timer.
|
|
go func() {
|
|
for !fc.HasWaiters() {
|
|
time.Sleep(time.Millisecond)
|
|
}
|
|
fc.Step(MaximumListWait)
|
|
|
|
// Add an object to make sure the test would
|
|
// eventually fail instead of just waiting
|
|
// forever.
|
|
time.Sleep(30 * time.Second)
|
|
store.Add(makeTestPod("bar", 5))
|
|
}()
|
|
|
|
_, _, err := store.WaitUntilFreshAndList(5)
|
|
if err == nil {
|
|
t.Fatalf("unexpected lack of timeout error")
|
|
}
|
|
}
|
|
|
|
type testLW struct {
|
|
ListFunc func(options api.ListOptions) (runtime.Object, error)
|
|
WatchFunc func(options api.ListOptions) (watch.Interface, error)
|
|
}
|
|
|
|
func (t *testLW) List(options api.ListOptions) (runtime.Object, error) {
|
|
return t.ListFunc(options)
|
|
}
|
|
func (t *testLW) Watch(options api.ListOptions) (watch.Interface, error) {
|
|
return t.WatchFunc(options)
|
|
}
|
|
|
|
func TestReflectorForWatchCache(t *testing.T) {
|
|
store := newTestWatchCache(5)
|
|
|
|
{
|
|
_, version, err := store.WaitUntilFreshAndList(0)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if version != 0 {
|
|
t.Errorf("unexpected resource version: %d", version)
|
|
}
|
|
}
|
|
|
|
lw := &testLW{
|
|
WatchFunc: func(options api.ListOptions) (watch.Interface, error) {
|
|
fw := watch.NewFake()
|
|
go fw.Stop()
|
|
return fw, nil
|
|
},
|
|
ListFunc: func(options api.ListOptions) (runtime.Object, error) {
|
|
return &api.PodList{ListMeta: unversioned.ListMeta{ResourceVersion: "10"}}, nil
|
|
},
|
|
}
|
|
r := cache.NewReflector(lw, &api.Pod{}, store, 0)
|
|
r.ListAndWatch(wait.NeverStop)
|
|
|
|
{
|
|
_, version, err := store.WaitUntilFreshAndList(10)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if version != 10 {
|
|
t.Errorf("unexpected resource version: %d", version)
|
|
}
|
|
}
|
|
}
|