Function TestTaintOutOfService

pkg/node/node_test.go:508–543

(t *testing.T)

}

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

	tNode, err := newNode(config.Config{EnableOutOfServiceTaint: true}, client)
	h.Ok(t, err)
	h.Equals(t, true, tNode.GetNthConfig().EnableOutOfServiceTaint)
	h.Equals(t, false, tNode.GetNthConfig().CordonOnly)

	err = tNode.TaintOutOfService(nodeName)
	h.Ok(t, err)

	updatedNode, err := client.CoreV1().Nodes().Get(context.Background(), nodeName, metav1.GetOptions{})
	h.Ok(t, err)
	taintFound := false
	expectedTaint := v1.Taint{
		Key:    outOfServiceTaintKey,
		Value:  outOfServiceTaintValue,
		Effect: v1.TaintEffectNoExecute,
	}
	for _, taint := range updatedNode.Spec.Taints {
		if taint.Key == expectedTaint.Key &&
			taint.Value == expectedTaint.Value &&
			taint.Effect == expectedTaint.Effect {
			taintFound = true
			break
		}
	}
	h.Equals(t, true, taintFound)
}