github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-23 11:50:44 +00:00
Code Issues Projects Releases Wiki Activity

chore(scheduler): add filter integration tests for missing part plugins: TaintToleration plugin

Browse Source
This commit is contained in:
googs1025 2025-03-02 15:41:29 +08:00
parent 3a812ec464
commit 032b05114c
2 changed files with 185 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/scheduler/testing/wrappers.go
View File

@ -409,6 +409,12 @@ func (p *PodWrapper) Node(s string) *PodWrapper {
return p
}
// Tolerations sets `tolerations` as the tolerations of the inner pod.
func (p *PodWrapper) Tolerations(tolerations []v1.Toleration) *PodWrapper {
p.Spec.Tolerations = tolerations
return p
}
// NodeSelector sets `m` as the nodeSelector of the inner pod.
func (p *PodWrapper) NodeSelector(m map[string]string) *PodWrapper {
p.Spec.NodeSelector = m

179
test/integration/scheduler/filters/filters_test.go
View File

@ -1350,6 +1350,185 @@ func TestInterPodAffinityWithNamespaceSelector(t *testing.T) {
}
}
func TestTaintTolerationFilter(t *testing.T) {
pause := imageutils.GetPauseImageName()
tests := []struct {
name string
nodes []*v1.Node
incomingPod *v1.Pod
fit bool
}{
{
name: "Pod tolerates node taint",
nodes: []*v1.Node{
st.MakeNode().Name("node-1").
Taints([]v1.Taint{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
}).Obj(),
},
incomingPod: st.MakePod().Name("pod-1").
Tolerations([]v1.Toleration{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
}).Container(pause).
Obj(),
fit: true,
},
{
name: "Pod does not tolerate node taint",
nodes: []*v1.Node{st.MakeNode().Name("node-2").
Taints([]v1.Taint{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
}).Obj(),
},
incomingPod: st.MakePod().Name("pod-2").Container(pause).Obj(),
fit: false,
},
{
name: "Pod tolerates some but not all node taints",
nodes: []*v1.Node{st.MakeNode().Name("node-3").Taints([]v1.Taint{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
{
Key: "example.com/taint2",
Value: "value2",
Effect: v1.TaintEffectNoSchedule,
},
}).Obj(),
},
incomingPod: st.MakePod().Name("pod-3").Tolerations([]v1.Toleration{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
}).Container(pause).
Obj(),
fit: false,
},
{
name: "Pod tolerates all node taints",
nodes: []*v1.Node{
st.MakeNode().Name("node-4").Taints([]v1.Taint{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
{
Key: "example.com/taint2",
Value: "value2",
Effect: v1.TaintEffectNoSchedule,
},
}).Obj(),
},
incomingPod: st.MakePod().Name("pod-4").Tolerations([]v1.Toleration{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
{
Key: "example.com/taint2",
Value: "value2",
Effect: v1.TaintEffectNoSchedule,
},
}).Container(pause).Obj(),
fit: true,
},
{
name: "Node has one taint, pod has multiple tolerations and tolerates all node taints",
nodes: []*v1.Node{
st.MakeNode().Name("node-1").
Taints([]v1.Taint{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
}).Obj(),
},
incomingPod: st.MakePod().Name("pod-5").Tolerations([]v1.Toleration{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
{
Key: "example.com/taint2",
Value: "value2",
Effect: v1.TaintEffectNoSchedule,
},
}).Container(pause).Obj(),
fit: true,
},
{
name: "Node has no taint, pod has one toleration",
nodes: []*v1.Node{
st.MakeNode().Name("node-6").Obj(),
},
incomingPod: st.MakePod().Name("pod-6").Tolerations([]v1.Toleration{
{
Key: "example.com/taint1",
Value: "value1",
Effect: v1.TaintEffectNoSchedule,
},
}).Container(pause).Obj(),
fit: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
testCtx := initTest(t, "taint-toleration-filter")
cs := testCtx.ClientSet
ns := testCtx.NS.Name
for i := range tt.nodes {
if _, err := createNode(cs, tt.nodes[i]); err != nil {
t.Fatalf("Failed to create node: %v", err)
}
}
// set namespace to pods
tt.incomingPod.SetNamespace(ns)
defer testutils.CleanupPods(testCtx.Ctx, cs, t, []*v1.Pod{tt.incomingPod})
testPod, err := cs.CoreV1().Pods(tt.incomingPod.Namespace).Create(testCtx.Ctx, tt.incomingPod, metav1.CreateOptions{})
if err != nil {
t.Fatalf("Failed to create pod during test: %v", err)
}
if tt.fit {
err = wait.PollUntilContextTimeout(testCtx.Ctx, pollInterval, wait.ForeverTestTimeout, false,
podScheduled(cs, testPod.Namespace, testPod.Name))
if err != nil {
t.Errorf("Test Failed: Expected pod %s/%s to be scheduled but got error: %v", testPod.Namespace, testPod.Name, err)
}
} else {
err = wait.PollUntilContextTimeout(testCtx.Ctx, pollInterval, wait.ForeverTestTimeout, false,
podUnschedulable(cs, testPod.Namespace, testPod.Name))
if err != nil {
t.Errorf("Test Failed: Expected pod %s/%s to be unschedulable but got error: %v", testPod.Namespace, testPod.Name, err)
}
}
})
}
}
// TestPodTopologySpreadFilter verifies that EvenPodsSpread predicate functions well.
func TestPodTopologySpreadFilter(t *testing.T) {
pause := imageutils.GetPauseImageName()