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

refactor(node_taint_test): inline addTaintsToNode

Browse Source
This commit is contained in:
Jan Chaloupka
2025-12-15 17:08:50 +01:00
parent b9ceb9144f
commit f65209d4fa
1 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pkg/framework/plugins/removepodsviolatingnodetaints/node_taint_test.go
View File

@@ -68,15 +68,6 @@ func createPreferNoScheduleTaint(key, value string, index int) v1.Taint {
} }
} }
func addTaintsToNode(node *v1.Node, key, value string, indices []int) *v1.Node {
taints := []v1.Taint{}
for _, index := range indices {
taints = append(taints, createNoScheduleTaint(key, value, index))
}
node.Spec.Taints = taints
return node
}
func addTolerationToPod(pod *v1.Pod, key, value string, index int, effect v1.TaintEffect) *v1.Pod { func addTolerationToPod(pod *v1.Pod, key, value string, index int, effect v1.TaintEffect) *v1.Pod {
if pod.Annotations == nil { if pod.Annotations == nil {
pod.Annotations = map[string]string{} pod.Annotations = map[string]string{}
@@ -89,10 +80,14 @@ func addTolerationToPod(pod *v1.Pod, key, value string, index int, effect v1.Tai
func TestDeletePodsViolatingNodeTaints(t *testing.T) { func TestDeletePodsViolatingNodeTaints(t *testing.T) {
node1 := buildTestNode(nodeName1, func(node *v1.Node) { node1 := buildTestNode(nodeName1, func(node *v1.Node) {
addTaintsToNode(node, "testTaint", "test", []int{1}) node.Spec.Taints = []v1.Taint{
createNoScheduleTaint("testTaint", "test", 1),
}
}) })
node2 := buildTestNode(nodeName2, func(node *v1.Node) { node2 := buildTestNode(nodeName2, func(node *v1.Node) {
addTaintsToNode(node, "testingTaint", "testing", []int{1}) node.Spec.Taints = []v1.Taint{
createNoScheduleTaint("testingTaint", "testing", 1),
}
}) })
node3 := buildTestNode(nodeName3, func(node *v1.Node) { node3 := buildTestNode(nodeName3, func(node *v1.Node) {
@@ -119,8 +114,10 @@ func TestDeletePodsViolatingNodeTaints(t *testing.T) {
}) })
node7 := buildTestNode(nodeName7, func(node *v1.Node) { node7 := buildTestNode(nodeName7, func(node *v1.Node) {
addTaintsToNode(node, "testTaint", "test", []int{1}) node.Spec.Taints = []v1.Taint{
addTaintsToNode(node, "testingTaint", "testing", []int{1}) createNoScheduleTaint("testTaint", "test", 1),
createNoScheduleTaint("testingTaint", "testing", 1),
}
}) })
p1 := buildTestPod("p1", nodeName1, func(pod *v1.Pod) { p1 := buildTestPod("p1", nodeName1, func(pod *v1.Pod) {