mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-10-30 21:30:16 +00:00
82 lines
2.8 KiB
Go
82 lines
2.8 KiB
Go
/*
|
|
Copyright 2017 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 scheduling
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/onsi/ginkgo"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
clientset "k8s.io/client-go/kubernetes"
|
|
"k8s.io/kubernetes/test/e2e/framework"
|
|
)
|
|
|
|
var (
|
|
timeout = 10 * time.Minute
|
|
waitTime = 2 * time.Second
|
|
)
|
|
|
|
// SIGDescribe annotates the test with the SIG label.
|
|
func SIGDescribe(text string, body func()) bool {
|
|
return ginkgo.Describe("[sig-scheduling] "+text, body)
|
|
}
|
|
|
|
// WaitForStableCluster waits until all existing pods are scheduled and returns their amount.
|
|
func WaitForStableCluster(c clientset.Interface, workerNodes sets.String) int {
|
|
startTime := time.Now()
|
|
// Wait for all pods to be scheduled.
|
|
allScheduledPods, allNotScheduledPods := getScheduledAndUnscheduledPods(c, workerNodes)
|
|
for len(allNotScheduledPods) != 0 {
|
|
time.Sleep(waitTime)
|
|
if startTime.Add(timeout).Before(time.Now()) {
|
|
framework.Logf("Timed out waiting for the following pods to schedule")
|
|
for _, p := range allNotScheduledPods {
|
|
framework.Logf("%v/%v", p.Namespace, p.Name)
|
|
}
|
|
framework.Failf("Timed out after %v waiting for stable cluster.", timeout)
|
|
break
|
|
}
|
|
allScheduledPods, allNotScheduledPods = getScheduledAndUnscheduledPods(c, workerNodes)
|
|
}
|
|
return len(allScheduledPods)
|
|
}
|
|
|
|
// getScheduledAndUnscheduledPods lists scheduled and not scheduled pods in all namespaces, with succeeded and failed pods filtered out.
|
|
func getScheduledAndUnscheduledPods(c clientset.Interface, workerNodes sets.String) (scheduledPods, notScheduledPods []v1.Pod) {
|
|
pods, err := c.CoreV1().Pods(metav1.NamespaceAll).List(context.TODO(), metav1.ListOptions{})
|
|
framework.ExpectNoError(err, fmt.Sprintf("listing all pods in namespace %q while waiting for stable cluster", metav1.NamespaceAll))
|
|
|
|
// API server returns also Pods that succeeded. We need to filter them out.
|
|
filteredPods := make([]v1.Pod, 0, len(pods.Items))
|
|
for _, p := range pods.Items {
|
|
if !podTerminated(p) {
|
|
filteredPods = append(filteredPods, p)
|
|
}
|
|
}
|
|
pods.Items = filteredPods
|
|
return GetPodsScheduled(workerNodes, pods)
|
|
}
|
|
|
|
func podTerminated(p v1.Pod) bool {
|
|
return p.Status.Phase == v1.PodSucceeded || p.Status.Phase == v1.PodFailed
|
|
}
|