Function TestDrainSuccess

pkg/node/node_test.go:124–171

(t *testing.T)

}

func TestDrainSuccess(t *testing.T) {
	isOwnerController := true
	client := fake.NewSimpleClientset()
	_, err := client.CoreV1().Nodes().Create(
		context.Background(),
		&v1.Node{
			ObjectMeta: metav1.ObjectMeta{Name: nodeName},
		},
		metav1.CreateOptions{})
	h.Ok(t, err)

	_, err = client.CoreV1().Pods("default").Create(
		context.Background(),
		&v1.Pod{
			ObjectMeta: metav1.ObjectMeta{
				GenerateName: "cool-app-pod-",
				OwnerReferences: []metav1.OwnerReference{
					{
						APIVersion: "apps/v1",
						Name:       "cool-app",
						Kind:       "ReplicaSet",
						Controller: &isOwnerController,
					},
				},
			},
			Spec: v1.PodSpec{
				NodeName: nodeName,
			},
		},
		metav1.CreateOptions{})
	h.Ok(t, err)

	fakeRecorder := record.NewFakeRecorder(recorderBufferSize)

	drainHelper := getDrainHelper(client)
	drainHelper.DisableEviction = true
	tNode := getNode(t, drainHelper)
	err = tNode.CordonAndDrain(nodeName, "cordonReason", fakeRecorder)
	close(fakeRecorder.Events)
	h.Ok(t, err)
	expectedEventArrived := false
	for event := range fakeRecorder.Events {
		if strings.Contains(event, "Normal PodEviction Pod evicted due to node drain") {
			expectedEventArrived = true
		}
	}
	h.Assert(t, expectedEventArrived, "PodEvicted event was not emitted")
}

func TestDrainCordonNodeFailure(t *testing.T) {
	fakeRecorder := record.NewFakeRecorder(recorderBufferSize)