github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-05 18:24:07 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #4294 from zmerlynn/util_funcs

Browse Source 
Utility functions for Gingko By and Fail
This commit is contained in:
Filipe Brandenburger 2015-02-11 16:53:38 -08:00
commit 854a3d1fad
2 changed files with 37 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
test/e2e/rc.go
View File

@ -99,12 +99,12 @@ func ServeImageOrFail(c *client.Client, test string, image string) {
// Resize the replication controller to zero to get rid of pods. // Resize the replication controller to zero to get rid of pods.
controller.Spec.Replicas = 0 controller.Spec.Replicas = 0
if _, err = c.ReplicationControllers(ns).Update(controller); err != nil { if _, err = c.ReplicationControllers(ns).Update(controller); err != nil {
By(fmt.Sprintf("Failed to resize replication controller %s to zero: %v", name, err)) Logf("Failed to resize replication controller %s to zero: %v", name, err)
} }
// Delete the replication controller. // Delete the replication controller.
if err = c.ReplicationControllers(ns).Delete(name); err != nil { if err = c.ReplicationControllers(ns).Delete(name); err != nil {
By(fmt.Sprintf("Failed to delete replication controller %s: %v", name, err)) Logf("Failed to delete replication controller %s: %v", name, err)
} }
}() }()
@ -115,20 +115,21 @@ func ServeImageOrFail(c *client.Client, test string, image string) {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
t := time.Now() t := time.Now()
for { for {
By(fmt.Sprintf("Controller %s: Found %d pods out of %d", name, len(pods.Items), replicas)) Logf("Controller %s: Found %d pods out of %d", name, len(pods.Items), replicas)
if len(pods.Items) == replicas { if len(pods.Items) == replicas {
break break
} }
if time.Since(t) > listTimeout { if time.Since(t) > listTimeout {
Fail(fmt.Sprintf( Failf("Controller %s: Gave up waiting for %d pods to come up after seeing only %d pods after %v seconds",
"Controller %s: Gave up waiting for %d pods to come up after seeing only %d pods after %v seconds", name, replicas, len(pods.Items), time.Since(t).Seconds())
name, replicas, len(pods.Items), time.Since(t).Seconds()))
} }
time.Sleep(5 * time.Second) time.Sleep(5 * time.Second)
pods, err = c.Pods(ns).List(label) pods, err = c.Pods(ns).List(label)
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
} }
By("Ensuring each pod is running and has a hostIP")
// Wait for the pods to enter the running state. Waiting loops until the pods // Wait for the pods to enter the running state. Waiting loops until the pods
// are running so non-running pods cause a timeout for this test. // are running so non-running pods cause a timeout for this test.
for _, pod := range pods.Items { for _, pod := range pods.Items {
@ -144,14 +145,14 @@ func ServeImageOrFail(c *client.Client, test string, image string) {
p, err := c.Pods(ns).Get(pod.Name) p, err := c.Pods(ns).Get(pod.Name)
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
if p.Status.HostIP != "" { if p.Status.HostIP != "" {
By(fmt.Sprintf("Controller %s: Replica %d has hostIP: %s", name, i+1, p.Status.HostIP)) Logf("Controller %s: Replica %d has hostIP: %s", name, i+1, p.Status.HostIP)
break break
} }
if time.Since(t) >= hostIPTimeout { if time.Since(t) >= hostIPTimeout {
Fail(fmt.Sprintf("Controller %s: Gave up waiting for hostIP of replica %d after %v seconds", Failf("Controller %s: Gave up waiting for hostIP of replica %d after %v seconds",
name, i, time.Since(t).Seconds())) name, i, time.Since(t).Seconds())
} }
By(fmt.Sprintf("Controller %s: Retrying to get the hostIP of replica %d", name, i+1)) Logf("Controller %s: Retrying to get the hostIP of replica %d", name, i+1)
time.Sleep(5 * time.Second) time.Sleep(5 * time.Second)
} }
} }
@ -161,24 +162,26 @@ func ServeImageOrFail(c *client.Client, test string, image string) {
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
// Verify that something is listening. // Verify that something is listening.
By("Trying to dial each unique pod")
for i, pod := range pods.Items { for i, pod := range pods.Items {
resp, err := http.Get(fmt.Sprintf("http://%s:8080", pod.Status.HostIP)) resp, err := http.Get(fmt.Sprintf("http://%s:8080", pod.Status.HostIP))
if err != nil { if err != nil {
Fail(fmt.Sprintf("Controller %s: Failed to GET from replica %d: %v", name, i+1, err)) Failf("Controller %s: Failed to GET from replica %d: %v", name, i+1, err)
} }
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
Fail(fmt.Sprintf("Controller %s: Expected OK status code for replica %d but got %d", name, i+1, resp.StatusCode)) Failf("Controller %s: Expected OK status code for replica %d but got %d", name, i+1, resp.StatusCode)
} }
body, err := ioutil.ReadAll(resp.Body) body, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil {
Fail(fmt.Sprintf("Controller %s: Failed to read the body of the GET response from replica %d: %v", Failf("Controller %s: Failed to read the body of the GET response from replica %d: %v",
name, i+1, err)) name, i+1, err)
} }
// The body should be the pod name. // The body should be the pod name.
if string(body) != pod.Name { if string(body) != pod.Name {
Fail(fmt.Sprintf("Controller %s: Replica %d expected response %s but got %s", name, i+1, pod.Name, string(body))) Failf("Controller %s: Replica %d expected response %s but got %s", name, i+1, pod.Name, string(body))
} }
By(fmt.Sprintf("Controller %s: Got expected result from replica %d: %s", name, i+1, string(body))) Logf("Controller %s: Got expected result from replica %d: %s", name, i+1, string(body))
} }
} }

28
test/e2e/util.go
View File

@ -40,6 +40,14 @@ type testContextType struct {
var testContext testContextType var testContext testContextType
func Logf(format string, a ...interface{}) {
fmt.Fprintf(GinkgoWriter, "INFO: "+format+"\n", a...)
}
func Failf(format string, a ...interface{}) {
Fail(fmt.Sprintf(format, a...))
}
func waitForPodRunning(c *client.Client, id string, tryFor time.Duration) error { func waitForPodRunning(c *client.Client, id string, tryFor time.Duration) error {
trySecs := int(tryFor.Seconds()) trySecs := int(tryFor.Seconds())
for i := 0; i <= trySecs; i += 5 { for i := 0; i <= trySecs; i += 5 {
@ -51,7 +59,7 @@ func waitForPodRunning(c *client.Client, id string, tryFor time.Duration) error
if pod.Status.Phase == api.PodRunning { if pod.Status.Phase == api.PodRunning {
return nil return nil
} }
By(fmt.Sprintf("Waiting for pod %s status to be %q (found %q) (%d secs)", id, api.PodRunning, pod.Status.Phase, i)) Logf("Waiting for pod %s status to be %q (found %q) (%d secs)", id, api.PodRunning, pod.Status.Phase, i)
} }
return fmt.Errorf("Gave up waiting for pod %s to be running after %d seconds", id, trySecs) return fmt.Errorf("Gave up waiting for pod %s to be running after %d seconds", id, trySecs)
} }
@ -65,14 +73,14 @@ func waitForPodNotPending(c *client.Client, ns, podName string, tryFor time.Dura
} }
pod, err := c.Pods(ns).Get(podName) pod, err := c.Pods(ns).Get(podName)
if err != nil { if err != nil {
By(fmt.Sprintf("Get pod %s in namespace %s failed, ignoring for 5s: %v", podName, ns, err)) Logf("Get pod %s in namespace %s failed, ignoring for 5s: %v", podName, ns, err)
continue continue
} }
if pod.Status.Phase != api.PodPending { if pod.Status.Phase != api.PodPending {
By(fmt.Sprintf("Saw pod %s in namespace %s out of pending state (found %q)", podName, ns, pod.Status.Phase)) Logf("Saw pod %s in namespace %s out of pending state (found %q)", podName, ns, pod.Status.Phase)
return nil return nil
} }
By(fmt.Sprintf("Waiting for status of pod %s in namespace %s to be !%q (found %q) (%v secs)", podName, ns, api.PodPending, pod.Status.Phase, i)) Logf("Waiting for status of pod %s in namespace %s to be !%q (found %q) (%v secs)", podName, ns, api.PodPending, pod.Status.Phase, i)
} }
return fmt.Errorf("Gave up waiting for status of pod %s in namespace %s to go out of pending after %d seconds", podName, ns, trySecs) return fmt.Errorf("Gave up waiting for status of pod %s in namespace %s to go out of pending after %d seconds", podName, ns, trySecs)
} }
@ -86,24 +94,24 @@ func waitForPodSuccess(c *client.Client, podName string, contName string, tryFor
} }
pod, err := c.Pods(api.NamespaceDefault).Get(podName) pod, err := c.Pods(api.NamespaceDefault).Get(podName)
if err != nil { if err != nil {
By(fmt.Sprintf("Get pod failed, ignoring for 5s: %v", err)) Logf("Get pod failed, ignoring for 5s: %v", err)
continue continue
} }
// Cannot use pod.Status.Phase == api.PodSucceeded/api.PodFailed due to #2632 // Cannot use pod.Status.Phase == api.PodSucceeded/api.PodFailed due to #2632
ci, ok := pod.Status.Info[contName] ci, ok := pod.Status.Info[contName]
if !ok { if !ok {
By(fmt.Sprintf("No Status.Info for container %s in pod %s yet", contName, podName)) Logf("No Status.Info for container %s in pod %s yet", contName, podName)
} else { } else {
if ci.State.Termination != nil { if ci.State.Termination != nil {
if ci.State.Termination.ExitCode == 0 { if ci.State.Termination.ExitCode == 0 {
By("Saw pod success") By("Saw pod success")
return nil return nil
} else { } else {
By(fmt.Sprintf("Saw pod failure: %+v", ci.State.Termination)) Logf("Saw pod failure: %+v", ci.State.Termination)
} }
By(fmt.Sprintf("Waiting for pod %q status to be success or failure", podName)) Logf("Waiting for pod %q status to be success or failure", podName)
} else { } else {
By(fmt.Sprintf("Nil State.Termination for container %s in pod %s so far", contName, podName)) Logf("Nil State.Termination for container %s in pod %s so far", contName, podName)
} }
} }
} }
@ -120,7 +128,7 @@ func loadClient() (*client.Client, error) {
} }
// If the certificate directory is provided, set the cert paths to be there. // If the certificate directory is provided, set the cert paths to be there.
if testContext.certDir != "" { if testContext.certDir != "" {
By(fmt.Sprintf("Expecting certs in %v.", testContext.certDir)) Logf("Expecting certs in %v.", testContext.certDir)
info.CAFile = filepath.Join(testContext.certDir, "ca.crt") info.CAFile = filepath.Join(testContext.certDir, "ca.crt")
info.CertFile = filepath.Join(testContext.certDir, "kubecfg.crt") info.CertFile = filepath.Join(testContext.certDir, "kubecfg.crt")
info.KeyFile = filepath.Join(testContext.certDir, "kubecfg.key") info.KeyFile = filepath.Join(testContext.certDir, "kubecfg.key")