add unit test cases for kubelet.util.sliceutils

pkg/kubelet/util/sliceutils/BUILD

@@ -5,6 +5,7 @@ licenses(["notice"])
 load(
     "@io_bazel_rules_go//go:def.bzl",
     "go_library",
+    "go_test",
 )
 
 go_library(
@@ -29,3 +30,15 @@ filegroup(
     srcs = [":package-srcs"],
     tags = ["automanaged"],
 )
+
+go_test(
+    name = "go_default_test",
+    srcs = ["sliceutils_test.go"],
+    library = ":go_default_library",
+    tags = ["automanaged"],
+    deps = [
+        "//pkg/api/v1:go_default_library",
+        "//pkg/kubelet/container:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
+    ],
+)

pkg/kubelet/util/sliceutils/sliceutils_test.go

@@ -0,0 +1,222 @@
+/*
+Copyright 2017 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 sliceutils
+
+import (
+	"testing"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/kubernetes/pkg/api/v1"
+	kubecontainer "k8s.io/kubernetes/pkg/kubelet/container"
+	"time"
+)
+
+func TestStringInSlice(t *testing.T) {
+	fooTests := []struct {
+		s    string
+		list []string
+		er   bool
+	}{
+		{"first", []string{"first", "second"}, true},
+		{"FIRST", []string{"first", "second"}, false},
+		{"third", []string{"first", "second"}, false},
+		{"first", nil, false},
+
+		{"", []string{"first", "second"}, false},
+		{"", []string{"first", "second", ""}, true},
+		{"", nil, false},
+	}
+
+	for _, fooTest := range fooTests {
+		r := StringInSlice(fooTest.s, fooTest.list)
+		if r != fooTest.er {
+			t.Errorf("returned %t but expected %t for s=%s & list=%s", r, fooTest.er, fooTest.s, fooTest.list)
+		}
+	}
+}
+
+func buildPodsByCreationTime() PodsByCreationTime {
+	return []*v1.Pod{
+		{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      "foo1",
+				Namespace: v1.NamespaceDefault,
+				CreationTimestamp: metav1.Time{
+					Time: time.Now(),
+				},
+			},
+		},
+		{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      "foo2",
+				Namespace: v1.NamespaceDefault,
+				CreationTimestamp: metav1.Time{
+					Time: time.Now().Add(time.Hour * 1),
+				},
+			},
+		},
+		{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      "foo3",
+				Namespace: v1.NamespaceDefault,
+				CreationTimestamp: metav1.Time{
+					Time: time.Now().Add(time.Hour * 2),
+				},
+			},
+		},
+	}
+}
+
+func TestPodsByCreationTimeLen(t *testing.T) {
+	fooTests := []struct {
+		pods PodsByCreationTime
+		el   int
+	}{
+		{[]*v1.Pod{}, 0},
+		{buildPodsByCreationTime(), 3},
+		{[]*v1.Pod{nil, {}}, 2},
+		{nil, 0},
+	}
+
+	for _, fooTest := range fooTests {
+		r := fooTest.pods.Len()
+		if r != fooTest.el {
+			t.Errorf("returned %d but expected %d for the len of PodsByCreationTime=%s", r, fooTest.el, fooTest.pods)
+		}
+	}
+}
+
+func TestPodsByCreationTimeSwap(t *testing.T) {
+	fooTests := []struct {
+		pods PodsByCreationTime
+		i    int
+		j    int
+	}{
+		{buildPodsByCreationTime(), 0, 1},
+		{buildPodsByCreationTime(), 2, 1},
+	}
+
+	for _, fooTest := range fooTests {
+		fooi := fooTest.pods[fooTest.i]
+		fooj := fooTest.pods[fooTest.j]
+		fooTest.pods.Swap(fooTest.i, fooTest.j)
+		if fooi.GetName() != fooTest.pods[fooTest.j].GetName() || fooj.GetName() != fooTest.pods[fooTest.i].GetName() {
+			t.Errorf("failed to swap for %v", fooTest)
+		}
+	}
+}
+
+func TestPodsByCreationTimeLess(t *testing.T) {
+	fooTests := []struct {
+		pods PodsByCreationTime
+		i    int
+		j    int
+		er   bool
+	}{
+		// ascending order
+		{buildPodsByCreationTime(), 0, 2, true},
+		{buildPodsByCreationTime(), 1, 0, false},
+	}
+
+	for _, fooTest := range fooTests {
+		r := fooTest.pods.Less(fooTest.i, fooTest.j)
+		if r != fooTest.er {
+			t.Errorf("returned %t but expected %t for the foo=%s", r, fooTest.er, fooTest.pods)
+		}
+	}
+}
+
+func buildByImageSize() ByImageSize {
+	return []kubecontainer.Image{
+		{
+			ID:          "1",
+			RepoTags:    []string{"foo-tag11", "foo-tag12"},
+			RepoDigests: []string{"foo-rd11", "foo-rd12"},
+			Size:        1,
+		},
+		{
+			ID:          "2",
+			RepoTags:    []string{"foo-tag21", "foo-tag22"},
+			RepoDigests: []string{"foo-rd21", "foo-rd22"},
+			Size:        2,
+		},
+		{
+			ID:          "3",
+			RepoTags:    []string{"foo-tag31", "foo-tag32"},
+			RepoDigests: []string{"foo-rd31", "foo-rd32"},
+			Size:        3,
+		},
+	}
+}
+
+func TestByImageSizeLen(t *testing.T) {
+	fooTests := []struct {
+		images ByImageSize
+		el     int
+	}{
+		{[]kubecontainer.Image{}, 0},
+		{buildByImageSize(), 3},
+		{nil, 0},
+	}
+
+	for _, fooTest := range fooTests {
+		r := fooTest.images.Len()
+		if r != fooTest.el {
+			t.Errorf("returned %d but expected %d for the len of ByImageSize=%v", r, fooTest.el, fooTest.images)
+		}
+	}
+}
+
+func TestByImageSizeSwap(t *testing.T) {
+	fooTests := []struct {
+		images ByImageSize
+		i      int
+		j      int
+	}{
+		{buildByImageSize(), 0, 1},
+		{buildByImageSize(), 2, 1},
+	}
+
+	for _, fooTest := range fooTests {
+		fooi := fooTest.images[fooTest.i]
+		fooj := fooTest.images[fooTest.j]
+		fooTest.images.Swap(fooTest.i, fooTest.j)
+		if fooi.ID != fooTest.images[fooTest.j].ID || fooj.ID != fooTest.images[fooTest.i].ID {
+			t.Errorf("failed to swap for %v", fooTest)
+		}
+	}
+}
+
+func TestByImageSizeLess(t *testing.T) {
+	fooTests := []struct {
+		images ByImageSize
+		i      int
+		j      int
+		er     bool
+	}{
+		// descending order
+		{buildByImageSize(), 0, 2, false},
+		{buildByImageSize(), 1, 0, true},
+	}
+
+	for _, fooTest := range fooTests {
+		r := fooTest.images.Less(fooTest.i, fooTest.j)
+		if r != fooTest.er {
+			t.Errorf("returned %t but expected %t for the foo=%v", r, fooTest.er, fooTest.images)
+		}
+	}
+}