Serve 404 if ctx is canceled while resolving a domain

2 files changed, 7 insertions, 1 deletions

diff --git a/internal/routing/middleware.go b/internal/routing/middleware.go
index bee7c9df..8272635a 100644
--- a/internal/routing/middleware.go
+++ b/internal/routing/middleware.go
@@ -1,6 +1,7 @@
 package routing
 
 import (
+	"context"
 	"errors"
 	"net/http"
 
@@ -21,6 +22,11 @@ func NewMiddleware(handler http.Handler, s source.Source) http.Handler {
 		if err != nil && !errors.Is(err, domain.ErrDomainDoesNotExist) {
 			logging.LogRequest(r).WithError(err).Error("could not fetch domain information from a source")
 
+			if errors.Is(err, context.Canceled) {
+				httperrors.Serve404(w)
+				return
+			}
+
 			httperrors.Serve502(w)
 			return
 		}
diff --git a/test/acceptance/serving_test.go b/test/acceptance/serving_test.go
index 92e9c8c7..cf50069d 100644
--- a/test/acceptance/serving_test.go
+++ b/test/acceptance/serving_test.go
@@ -569,7 +569,7 @@ func TestSlowRequests(t *testing.T) {
 	defer cancel()
 
 	url := httpListener.URL("/index.html")
-	req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
 	require.NoError(t, err)
 
 	req.Host = "group.gitlab-example.com"