Add tests for reverse proxy

2 files changed, 33 insertions, 0 deletions

diff --git a/acceptance_test.go b/acceptance_test.go
index e971588a..361cba68 100644
--- a/acceptance_test.go
+++ b/acceptance_test.go
@@ -553,3 +553,23 @@ func TestMultiFlagEnvironmentVariables(t *testing.T) {
 		assert.Equal(t, http.StatusOK, rsp.StatusCode)
 	}
 }
+
+func TestKnownHostInReverseProxySetupReturns200(t *testing.T) {
+	skipUnlessEnabled(t)
+
+	var listeners = []ListenSpec{
+		{"proxy", "127.0.0.1", "37002"},
+		{"proxy", "::1", "37002"},
+	}
+
+	teardown := RunPagesProcess(t, *pagesBinary, listeners, "")
+	defer teardown()
+
+	for _, spec := range listeners {
+		rsp, err := GetProxiedPageFromListener(t, spec, "localhost", "group.gitlab-example.com", "project/")
+
+		require.NoError(t, err)
+		rsp.Body.Close()
+		assert.Equal(t, http.StatusOK, rsp.StatusCode)
+	}
+}
diff --git a/helpers_test.go b/helpers_test.go
index e853e1c4..ccbbb6e3 100644
--- a/helpers_test.go
+++ b/helpers_test.go
@@ -259,6 +259,19 @@ func GetPageFromListener(t *testing.T, spec ListenSpec, host, urlsuffix string)
 	return DoPagesRequest(t, req)
 }
 
+func GetProxiedPageFromListener(t *testing.T, spec ListenSpec, host, xForwardedHost, urlsuffix string) (*http.Response, error) {
+	url := spec.URL(urlsuffix)
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Host = host
+	req.Header.Set("X-Forwarded-Host", xForwardedHost)
+
+	return DoPagesRequest(t, req)
+}
+
 func DoPagesRequest(t *testing.T, req *http.Request) (*http.Response, error) {
 	t.Logf("curl -X %s -H'Host: %s' %s", req.Method, req.Host, req.URL)