224 lines
6.8 KiB
Go
224 lines
6.8 KiB
Go
/*
|
|
Copyright 2016 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 priorities
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
framework "k8s.io/kubernetes/pkg/scheduler/framework/v1alpha1"
|
|
nodeinfosnapshot "k8s.io/kubernetes/pkg/scheduler/nodeinfo/snapshot"
|
|
)
|
|
|
|
func TestMostRequested(t *testing.T) {
|
|
labels1 := map[string]string{
|
|
"foo": "bar",
|
|
"baz": "blah",
|
|
}
|
|
labels2 := map[string]string{
|
|
"bar": "foo",
|
|
"baz": "blah",
|
|
}
|
|
noResources := v1.PodSpec{
|
|
Containers: []v1.Container{},
|
|
}
|
|
cpuOnly := v1.PodSpec{
|
|
NodeName: "machine1",
|
|
Containers: []v1.Container{
|
|
{
|
|
Resources: v1.ResourceRequirements{
|
|
Requests: v1.ResourceList{
|
|
v1.ResourceCPU: resource.MustParse("1000m"),
|
|
v1.ResourceMemory: resource.MustParse("0"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Resources: v1.ResourceRequirements{
|
|
Requests: v1.ResourceList{
|
|
v1.ResourceCPU: resource.MustParse("2000m"),
|
|
v1.ResourceMemory: resource.MustParse("0"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
cpuOnly2 := cpuOnly
|
|
cpuOnly2.NodeName = "machine2"
|
|
cpuAndMemory := v1.PodSpec{
|
|
NodeName: "machine2",
|
|
Containers: []v1.Container{
|
|
{
|
|
Resources: v1.ResourceRequirements{
|
|
Requests: v1.ResourceList{
|
|
v1.ResourceCPU: resource.MustParse("1000m"),
|
|
v1.ResourceMemory: resource.MustParse("2000"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Resources: v1.ResourceRequirements{
|
|
Requests: v1.ResourceList{
|
|
v1.ResourceCPU: resource.MustParse("2000m"),
|
|
v1.ResourceMemory: resource.MustParse("3000"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
bigCPUAndMemory := v1.PodSpec{
|
|
NodeName: "machine1",
|
|
Containers: []v1.Container{
|
|
{
|
|
Resources: v1.ResourceRequirements{
|
|
Requests: v1.ResourceList{
|
|
v1.ResourceCPU: resource.MustParse("2000m"),
|
|
v1.ResourceMemory: resource.MustParse("4000"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Resources: v1.ResourceRequirements{
|
|
Requests: v1.ResourceList{
|
|
v1.ResourceCPU: resource.MustParse("3000m"),
|
|
v1.ResourceMemory: resource.MustParse("5000"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
tests := []struct {
|
|
pod *v1.Pod
|
|
pods []*v1.Pod
|
|
nodes []*v1.Node
|
|
expectedList framework.NodeScoreList
|
|
name string
|
|
}{
|
|
{
|
|
/*
|
|
Node1 scores (used resources) on 0-10 scale
|
|
CPU Score: (0 * 100) / 4000 = 0
|
|
Memory Score: (0 * 100) / 10000 = 0
|
|
Node1 Score: (0 + 0) / 2 = 0
|
|
|
|
Node2 scores (used resources) on 0-10 scale
|
|
CPU Score: (0 * 100) / 4000 = 0
|
|
Memory Score: (0 * 100) / 10000 = 0
|
|
Node2 Score: (0 + 0) / 2 = 0
|
|
*/
|
|
pod: &v1.Pod{Spec: noResources},
|
|
nodes: []*v1.Node{makeNode("machine1", 4000, 10000), makeNode("machine2", 4000, 10000)},
|
|
expectedList: []framework.NodeScore{{Name: "machine1", Score: 0}, {Name: "machine2", Score: 0}},
|
|
name: "nothing scheduled, nothing requested",
|
|
},
|
|
{
|
|
/*
|
|
Node1 scores on 0-10 scale
|
|
CPU Score: (3000 * 100) / 4000 = 75
|
|
Memory Score: (5000 * 100) / 10000 = 50
|
|
Node1 Score: (75 + 50) / 2 = 6
|
|
|
|
Node2 scores on 0-10 scale
|
|
CPU Score: (3000 * 100) / 6000 = 50
|
|
Memory Score: (5000 * 100) / 10000 = 50
|
|
Node2 Score: (50 + 50) / 2 = 50
|
|
*/
|
|
pod: &v1.Pod{Spec: cpuAndMemory},
|
|
nodes: []*v1.Node{makeNode("machine1", 4000, 10000), makeNode("machine2", 6000, 10000)},
|
|
expectedList: []framework.NodeScore{{Name: "machine1", Score: 62}, {Name: "machine2", Score: 50}},
|
|
name: "nothing scheduled, resources requested, differently sized machines",
|
|
},
|
|
{
|
|
/*
|
|
Node1 scores on 0-10 scale
|
|
CPU Score: (6000 * 100) / 10000 = 60
|
|
Memory Score: (0 * 100) / 20000 = 100
|
|
Node1 Score: (60 + 0) / 2 = 30
|
|
|
|
Node2 scores on 0-10 scale
|
|
CPU Score: (6000 * 100) / 10000 = 60
|
|
Memory Score: (5000 * 100) / 20000 = 25
|
|
Node2 Score: (60 + 25) / 2 = 42
|
|
*/
|
|
pod: &v1.Pod{Spec: noResources},
|
|
nodes: []*v1.Node{makeNode("machine1", 10000, 20000), makeNode("machine2", 10000, 20000)},
|
|
expectedList: []framework.NodeScore{{Name: "machine1", Score: 30}, {Name: "machine2", Score: 42}},
|
|
name: "no resources requested, pods scheduled with resources",
|
|
pods: []*v1.Pod{
|
|
{Spec: cpuOnly, ObjectMeta: metav1.ObjectMeta{Labels: labels2}},
|
|
{Spec: cpuOnly, ObjectMeta: metav1.ObjectMeta{Labels: labels1}},
|
|
{Spec: cpuOnly2, ObjectMeta: metav1.ObjectMeta{Labels: labels1}},
|
|
{Spec: cpuAndMemory, ObjectMeta: metav1.ObjectMeta{Labels: labels1}},
|
|
},
|
|
},
|
|
{
|
|
/*
|
|
Node1 scores on 0-10 scale
|
|
CPU Score: (6000 * 100) / 10000 = 60
|
|
Memory Score: (5000 * 100) / 20000 = 25
|
|
Node1 Score: (60 + 25) / 2 = 42
|
|
|
|
Node2 scores on 0-10 scale
|
|
CPU Score: (6000 * 100) / 10000 = 60
|
|
Memory Score: (10000 * 100) / 20000 = 50
|
|
Node2 Score: (60 + 50) / 2 = 55
|
|
*/
|
|
pod: &v1.Pod{Spec: cpuAndMemory},
|
|
nodes: []*v1.Node{makeNode("machine1", 10000, 20000), makeNode("machine2", 10000, 20000)},
|
|
expectedList: []framework.NodeScore{{Name: "machine1", Score: 42}, {Name: "machine2", Score: 55}},
|
|
name: "resources requested, pods scheduled with resources",
|
|
pods: []*v1.Pod{
|
|
{Spec: cpuOnly},
|
|
{Spec: cpuAndMemory},
|
|
},
|
|
},
|
|
{
|
|
/*
|
|
Node1 scores on 0-10 scale
|
|
CPU Score: 5000 > 4000 return 0
|
|
Memory Score: (9000 * 100) / 10000 = 90
|
|
Node1 Score: (0 + 90) / 2 = 45
|
|
|
|
Node2 scores on 0-10 scale
|
|
CPU Score: (5000 * 100) / 10000 = 50
|
|
Memory Score: 9000 > 8000 return 0
|
|
Node2 Score: (50 + 0) / 2 = 25
|
|
*/
|
|
pod: &v1.Pod{Spec: bigCPUAndMemory},
|
|
nodes: []*v1.Node{makeNode("machine1", 4000, 10000), makeNode("machine2", 10000, 8000)},
|
|
expectedList: []framework.NodeScore{{Name: "machine1", Score: 45}, {Name: "machine2", Score: 25}},
|
|
name: "resources requested with more than the node, pods scheduled with resources",
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
snapshot := nodeinfosnapshot.NewSnapshot(nodeinfosnapshot.CreateNodeInfoMap(test.pods, test.nodes))
|
|
list, err := runMapReducePriority(MostRequestedPriorityMap, nil, nil, test.pod, snapshot, test.nodes)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if !reflect.DeepEqual(test.expectedList, list) {
|
|
t.Errorf("expected %#v, got %#v", test.expectedList, list)
|
|
}
|
|
})
|
|
}
|
|
}
|