fix: pod worker test

Signed-off-by: jonyhy96 <hy352144278@gmail.com>
This commit is contained in:
jonyhy96 2022-02-24 16:35:33 +08:00
parent fb82a0d7eb
commit 60cd896602

View File

@ -288,7 +288,7 @@ func TestUpdatePod(t *testing.T) {
func TestUpdatePodWithTerminatedPod(t *testing.T) { func TestUpdatePodWithTerminatedPod(t *testing.T) {
podWorkers, _ := createPodWorkers() podWorkers, _ := createPodWorkers()
terminatedPod := newPodWithPhase("0000-0000-0000", "done-pod", v1.PodSucceeded) terminatedPod := newPodWithPhase("0000-0000-0000", "done-pod", v1.PodSucceeded)
runningPod := &kubecontainer.Pod{ID: "0000-0000-0001", Name: "done-pod"} orphanedPod := &kubecontainer.Pod{ID: "0000-0000-0001", Name: "orphaned-pod"}
pod := newPod("0000-0000-0002", "running-pod") pod := newPod("0000-0000-0002", "running-pod")
podWorkers.UpdatePod(UpdatePodOptions{ podWorkers.UpdatePod(UpdatePodOptions{
@ -301,17 +301,17 @@ func TestUpdatePodWithTerminatedPod(t *testing.T) {
}) })
podWorkers.UpdatePod(UpdatePodOptions{ podWorkers.UpdatePod(UpdatePodOptions{
UpdateType: kubetypes.SyncPodKill, UpdateType: kubetypes.SyncPodKill,
RunningPod: runningPod, RunningPod: orphanedPod,
}) })
drainAllWorkers(podWorkers)
if podWorkers.IsPodKnownTerminated(pod.UID) == true { if podWorkers.IsPodKnownTerminated(pod.UID) == true {
t.Errorf("podWorker state should not be terminated") t.Errorf("podWorker state should not be terminated")
} }
if podWorkers.IsPodKnownTerminated(terminatedPod.UID) == false { if podWorkers.IsPodKnownTerminated(terminatedPod.UID) == false {
t.Errorf("podWorker state should be terminated") t.Errorf("podWorker state should be terminated")
} }
if podWorkers.IsPodKnownTerminated(runningPod.ID) == true { if podWorkers.IsPodKnownTerminated(orphanedPod.ID) == false {
t.Errorf("podWorker state should not be marked terminated for a running pod") t.Errorf("podWorker state should be terminated for orphaned pod")
} }
} }