test: update healthcheck test for disk source removal

1 files changed, 67 insertions, 0 deletions

diff --git a/app_test.go b/app_test.go
index d78d6737..483ffb39 100644
--- a/app_test.go
+++ b/app_test.go
@@ -3,12 +3,17 @@ package main
 import (
 	"crypto/tls"
 	"fmt"
+	"io"
 	"net/http"
+	"net/http/httptest"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/require"
 
+	"gitlab.com/gitlab-org/gitlab-pages/internal/config"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/request"
+	"gitlab.com/gitlab-org/gitlab-pages/internal/source"
 )
 
 func Test_setRequestScheme(t *testing.T) {
@@ -58,3 +63,65 @@ func newGetRequestWithScheme(t *testing.T, scheme string, withTLS bool) *http.Re
 
 	return req
 }
+
+func TestHealthCheckMiddleware(t *testing.T) {
+	tests := []struct {
+		name   string
+		path   string
+		status int
+		body   string
+	}{
+		{
+			name:   "Not a healthcheck request",
+			path:   "/foo/bar",
+			status: http.StatusOK,
+			body:   "Hello from inner handler",
+		},
+		{
+			name:   "Healthcheck request",
+			path:   "/-/healthcheck",
+			status: http.StatusOK,
+			body:   "success\n",
+		},
+	}
+
+	validCfg := config.GitLab{
+		InternalServer:     "server",
+		APISecretKey:       []byte("secret"),
+		ClientHTTPTimeout:  time.Second,
+		JWTTokenExpiration: time.Second,
+	}
+
+	domains, err := source.NewDomains(&validCfg)
+	require.NoError(t, err)
+
+	cfg := config.Config{
+		General: config.General{
+			StatusPath: "/-/healthcheck",
+		},
+	}
+
+	app := theApp{
+		config:  &cfg,
+		domains: domains,
+	}
+
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		io.WriteString(w, "Hello from inner handler")
+	})
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			r := httptest.NewRequest("GET", tc.path, nil)
+			rr := httptest.NewRecorder()
+
+			middleware, err := app.healthCheckMiddleware(handler)
+			require.NoError(t, err)
+			middleware.ServeHTTP(rr, r)
+
+			require.Equal(t, tc.status, rr.Code)
+			require.Equal(t, tc.body, rr.Body.String())
+		})
+	}
+}