/*
|
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 cache
|
|
import (
|
"k8s.io/apimachinery/pkg/util/sets"
|
"strings"
|
"testing"
|
|
"k8s.io/api/core/v1"
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
)
|
|
func testIndexFunc(obj interface{}) ([]string, error) {
|
pod := obj.(*v1.Pod)
|
return []string{pod.Labels["foo"]}, nil
|
}
|
|
func TestGetIndexFuncValues(t *testing.T) {
|
index := NewIndexer(MetaNamespaceKeyFunc, Indexers{"testmodes": testIndexFunc})
|
|
pod1 := &v1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "one", Labels: map[string]string{"foo": "bar"}}}
|
pod2 := &v1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "two", Labels: map[string]string{"foo": "bar"}}}
|
pod3 := &v1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "tre", Labels: map[string]string{"foo": "biz"}}}
|
|
index.Add(pod1)
|
index.Add(pod2)
|
index.Add(pod3)
|
|
keys := index.ListIndexFuncValues("testmodes")
|
if len(keys) != 2 {
|
t.Errorf("Expected 2 keys but got %v", len(keys))
|
}
|
|
for _, key := range keys {
|
if key != "bar" && key != "biz" {
|
t.Errorf("Expected only 'bar' or 'biz' but got %s", key)
|
}
|
}
|
}
|
|
func testUsersIndexFunc(obj interface{}) ([]string, error) {
|
pod := obj.(*v1.Pod)
|
usersString := pod.Annotations["users"]
|
|
return strings.Split(usersString, ","), nil
|
}
|
|
func TestMultiIndexKeys(t *testing.T) {
|
index := NewIndexer(MetaNamespaceKeyFunc, Indexers{"byUser": testUsersIndexFunc})
|
|
pod1 := &v1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "one", Annotations: map[string]string{"users": "ernie,bert"}}}
|
pod2 := &v1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "two", Annotations: map[string]string{"users": "bert,oscar"}}}
|
pod3 := &v1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "tre", Annotations: map[string]string{"users": "ernie,elmo"}}}
|
|
index.Add(pod1)
|
index.Add(pod2)
|
index.Add(pod3)
|
|
expected := map[string]sets.String{}
|
expected["ernie"] = sets.NewString("one", "tre")
|
expected["bert"] = sets.NewString("one", "two")
|
expected["elmo"] = sets.NewString("tre")
|
expected["oscar"] = sets.NewString("two")
|
expected["elmo"] = sets.NewString() // let's just make sure we don't get anything back in this case
|
{
|
for k, v := range expected {
|
found := sets.String{}
|
indexResults, err := index.ByIndex("byUser", k)
|
if err != nil {
|
t.Errorf("Unexpected error %v", err)
|
}
|
for _, item := range indexResults {
|
found.Insert(item.(*v1.Pod).Name)
|
}
|
items := v.List()
|
if !found.HasAll(items...) {
|
t.Errorf("missing items, index %s, expected %v but found %v", k, items, found.List())
|
}
|
}
|
}
|
|
index.Delete(pod3)
|
erniePods, err := index.ByIndex("byUser", "ernie")
|
if err != nil {
|
t.Errorf("unexpected error: %v", err)
|
}
|
if len(erniePods) != 1 {
|
t.Errorf("Expected 1 pods but got %v", len(erniePods))
|
}
|
for _, erniePod := range erniePods {
|
if erniePod.(*v1.Pod).Name != "one" {
|
t.Errorf("Expected only 'one' but got %s", erniePod.(*v1.Pod).Name)
|
}
|
}
|
|
elmoPods, err := index.ByIndex("byUser", "elmo")
|
if err != nil {
|
t.Errorf("unexpected error: %v", err)
|
}
|
if len(elmoPods) != 0 {
|
t.Errorf("Expected 0 pods but got %v", len(elmoPods))
|
}
|
|
copyOfPod2 := pod2.DeepCopy()
|
copyOfPod2.Annotations["users"] = "oscar"
|
index.Update(copyOfPod2)
|
bertPods, err := index.ByIndex("byUser", "bert")
|
if err != nil {
|
t.Errorf("unexpected error: %v", err)
|
}
|
if len(bertPods) != 1 {
|
t.Errorf("Expected 1 pods but got %v", len(bertPods))
|
}
|
for _, bertPod := range bertPods {
|
if bertPod.(*v1.Pod).Name != "one" {
|
t.Errorf("Expected only 'one' but got %s", bertPod.(*v1.Pod).Name)
|
}
|
}
|
}
|
