func TestRunOnce(t *testing.T) { cadvisor := &cadvisor.Mock{} cadvisor.On("MachineInfo").Return(&cadvisorApi.MachineInfo{}, nil) podManager, _ := newFakePodManager() diskSpaceManager, _ := newDiskSpaceManager(cadvisor, DiskSpacePolicy{}) fakeRuntime := &kubecontainer.FakeRuntime{} kb := &Kubelet{ rootDirectory: "/tmp/kubelet", recorder: &record.FakeRecorder{}, cadvisor: cadvisor, nodeLister: testNodeLister{}, statusManager: status.NewManager(nil), containerRefManager: kubecontainer.NewRefManager(), readinessManager: kubecontainer.NewReadinessManager(), podManager: podManager, os: kubecontainer.FakeOS{}, volumeManager: newVolumeManager(), diskSpaceManager: diskSpaceManager, containerRuntime: fakeRuntime, } kb.containerManager, _ = newContainerManager(fakeContainerMgrMountInt(), cadvisor, "", "", "") kb.networkPlugin, _ = network.InitNetworkPlugin([]network.NetworkPlugin{}, "", network.NewFakeHost(nil)) if err := kb.setupDataDirs(); err != nil { t.Errorf("Failed to init data dirs: %v", err) } pods := []*api.Pod{ { ObjectMeta: api.ObjectMeta{ UID: "12345678", Name: "foo", Namespace: "new", }, Spec: api.PodSpec{ Containers: []api.Container{ {Name: "bar"}, }, }, }, } podManager.SetPods(pods) results, err := kb.runOnce(pods, time.Millisecond) if err != nil { t.Errorf("unexpected error: %v", err) } if results[0].Err != nil { t.Errorf("unexpected run pod error: %v", results[0].Err) } if results[0].Pod.Name != "foo" { t.Errorf("unexpected pod: %q", results[0].Pod.Name) } }
func newTestDockerManager() (*dockertools.DockerManager, *dockertools.FakeDockerClient) { fakeDocker := &dockertools.FakeDockerClient{VersionInfo: docker.Env{"Version=1.1.3", "ApiVersion=1.15"}, Errors: make(map[string]error), RemovedImages: sets.String{}} fakeRecorder := &record.FakeRecorder{} readinessManager := kubecontainer.NewReadinessManager() containerRefManager := kubecontainer.NewRefManager() networkPlugin, _ := network.InitNetworkPlugin([]network.NetworkPlugin{}, "", network.NewFakeHost(nil)) dockerManager := dockertools.NewFakeDockerManager( fakeDocker, fakeRecorder, readinessManager, containerRefManager, &cadvisorApi.MachineInfo{}, dockertools.PodInfraContainerImage, 0, 0, "", kubecontainer.FakeOS{}, networkPlugin, nil, nil) return dockerManager, fakeDocker }
func TestRunOnce(t *testing.T) { cadvisor := &cadvisor.Mock{} cadvisor.On("MachineInfo").Return(&cadvisorApi.MachineInfo{}, nil) podManager, _ := newFakePodManager() kb := &Kubelet{ rootDirectory: "/tmp/kubelet", recorder: &record.FakeRecorder{}, cadvisor: cadvisor, nodeLister: testNodeLister{}, statusManager: newStatusManager(nil), containerRefManager: kubecontainer.NewRefManager(), readinessManager: kubecontainer.NewReadinessManager(), podManager: podManager, os: kubecontainer.FakeOS{}, volumeManager: newVolumeManager(), } kb.containerManager, _ = newContainerManager(cadvisor, "", "", "") kb.networkPlugin, _ = network.InitNetworkPlugin([]network.NetworkPlugin{}, "", network.NewFakeHost(nil)) if err := kb.setupDataDirs(); err != nil { t.Errorf("Failed to init data dirs: %v", err) } podContainers := []docker.APIContainers{ { Names: []string{"/k8s_bar." + strconv.FormatUint(kubecontainer.HashContainer(&api.Container{Name: "bar"}), 16) + "_foo_new_12345678_42"}, ID: "1234", Status: "running", }, { Names: []string{"/k8s_net_foo.new.test_abcdefgh_42"}, ID: "9876", Status: "running", }, } kb.dockerClient = &testDocker{ listContainersResults: []listContainersResult{ {label: "list pod container", containers: []docker.APIContainers{}}, {label: "syncPod", containers: []docker.APIContainers{}}, {label: "list pod container", containers: []docker.APIContainers{}}, {label: "syncPod", containers: podContainers}, {label: "list pod container", containers: podContainers}, {label: "list pod container", containers: podContainers}, }, inspectContainersResults: []inspectContainersResult{ { label: "syncPod", container: docker.Container{ Config: &docker.Config{Image: "someimage"}, State: docker.State{Running: true, Pid: 42}, }, }, { label: "syncPod", container: docker.Container{ Config: &docker.Config{Image: "someimage"}, State: docker.State{Running: true, Pid: 42}, }, }, { label: "syncPod", container: docker.Container{ Config: &docker.Config{Image: "someimage"}, State: docker.State{Running: true, Pid: 42}, }, }, { label: "syncPod", container: docker.Container{ Config: &docker.Config{Image: "someimage"}, State: docker.State{Running: true, Pid: 42}, }, }, }, t: t, } kb.containerRuntime = dockertools.NewFakeDockerManager( kb.dockerClient, kb.recorder, kb.readinessManager, kb.containerRefManager, dockertools.PodInfraContainerImage, 0, 0, "", kubecontainer.FakeOS{}, kb.networkPlugin, kb, nil, newKubeletRuntimeHooks(kb.recorder)) pods := []*api.Pod{ { ObjectMeta: api.ObjectMeta{ UID: "12345678", Name: "foo", Namespace: "new", }, Spec: api.PodSpec{ Containers: []api.Container{ {Name: "bar"}, }, }, }, } podManager.SetPods(pods) results, err := kb.runOnce(pods, time.Millisecond) if err != nil { t.Errorf("unexpected error: %v", err) } if results[0].Err != nil { t.Errorf("unexpected run pod error: %v", results[0].Err) } if results[0].Pod.Name != "foo" { t.Errorf("unexpected pod: %q", results[0].Pod.Name) } }
// TestProbeContainer tests the functionality of probeContainer. // Test cases are: // // No probe. // Only LivenessProbe. // Only ReadinessProbe. // Both probes. // // Also, for each probe, there will be several cases covering whether the initial // delay has passed, whether the probe handler will return Success, Failure, // Unknown or error. // // PLEASE READ THE PROBE DOCS BEFORE CHANGING THIS TEST IF YOU ARE UNSURE HOW PROBES ARE SUPPOSED TO WORK: // (See https://github.com/GoogleCloudPlatform/kubernetes/blob/master/docs/user-guide/pod-states.md#pod-conditions) func TestProbeContainer(t *testing.T) { readinessManager := kubecontainer.NewReadinessManager() prober := &prober{ readinessManager: readinessManager, refManager: kubecontainer.NewRefManager(), recorder: &record.FakeRecorder{}, } containerID := "foobar" createdAt := time.Now().Unix() tests := []struct { testContainer api.Container expectError bool expectedResult probe.Result expectedReadiness bool }{ // No probes. { testContainer: api.Container{}, expectedResult: probe.Success, expectedReadiness: true, }, // Only LivenessProbe. expectedReadiness should always be true here. { testContainer: api.Container{ LivenessProbe: &api.Probe{InitialDelaySeconds: 100}, }, expectedResult: probe.Success, expectedReadiness: true, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{InitialDelaySeconds: -100}, }, expectedResult: probe.Unknown, expectedReadiness: true, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Failure, expectedReadiness: true, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Success, expectedReadiness: true, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Unknown, expectedReadiness: true, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectError: true, expectedResult: probe.Unknown, expectedReadiness: true, }, // // Only ReadinessProbe. expectedResult should always be probe.Success here. { testContainer: api.Container{ ReadinessProbe: &api.Probe{InitialDelaySeconds: 100}, }, expectedResult: probe.Success, expectedReadiness: false, }, { testContainer: api.Container{ ReadinessProbe: &api.Probe{InitialDelaySeconds: -100}, }, expectedResult: probe.Success, expectedReadiness: false, }, { testContainer: api.Container{ ReadinessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Success, expectedReadiness: true, }, { testContainer: api.Container{ ReadinessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Success, expectedReadiness: true, }, { testContainer: api.Container{ ReadinessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Success, expectedReadiness: true, }, { testContainer: api.Container{ ReadinessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectError: false, expectedResult: probe.Success, expectedReadiness: true, }, // Both LivenessProbe and ReadinessProbe. { testContainer: api.Container{ LivenessProbe: &api.Probe{InitialDelaySeconds: 100}, ReadinessProbe: &api.Probe{InitialDelaySeconds: 100}, }, expectedResult: probe.Success, expectedReadiness: false, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{InitialDelaySeconds: 100}, ReadinessProbe: &api.Probe{InitialDelaySeconds: -100}, }, expectedResult: probe.Success, expectedReadiness: false, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{InitialDelaySeconds: -100}, ReadinessProbe: &api.Probe{InitialDelaySeconds: 100}, }, expectedResult: probe.Unknown, expectedReadiness: false, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{InitialDelaySeconds: -100}, ReadinessProbe: &api.Probe{InitialDelaySeconds: -100}, }, expectedResult: probe.Unknown, expectedReadiness: false, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, ReadinessProbe: &api.Probe{InitialDelaySeconds: -100}, }, expectedResult: probe.Unknown, expectedReadiness: false, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, ReadinessProbe: &api.Probe{InitialDelaySeconds: -100}, }, expectedResult: probe.Failure, expectedReadiness: false, }, { testContainer: api.Container{ LivenessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, ReadinessProbe: &api.Probe{ InitialDelaySeconds: -100, Handler: api.Handler{ Exec: &api.ExecAction{}, }, }, }, expectedResult: probe.Success, expectedReadiness: true, }, } for i, test := range tests { if test.expectError { prober.exec = fakeExecProber{test.expectedResult, errors.New("exec error")} } else { prober.exec = fakeExecProber{test.expectedResult, nil} } result, err := prober.Probe(&api.Pod{}, api.PodStatus{}, test.testContainer, containerID, createdAt) if test.expectError && err == nil { t.Errorf("[%d] Expected error but no error was returned.", i) } if !test.expectError && err != nil { t.Errorf("[%d] Didn't expect error but got: %v", i, err) } if test.expectedResult != result { t.Errorf("[%d] Expected result to be %v but was %v", i, test.expectedResult, result) } if test.expectedReadiness != readinessManager.GetReadiness(containerID) { t.Errorf("[%d] Expected readiness to be %v but was %v", i, test.expectedReadiness, readinessManager.GetReadiness(containerID)) } } }