pods.go

package helper

import (
	"context"
	"log"
	"time"

	. "github.com/onsi/gomega"

	kubev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

// WaitForPodPhase until in target, checking n times and sleeping dur between them. Last known phase is returned.
func (h *H) WaitForPodPhase(pod *kubev1.Pod, target kubev1.PodPhase, n int, dur time.Duration) (phase kubev1.PodPhase) {
	var err error
	for i := 0; i < n; i++ {
		if pod, err = h.Kube().CoreV1().Pods(pod.Namespace).Get(context.TODO(), pod.Name, metav1.GetOptions{}); err != nil {
			log.Println(err)
		} else if pod != nil {
			phase = pod.Status.Phase

			// stop checking if Pod has reached state or failed
			if phase == target || phase == kubev1.PodFailed {
				return
			}
		}

		log.Printf("Waiting for Pod '%s/%s' to be %s, currently %s...", pod.Namespace, pod.Name, target, phase)
		time.Sleep(dur)
	}

	Expect(phase).NotTo(BeEmpty())
	return
}