fix: workspaceapps: overloaded test server responds with 502s (#5255)

coderd/client_test.go

@@ -0,0 +1,49 @@
+package coderd_test
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/codersdk"
+	"github.com/coder/coder/testutil"
+)
+
+// Issue: https://github.com/coder/coder/issues/5249
+// While running tests in parallel, the web server seems to be overloaded and responds with HTTP 502.
+// require.Eventually expects correct HTTP responses.
+
+func doWithRetries(t require.TestingT, client *codersdk.Client, req *http.Request) (*http.Response, error) {
+	var resp *http.Response
+	var err error
+	require.Eventually(t, func() bool {
+		// nolint // only requests which are not passed upstream have a body closed
+		resp, err = client.HTTPClient.Do(req)
+		if resp != nil && resp.StatusCode == http.StatusBadGateway {
+			if resp.Body != nil {
+				resp.Body.Close()
+			}
+			return false
+		}
+		return true
+	}, testutil.WaitLong, testutil.IntervalFast)
+	return resp, err
+}
+
+func requestWithRetries(ctx context.Context, t require.TestingT, client *codersdk.Client, method, path string, body interface{}, opts ...codersdk.RequestOption) (*http.Response, error) {
+	var resp *http.Response
+	var err error
+	require.Eventually(t, func() bool {
+		// nolint // only requests which are not passed upstream have a body closed
+		resp, err = client.Request(ctx, method, path, body, opts...)
+		if resp != nil && resp.StatusCode == http.StatusBadGateway {
+			if resp.Body != nil {
+				resp.Body.Close()
+			}
+			return false
+		}
+		return true
+	}, testutil.WaitLong, testutil.IntervalFast)
+	return resp, err
+}