Sok3t/descheduler
1
0
Fork 0
mirror of https://github.com/kubernetes-sigs/descheduler.git synced 2026-01-26 05:14:13 +01:00
Code Packages Releases Activity

refactor(TestDefaultEvictor): add setPodNodeSelector helper function

Browse Source
This commit is contained in:
Jan Chaloupka
2025-12-20 14:11:46 +01:00
parent dfd2b95d2d
commit 3068f8431a
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/framework/plugins/defaultevictor/defaultevictor_test.go
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{ testCases := []testCase{
{ {
description: "Pod with no tolerations running on normal node, all other nodes tainted", 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{ pods: []*v1.Pod{
buildTestPod("p1", n1.Name, func(pod *v1.Pod) { buildTestPod("p1", n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod) test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{ setPodNodeSelector(pod)
nodeLabelKey: nodeLabelValue,
}
}), }),
}, },
nodes: []*v1.Node{ nodes: []*v1.Node{
@@ -160,9 +164,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{ pods: []*v1.Pod{
test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) { test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod) test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{ setPodNodeSelector(pod)
nodeLabelKey: nodeLabelValue,
}
}), }),
}, },
nodes: []*v1.Node{ nodes: []*v1.Node{
@@ -175,9 +177,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{ pods: []*v1.Pod{
test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) { test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod) test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{ setPodNodeSelector(pod)
nodeLabelKey: nodeLabelValue,
}
}), }),
test.BuildTestPod("node2-pod-10GB-mem", 20, 10, "node2", func(pod *v1.Pod) { test.BuildTestPod("node2-pod-10GB-mem", 20, 10, "node2", func(pod *v1.Pod) {
pod.ObjectMeta.Labels = map[string]string{ pod.ObjectMeta.Labels = map[string]string{
@@ -201,9 +201,7 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
pods: []*v1.Pod{ pods: []*v1.Pod{
test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) { test.BuildTestPod("p1", 12, 8, n1.Name, func(pod *v1.Pod) {
test.SetNormalOwnerRef(pod) test.SetNormalOwnerRef(pod)
pod.Spec.NodeSelector = map[string]string{ setPodNodeSelector(pod)
nodeLabelKey: nodeLabelValue,
}
}), }),
test.BuildTestPod("node2-pod-10GB-mem", 10, 10, "node2", func(pod *v1.Pod) { test.BuildTestPod("node2-pod-10GB-mem", 10, 10, "node2", func(pod *v1.Pod) {
pod.ObjectMeta.Labels = map[string]string{ pod.ObjectMeta.Labels = map[string]string{