From 54d8efbe3408d265fcabea612df3c87a5440c28f Mon Sep 17 00:00:00 2001
From: feistel <6742251-feistel@users.noreply.gitlab.com>
Date: Tue, 28 Jun 2022 14:29:08 +0200
Subject: Fix Flaky TestDomainResolverError race

---
 test/acceptance/serving_test.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'test')

diff --git a/test/acceptance/serving_test.go b/test/acceptance/serving_test.go
index c7aed337..84808f46 100644
--- a/test/acceptance/serving_test.go
+++ b/test/acceptance/serving_test.go
@@ -378,7 +378,7 @@ func TestDomainResolverError(t *testing.T) {
 
 	for name, test := range tests {
 		t.Run(name, func(t *testing.T) {
-			called := false
+			called := make(chan struct{})
 
 			// handler setup
 			pagesHandler := func(w http.ResponseWriter, r *http.Request) {
@@ -387,7 +387,7 @@ func TestDomainResolverError(t *testing.T) {
 					return
 				}
 
-				called = true
+				close(called)
 
 				if test.panic {
 					panic("server failed")
@@ -413,7 +413,11 @@ func TestDomainResolverError(t *testing.T) {
 			require.NoError(t, err)
 			testhelpers.Close(t, response.Body)
 
-			require.True(t, called, "api must have been called")
+			select {
+			case <-called:
+			case <-time.After(2 * time.Second):
+				t.Fatal("timed out waiting for the pages handler")
+			}
 
 			require.Equal(t, http.StatusBadGateway, response.StatusCode)
 
-- 
cgit v1.2.3