Add more test cases for resolver errors

Fix imports and enable test

1 files changed, 2 insertions, 1 deletions

diff --git a/app.go b/app.go
index ec0c36a5..70ad6e1e 100644
--- a/app.go
+++ b/app.go
@@ -33,6 +33,7 @@ import (
 	"gitlab.com/gitlab-org/gitlab-pages/internal/request"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/serving/disk/zip"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/source"
+	"gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/client"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/tlsconfig"
 	"gitlab.com/gitlab-org/gitlab-pages/metrics"
 )
@@ -152,7 +153,7 @@ func (a *theApp) routingMiddleware(handler http.Handler) http.Handler {
 		// if we could not retrieve a domain from domains source we break the
 		// middleware chain and simply respond with 502 after logging this
 		host, d, err := a.getHostAndDomain(r)
-		if err != nil && !errors.Is(err, domain.ErrDomainDoesNotExist) {
+		if err != nil && !errors.Is(err, client.ErrDomainDoesNotExist) {
 			metrics.DomainsSourceFailures.Inc()
 			log.WithError(err).Error("could not fetch domain information from a source")