1
0
mirror of https://github.com/kubernetes-sigs/descheduler.git synced 2026-01-28 06:29:29 +01:00

refactor(TestRemovePodsViolatingNodeAffinity): add constants for node names

This commit is contained in:
Jan Chaloupka
2025-12-15 15:28:03 +01:00
parent a2ffbc1261
commit 183a138d82

View File

@@ -32,6 +32,12 @@ import (
"sigs.k8s.io/descheduler/test" "sigs.k8s.io/descheduler/test"
) )
const (
nodeWithLabelsName = "nodeWithLabels"
nodeWithoutLabelsName = "nodeWithoutLabels"
unschedulableNodeWithLabelsName = "unschedulableNodeWithLabels"
)
func buildTestNode(name string, apply func(*v1.Node)) *v1.Node { func buildTestNode(name string, apply func(*v1.Node)) *v1.Node {
return test.BuildTestNode(name, 2000, 3000, 10, apply) return test.BuildTestNode(name, 2000, 3000, 10, apply)
} }
@@ -43,13 +49,13 @@ func buildTestPod(name string, nodeName string, apply func(*v1.Pod)) *v1.Pod {
func TestRemovePodsViolatingNodeAffinity(t *testing.T) { func TestRemovePodsViolatingNodeAffinity(t *testing.T) {
nodeLabelKey := "kubernetes.io/desiredNode" nodeLabelKey := "kubernetes.io/desiredNode"
nodeLabelValue := "yes" nodeLabelValue := "yes"
nodeWithLabels := buildTestNode("nodeWithLabels", func(node *v1.Node) { nodeWithLabels := buildTestNode(nodeWithLabelsName, func(node *v1.Node) {
node.Labels[nodeLabelKey] = nodeLabelValue node.Labels[nodeLabelKey] = nodeLabelValue
}) })
nodeWithoutLabels := buildTestNode("nodeWithoutLabels", nil) nodeWithoutLabels := buildTestNode(nodeWithoutLabelsName, nil)
unschedulableNodeWithLabels := buildTestNode("unschedulableNodeWithLabels", func(node *v1.Node) { unschedulableNodeWithLabels := buildTestNode(unschedulableNodeWithLabelsName, func(node *v1.Node) {
node.Labels[nodeLabelKey] = nodeLabelValue node.Labels[nodeLabelKey] = nodeLabelValue
node.Spec.Unschedulable = true node.Spec.Unschedulable = true
}) })