Fix Flaky TestDomainResolverError race

1 files changed, 7 insertions, 3 deletions

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)