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

refactor(TestDefaultEvictor): add setPodNodeSelector helper function

This commit is contained in:
Jan Chaloupka
2025-12-20 14:11:46 +01:00
parent dfd2b95d2d
commit 3068f8431a

View File

@@ -93,6 +93,12 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
}
}
setPodNodeSelector := func(pod *v1.Pod) {
pod.Spec.NodeSelector = map[string]string{
nodeLabelKey: nodeLabelValue,
}
}
testCases := []testCase{
{
description: "Pod with no tolerations running on normal node, all other nodes tainted",
@@ -144,9 +150,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{
buildTestPod("p1", n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{
nodeLabelKey: nodeLabelValue,
}
setPodNodeSelector(pod)
}),
},
nodes: []*v1.Node{
@@ -160,9 +164,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{
test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{
nodeLabelKey: nodeLabelValue,
}
setPodNodeSelector(pod)
}),
},
nodes: []*v1.Node{
@@ -175,9 +177,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{
test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{
nodeLabelKey: nodeLabelValue,
}
setPodNodeSelector(pod)
}),
test.BuildTestPod("node2-pod-10GB-mem", 20, 10, "node2", func(pod *v1.Pod) {
pod.ObjectMeta.Labels = map[string]string{
@@ -201,9 +201,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{
test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{
nodeLabelKey: nodeLabelValue,
}
setPodNodeSelector(pod)
}),
test.BuildTestPod("node2-pod-10GB-mem", 10, 10, "node2", func(pod *v1.Pod) {
pod.ObjectMeta.Labels = map[string]string{