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

refactor(TestDefaultEvictor): add setNodeTaint helper function

Browse Source
This commit is contained in:
Jan Chaloupka
2025-12-20 13:54:50 +01:00
parent d666e4b830
commit b8765bd8ee
1 changed files with 25 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
pkg/framework/plugins/defaultevictor/defaultevictor_test.go
View File

@@ -77,6 +77,16 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
nodeLabelKey := "datacenter"
nodeLabelValue := "east"
setNodeTaint := func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}
testCases := []testCase{
{
description: "Pod with no tolerations running on normal node, all other nodes tainted",
@@ -86,24 +96,8 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
}),
},
nodes: []*v1.Node{
buildTestNode("node2", func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}),
buildTestNode("node3", func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}),
buildTestNode("node2", setNodeTaint),
buildTestNode("node3", setNodeTaint),
},
nodeFit: true,
}, {
@@ -121,24 +115,8 @@ func TestDefaultEvictorPreEvictionFilter(t *testing.T) {
}),
},
nodes: []*v1.Node{
buildTestNode("node2", func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}),
buildTestNode("node3", func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}),
buildTestNode("node2", setNodeTaint),
buildTestNode("node3", setNodeTaint),
},
nodeFit: true,
result: true,
@@ -330,8 +308,15 @@ func TestDefaultEvictorFilter(t *testing.T) {
minPodAge := metav1.Duration{Duration: 50 * time.Minute}
nodeTaintKey := "hardware"
nodeTaintValue := "gpu"
setNodeTaint := func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: "hardware",
Value: "gpu",
Effect: v1.TaintEffectNoSchedule,
},
}
}
ownerRefUUID := uuid.NewUUID()
@@ -645,24 +630,8 @@ func TestDefaultEvictorFilter(t *testing.T) {
}),
},
nodes: []*v1.Node{
buildTestNode("node2", func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}),
buildTestNode("node3", func(node *v1.Node) {
node.Spec.Taints = []v1.Taint{
{
Key: nodeTaintKey,
Value: nodeTaintValue,
Effect: v1.TaintEffectNoSchedule,
},
}
}),
buildTestNode("node2", setNodeTaint),
buildTestNode("node3", setNodeTaint),
},
nodeFit: true,
result: true,