mirror of
https://github.com/kubernetes-sigs/descheduler.git
synced 2026-01-28 14:41:10 +01:00
277 lines
9.0 KiB
Go
277 lines
9.0 KiB
Go
/*
|
|
Copyright 2020 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 strategies
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
policyv1 "k8s.io/api/policy/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
core "k8s.io/client-go/testing"
|
|
"sigs.k8s.io/descheduler/pkg/api"
|
|
"sigs.k8s.io/descheduler/pkg/descheduler/evictions"
|
|
"sigs.k8s.io/descheduler/test"
|
|
)
|
|
|
|
func TestPodLifeTime(t *testing.T) {
|
|
ctx := context.Background()
|
|
node1 := test.BuildTestNode("n1", 2000, 3000, 10, nil)
|
|
olderPodCreationTime := metav1.NewTime(time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC))
|
|
newerPodCreationTime := metav1.NewTime(time.Now())
|
|
|
|
// Setup pods, one should be evicted
|
|
p1 := test.BuildTestPod("p1", 100, 0, node1.Name, nil)
|
|
p1.Namespace = "dev"
|
|
p1.ObjectMeta.CreationTimestamp = newerPodCreationTime
|
|
p2 := test.BuildTestPod("p2", 100, 0, node1.Name, nil)
|
|
p2.Namespace = "dev"
|
|
p2.ObjectMeta.CreationTimestamp = olderPodCreationTime
|
|
|
|
ownerRef1 := test.GetReplicaSetOwnerRefList()
|
|
p1.ObjectMeta.OwnerReferences = ownerRef1
|
|
p2.ObjectMeta.OwnerReferences = ownerRef1
|
|
|
|
// Setup pods, zero should be evicted
|
|
p3 := test.BuildTestPod("p3", 100, 0, node1.Name, nil)
|
|
p3.Namespace = "dev"
|
|
p3.ObjectMeta.CreationTimestamp = newerPodCreationTime
|
|
p4 := test.BuildTestPod("p4", 100, 0, node1.Name, nil)
|
|
p4.Namespace = "dev"
|
|
p4.ObjectMeta.CreationTimestamp = newerPodCreationTime
|
|
|
|
ownerRef2 := test.GetReplicaSetOwnerRefList()
|
|
p3.ObjectMeta.OwnerReferences = ownerRef2
|
|
p4.ObjectMeta.OwnerReferences = ownerRef2
|
|
|
|
// Setup pods, one should be evicted
|
|
p5 := test.BuildTestPod("p5", 100, 0, node1.Name, nil)
|
|
p5.Namespace = "dev"
|
|
p5.ObjectMeta.CreationTimestamp = newerPodCreationTime
|
|
p6 := test.BuildTestPod("p6", 100, 0, node1.Name, nil)
|
|
p6.Namespace = "dev"
|
|
p6.ObjectMeta.CreationTimestamp = metav1.NewTime(time.Now().Add(time.Second * 605))
|
|
|
|
ownerRef3 := test.GetReplicaSetOwnerRefList()
|
|
p5.ObjectMeta.OwnerReferences = ownerRef3
|
|
p6.ObjectMeta.OwnerReferences = ownerRef3
|
|
|
|
// Setup pods, zero should be evicted
|
|
p7 := test.BuildTestPod("p7", 100, 0, node1.Name, nil)
|
|
p7.Namespace = "dev"
|
|
p7.ObjectMeta.CreationTimestamp = newerPodCreationTime
|
|
p8 := test.BuildTestPod("p8", 100, 0, node1.Name, nil)
|
|
p8.Namespace = "dev"
|
|
p8.ObjectMeta.CreationTimestamp = metav1.NewTime(time.Now().Add(time.Second * 595))
|
|
|
|
ownerRef4 := test.GetReplicaSetOwnerRefList()
|
|
p5.ObjectMeta.OwnerReferences = ownerRef4
|
|
p6.ObjectMeta.OwnerReferences = ownerRef4
|
|
|
|
// Setup two old pods with different status phases
|
|
p9 := test.BuildTestPod("p9", 100, 0, node1.Name, nil)
|
|
p9.Namespace = "dev"
|
|
p9.ObjectMeta.CreationTimestamp = olderPodCreationTime
|
|
p10 := test.BuildTestPod("p10", 100, 0, node1.Name, nil)
|
|
p10.Namespace = "dev"
|
|
p10.ObjectMeta.CreationTimestamp = olderPodCreationTime
|
|
|
|
p9.Status.Phase = "Pending"
|
|
p10.Status.Phase = "Running"
|
|
p9.ObjectMeta.OwnerReferences = ownerRef1
|
|
p10.ObjectMeta.OwnerReferences = ownerRef1
|
|
|
|
p11 := test.BuildTestPod("p11", 100, 0, node1.Name, func(pod *v1.Pod) {
|
|
pod.Spec.Volumes = []v1.Volume{
|
|
{
|
|
Name: "pvc", VolumeSource: v1.VolumeSource{
|
|
PersistentVolumeClaim: &v1.PersistentVolumeClaimVolumeSource{ClaimName: "foo"},
|
|
},
|
|
},
|
|
}
|
|
pod.Namespace = "dev"
|
|
pod.ObjectMeta.CreationTimestamp = olderPodCreationTime
|
|
pod.ObjectMeta.OwnerReferences = ownerRef1
|
|
})
|
|
|
|
// Setup two old pods with different labels
|
|
p12 := test.BuildTestPod("p12", 100, 0, node1.Name, nil)
|
|
p12.Namespace = "dev"
|
|
p12.ObjectMeta.CreationTimestamp = olderPodCreationTime
|
|
p13 := test.BuildTestPod("p13", 100, 0, node1.Name, nil)
|
|
p13.Namespace = "dev"
|
|
p13.ObjectMeta.CreationTimestamp = olderPodCreationTime
|
|
|
|
p12.ObjectMeta.Labels = map[string]string{"foo": "bar"}
|
|
p13.ObjectMeta.Labels = map[string]string{"foo": "bar1"}
|
|
p12.ObjectMeta.OwnerReferences = ownerRef1
|
|
p13.ObjectMeta.OwnerReferences = ownerRef1
|
|
|
|
var maxLifeTime uint = 600
|
|
testCases := []struct {
|
|
description string
|
|
strategy api.DeschedulerStrategy
|
|
maxPodsToEvictPerNode int
|
|
pods []v1.Pod
|
|
nodes []*v1.Node
|
|
expectedEvictedPodCount int
|
|
ignorePvcPods bool
|
|
}{
|
|
{
|
|
description: "Two pods in the `dev` Namespace, 1 is new and 1 very is old. 1 should be evicted.",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p1, *p2},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 1,
|
|
},
|
|
{
|
|
description: "Two pods in the `dev` Namespace, 2 are new and 0 are old. 0 should be evicted.",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p3, *p4},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 0,
|
|
},
|
|
{
|
|
description: "Two pods in the `dev` Namespace, 1 created 605 seconds ago. 1 should be evicted.",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p5, *p6},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 1,
|
|
},
|
|
{
|
|
description: "Two pods in the `dev` Namespace, 1 created 595 seconds ago. 0 should be evicted.",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p7, *p8},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 0,
|
|
},
|
|
{
|
|
description: "Two old pods with different status phases. 1 should be evicted.",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{
|
|
MaxPodLifeTimeSeconds: &maxLifeTime,
|
|
PodStatusPhases: []string{"Pending"},
|
|
},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p9, *p10},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 1,
|
|
},
|
|
{
|
|
description: "does not evict pvc pods with ignorePvcPods set to true",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p11},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 0,
|
|
ignorePvcPods: true,
|
|
},
|
|
{
|
|
description: "evicts pvc pods with ignorePvcPods set to false (or unset)",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p11},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 1,
|
|
},
|
|
{
|
|
description: "Two old pods with different labels, 1 selected by labelSelector",
|
|
strategy: api.DeschedulerStrategy{
|
|
Enabled: true,
|
|
Params: &api.StrategyParameters{
|
|
PodLifeTime: &api.PodLifeTime{MaxPodLifeTimeSeconds: &maxLifeTime},
|
|
LabelSelector: &metav1.LabelSelector{
|
|
MatchLabels: map[string]string{"foo": "bar"},
|
|
},
|
|
},
|
|
},
|
|
maxPodsToEvictPerNode: 5,
|
|
pods: []v1.Pod{*p12, *p13},
|
|
nodes: []*v1.Node{node1},
|
|
expectedEvictedPodCount: 1,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
fakeClient := &fake.Clientset{}
|
|
|
|
fakeClient.Fake.AddReactor("list", "pods", func(action core.Action) (bool, runtime.Object, error) {
|
|
return true, &v1.PodList{Items: tc.pods}, nil
|
|
})
|
|
|
|
podEvictor := evictions.NewPodEvictor(
|
|
fakeClient,
|
|
policyv1.SchemeGroupVersion.String(),
|
|
false,
|
|
tc.maxPodsToEvictPerNode,
|
|
tc.nodes,
|
|
false,
|
|
false,
|
|
tc.ignorePvcPods,
|
|
)
|
|
|
|
PodLifeTime(ctx, fakeClient, tc.strategy, tc.nodes, podEvictor)
|
|
podsEvicted := podEvictor.TotalEvicted()
|
|
if podsEvicted != tc.expectedEvictedPodCount {
|
|
t.Errorf("Test error for description: %s. Expected evicted pods count %v, got %v", tc.description, tc.expectedEvictedPodCount, podsEvicted)
|
|
}
|
|
}
|
|
|
|
}
|