|
|
|
@@ -0,0 +1,479 @@
|
|
|
|
|
/*
|
|
|
|
|
Copyright 2019 The Kubernetes Authors.
|
|
|
|
|
|
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
|
you may not use this file except in compliance with the License.
|
|
|
|
|
You may obtain a copy of the License at
|
|
|
|
|
|
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
|
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
See the License for the specific language governing permissions and
|
|
|
|
|
limitations under the License.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
package podtopologyspread
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
|
|
"k8s.io/api/core/v1"
|
|
|
|
|
"k8s.io/kubernetes/pkg/scheduler/algorithm/predicates"
|
|
|
|
|
"k8s.io/kubernetes/pkg/scheduler/framework/plugins/migration"
|
|
|
|
|
"k8s.io/kubernetes/pkg/scheduler/framework/v1alpha1"
|
|
|
|
|
schedulernodeinfo "k8s.io/kubernetes/pkg/scheduler/nodeinfo"
|
|
|
|
|
st "k8s.io/kubernetes/pkg/scheduler/testing"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
var hardSpread = v1.DoNotSchedule
|
|
|
|
|
|
|
|
|
|
func TestPodTopologySpreadFilter_SingleConstraint(t *testing.T) {
|
|
|
|
|
tests := []struct {
|
|
|
|
|
name string
|
|
|
|
|
pod *v1.Pod
|
|
|
|
|
nodes []*v1.Node
|
|
|
|
|
existingPods []*v1.Pod
|
|
|
|
|
fits map[string]bool
|
|
|
|
|
}{
|
|
|
|
|
{
|
|
|
|
|
name: "no existing pods",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": true,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
name: "no existing pods, incoming pod doesn't match itself",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
1, "zone", hardSpread, st.MakeLabelSelector().Exists("bar").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": true,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
name: "existing pods with mis-matched namespace doens't count",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Namespace("ns1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Namespace("ns2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-x1").Node("node-x").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": false,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
name: "pods spread across zones as 3/3, all nodes fit",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": true,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// TODO(Huang-Wei): maybe document this to remind users that typos on node labels
|
|
|
|
|
// can cause unexpected behavior
|
|
|
|
|
name: "pods spread across zones as 1/2 due to absence of label 'zone' on node-b",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zon", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-x1").Node("node-x").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": false,
|
|
|
|
|
"node-x": false,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
name: "pods spread across nodes as 2/1/0/3, only node-x fits",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
1, "node", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": false,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
name: "pods spread across nodes as 2/1/0/3, maxSkew is 2, node-b and node-x fit",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").SpreadConstraint(
|
|
|
|
|
2, "node", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// not a desired case, but it can happen
|
|
|
|
|
// TODO(Huang-Wei): document this "pod-not-match-itself" case
|
|
|
|
|
// in this case, placement of the new pod doesn't change pod distribution of the cluster
|
|
|
|
|
// as the incoming pod doesn't have label "foo"
|
|
|
|
|
name: "pods spread across nodes as 2/1/0/3, but pod doesn't match itself",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("bar", "").SpreadConstraint(
|
|
|
|
|
1, "node", hardSpread, st.MakeLabelSelector().Exists("foo").Obj(),
|
|
|
|
|
).Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// only node-a and node-y are considered, so pods spread as 2/~1~/~0~/3
|
|
|
|
|
// ps: '~num~' is a markdown symbol to denote a crossline through 'num'
|
|
|
|
|
// but in this unit test, we don't run NodeAffinityPredicate, so node-b and node-x are
|
|
|
|
|
// still expected to be fits;
|
|
|
|
|
// the fact that node-a fits can prove the underlying logic works
|
|
|
|
|
name: "incoming pod has nodeAffinity, pods spread as 2/~1~/~0~/3, hence node-a fits",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").
|
|
|
|
|
NodeAffinityIn("node", []string{"node-a", "node-y"}).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": true, // in real case, it's false
|
|
|
|
|
"node-x": true, // in real case, it's false
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
}
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
|
nodeInfoMap := schedulernodeinfo.CreateNodeNameToInfoMap(tt.existingPods, tt.nodes)
|
|
|
|
|
meta := predicates.GetPredicateMetadata(tt.pod, nodeInfoMap)
|
|
|
|
|
state := v1alpha1.NewCycleState()
|
|
|
|
|
state.Write(migration.PredicatesStateKey, &migration.PredicatesStateData{Reference: meta})
|
|
|
|
|
plugin, _ := New(nil, nil)
|
|
|
|
|
for _, node := range tt.nodes {
|
|
|
|
|
status := plugin.(*PodTopologySpread).Filter(context.Background(), state, tt.pod, nodeInfoMap[node.Name])
|
|
|
|
|
if status.IsSuccess() != tt.fits[node.Name] {
|
|
|
|
|
t.Errorf("[%s]: expected %v got %v", node.Name, tt.fits[node.Name], status.IsSuccess())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func TestPodTopologySpreadFilter_MultipleConstraints(t *testing.T) {
|
|
|
|
|
tests := []struct {
|
|
|
|
|
name string
|
|
|
|
|
pod *v1.Pod
|
|
|
|
|
nodes []*v1.Node
|
|
|
|
|
existingPods []*v1.Pod
|
|
|
|
|
fits map[string]bool
|
|
|
|
|
}{
|
|
|
|
|
{
|
|
|
|
|
// 1. to fulfil "zone" constraint, incoming pod can be placed on any zone (hence any node)
|
|
|
|
|
// 2. to fulfil "node" constraint, incoming pod can be placed on node-x
|
|
|
|
|
// intersection of (1) and (2) returns node-x
|
|
|
|
|
name: "two constraints on zone and node, spreads = [3/3, 2/1/0/3]",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").
|
|
|
|
|
SpreadConstraint(1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": false,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// 1. to fulfil "zone" constraint, incoming pod can be placed on zone1 (node-a or node-b)
|
|
|
|
|
// 2. to fulfil "node" constraint, incoming pod can be placed on node-x
|
|
|
|
|
// intersection of (1) and (2) returns no node
|
|
|
|
|
name: "two constraints on zone and node, spreads = [3/4, 2/1/0/4]",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").
|
|
|
|
|
SpreadConstraint(1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-b1").Node("node-b").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y4").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": false,
|
|
|
|
|
"node-x": false,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// 1. to fulfil "zone" constraint, incoming pod can be placed on zone2 (node-x or node-y)
|
|
|
|
|
// 2. to fulfil "node" constraint, incoming pod can be placed on node-b or node-x
|
|
|
|
|
// intersection of (1) and (2) returns node-x
|
|
|
|
|
name: "constraints hold different labelSelectors, spreads = [1/0, 1/0/0/1]",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").Label("bar", "").
|
|
|
|
|
SpreadConstraint(1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("bar").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("bar", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": false,
|
|
|
|
|
"node-x": true,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// 1. to fulfil "zone" constraint, incoming pod can be placed on zone2 (node-x or node-y)
|
|
|
|
|
// 2. to fulfil "node" constraint, incoming pod can be placed on node-a or node-b
|
|
|
|
|
// intersection of (1) and (2) returns no node
|
|
|
|
|
name: "constraints hold different labelSelectors, spreads = [1/0, 0/0/1/1]",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").Label("bar", "").
|
|
|
|
|
SpreadConstraint(1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("bar").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-x1").Node("node-x").Label("bar", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("bar", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": false,
|
|
|
|
|
"node-x": false,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// 1. to fulfil "zone" constraint, incoming pod can be placed on zone1 (node-a or node-b)
|
|
|
|
|
// 2. to fulfil "node" constraint, incoming pod can be placed on node-b or node-x
|
|
|
|
|
// intersection of (1) and (2) returns node-b
|
|
|
|
|
name: "constraints hold different labelSelectors, spreads = [2/3, 1/0/0/1]",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("foo", "").Label("bar", "").
|
|
|
|
|
SpreadConstraint(1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("bar").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-a2").Node("node-a").Label("foo", "").Label("bar", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y2").Node("node-y").Label("foo", "").Label("bar", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y3").Node("node-y").Label("foo", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": false,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": false,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
// 1. pod doesn't match itself on "zone" constraint, so it can be put onto any zone
|
|
|
|
|
// 2. to fulfil "node" constraint, incoming pod can be placed on node-a or node-b
|
|
|
|
|
// intersection of (1) and (2) returns node-a and node-b
|
|
|
|
|
name: "constraints hold different labelSelectors but pod doesn't match itself on 'zone' constraint",
|
|
|
|
|
pod: st.MakePod().Name("p").Label("bar", "").
|
|
|
|
|
SpreadConstraint(1, "zone", hardSpread, st.MakeLabelSelector().Exists("foo").Obj()).
|
|
|
|
|
SpreadConstraint(1, "node", hardSpread, st.MakeLabelSelector().Exists("bar").Obj()).
|
|
|
|
|
Obj(),
|
|
|
|
|
nodes: []*v1.Node{
|
|
|
|
|
st.MakeNode().Name("node-a").Label("zone", "zone1").Label("node", "node-a").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-b").Label("zone", "zone1").Label("node", "node-b").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-x").Label("zone", "zone2").Label("node", "node-x").Obj(),
|
|
|
|
|
st.MakeNode().Name("node-y").Label("zone", "zone2").Label("node", "node-y").Obj(),
|
|
|
|
|
},
|
|
|
|
|
existingPods: []*v1.Pod{
|
|
|
|
|
st.MakePod().Name("p-a1").Node("node-a").Label("foo", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-x1").Node("node-x").Label("bar", "").Obj(),
|
|
|
|
|
st.MakePod().Name("p-y1").Node("node-y").Label("bar", "").Obj(),
|
|
|
|
|
},
|
|
|
|
|
fits: map[string]bool{
|
|
|
|
|
"node-a": true,
|
|
|
|
|
"node-b": true,
|
|
|
|
|
"node-x": false,
|
|
|
|
|
"node-y": false,
|
|
|
|
|
},
|
|
|
|
|
},
|
|
|
|
|
}
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
|
nodeInfoMap := schedulernodeinfo.CreateNodeNameToInfoMap(tt.existingPods, tt.nodes)
|
|
|
|
|
meta := predicates.GetPredicateMetadata(tt.pod, nodeInfoMap)
|
|
|
|
|
state := v1alpha1.NewCycleState()
|
|
|
|
|
state.Write(migration.PredicatesStateKey, &migration.PredicatesStateData{Reference: meta})
|
|
|
|
|
plugin, _ := New(nil, nil)
|
|
|
|
|
for _, node := range tt.nodes {
|
|
|
|
|
status := plugin.(*PodTopologySpread).Filter(context.Background(), state, tt.pod, nodeInfoMap[node.Name])
|
|
|
|
|
if status.IsSuccess() != tt.fits[node.Name] {
|
|
|
|
|
t.Errorf("[%s]: expected %v got %v", node.Name, tt.fits[node.Name], status.IsSuccess())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
}
|