custom-kube-scheduler/kubernetes/pkg/kubelet/checkpoint/checkpoint_test.go

/*
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 checkpoint

import (
	"io/ioutil"
	"os"
	"reflect"
	"testing"

	"k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/kubernetes/pkg/apis/core"
	"k8s.io/kubernetes/pkg/kubelet/checkpointmanager"
)

// TestWriteLoadDeletePods validates all combinations of write, load, and delete
func TestWriteLoadDeletePods(t *testing.T) {
	testPods := []struct {
		pod     *v1.Pod
		written bool
	}{
		{
			pod: &v1.Pod{
				ObjectMeta: metav1.ObjectMeta{
					Name:        "Foo",
					Annotations: map[string]string{core.BootstrapCheckpointAnnotationKey: "true"},
					UID:         "1",
				},
			},
			written: true,
		},
		{
			pod: &v1.Pod{
				ObjectMeta: metav1.ObjectMeta{
					Name:        "Foo2",
					Annotations: map[string]string{core.BootstrapCheckpointAnnotationKey: "true"},
					UID:         "2",
				},
			},
			written: true,
		},
		{
			pod: &v1.Pod{
				ObjectMeta: metav1.ObjectMeta{
					Name: "Bar",
					UID:  "3",
				},
			},
			written: false,
		},
	}

	dir, err := ioutil.TempDir("", "checkpoint")
	if err != nil {
		t.Errorf("Failed to allocate temp directory for TestWriteLoadDeletePods error=%v", err)
	}
	defer os.RemoveAll(dir)

	cpm, err := checkpointmanager.NewCheckpointManager(dir)
	if err != nil {
		t.Errorf("Failed to initialize checkpoint manager error=%v", err)
	}
	for _, p := range testPods {
		// Write pods should always pass unless there is an fs error
		if err := WritePod(cpm, p.pod); err != nil {
			t.Errorf("Failed to Write Pod: %v", err)
		}
	}
	// verify the correct written files are loaded from disk
	pods, err := LoadPods(cpm)
	if err != nil {
		t.Errorf("Failed to Load Pods: %v", err)
	}
	// loop through contents and check make sure
	// what was loaded matched the expected results.
	for _, p := range testPods {
		pname := p.pod.GetName()
		var lpod *v1.Pod
		for _, check := range pods {
			if check.GetName() == pname {
				lpod = check
				break
			}
		}
		if p.written {
			if lpod != nil {
				if !reflect.DeepEqual(p.pod, lpod) {
					t.Errorf("expected %#v, \ngot %#v", p.pod, lpod)
				}
			} else {
				t.Errorf("Got unexpected result for %v, should have been loaded", pname)
			}
		} else if lpod != nil {
			t.Errorf("Got unexpected result for %v, should not have been loaded", pname)
		}
		err = DeletePod(cpm, p.pod)
		if err != nil {
			t.Errorf("Failed to delete pod %v", pname)
		}
	}
	// finally validate the contents of the directory is empty.
	files, err := ioutil.ReadDir(dir)
	if err != nil {
		t.Errorf("Failed to read directory %v", dir)
	}
	if len(files) > 0 {
		t.Errorf("Directory %v should be empty but found %#v", dir, files)
	}
}