From cdc694505645168c7992e908acc94c933416ebeb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dennis=20G=C3=BCnnewig?= <dev@fedux.org>
Date: Tue, 19 Jun 2018 22:59:47 +0200
Subject: Add tests for reverse proxy

---
 helpers_test.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'helpers_test.go')

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)
 
-- 
cgit v1.2.3