From a15dcd718bf2337a266d449dc7149ba17d668b95 Mon Sep 17 00:00:00 2001 From: Manuel de Brito Fontes Date: Sat, 26 May 2018 17:11:15 -0400 Subject: [PATCH] Fix influxdb test flake --- test/e2e/annotations/influxdb.go | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/test/e2e/annotations/influxdb.go b/test/e2e/annotations/influxdb.go index f72825ba5..0adfd4657 100644 --- a/test/e2e/annotations/influxdb.go +++ b/test/e2e/annotations/influxdb.go @@ -31,6 +31,7 @@ import ( corev1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/util/intstr" + "k8s.io/apimachinery/pkg/util/wait" "k8s.io/ingress-nginx/test/e2e/framework" ) @@ -77,12 +78,20 @@ var _ = framework.IngressNginxDescribe("Annotations - influxdb", func() { Expect(res.StatusCode).Should(Equal(http.StatusOK)) time.Sleep(5 * time.Second) - measurements, err := extractInfluxDBMeasurements(f) + + var measurements string + err = wait.PollImmediate(time.Second, time.Minute, func() (bool, error) { + measurements, err = extractInfluxDBMeasurements(f) + if err != nil { + return false, nil + } + return true, nil + }) + Expect(err).NotTo(HaveOccurred()) var results map[string][]map[string]interface{} json.Unmarshal([]byte(measurements), &results) - Expect(err).NotTo(HaveOccurred()) Expect(len(measurements)).ShouldNot(Equal(0)) for _, elem := range results["results"] { Expect(len(elem)).ShouldNot(Equal(0))