diff options
author | Max Wittig <max.wittig@siemens.com> | 2019-06-28 13:35:29 +0300 |
---|---|---|
committer | Max Wittig <max.wittig@siemens.com> | 2019-07-12 15:37:16 +0300 |
commit | 5199c4c8b646f3e66b0f03dd51fbaa704d9fd94f (patch) | |
tree | c60baf57319ea68b7b7213c4bba07662ced06d86 /acceptance_test.go | |
parent | 7d39822ce2221156c3479ceae0e4f8e24c7373b2 (diff) |
feat: add flag to define custom response headers
Diffstat (limited to 'acceptance_test.go')
-rw-r--r-- | acceptance_test.go | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/acceptance_test.go b/acceptance_test.go index eaa32318..28ebdeb8 100644 --- a/acceptance_test.go +++ b/acceptance_test.go @@ -241,6 +241,19 @@ func TestCORSForbidsPOST(t *testing.T) { } } +func TestCustomHeaders(t *testing.T) { + teardown := RunPagesProcess(t, *pagesBinary, listeners, "", "-header", "X-Test1:Testing1", "-header", "X-Test2:Testing2") + defer teardown() + + for _, spec := range listeners { + rsp, err := GetPageFromListener(t, spec, "group.gitlab-example.com:", "project/") + require.NoError(t, err) + assert.Equal(t, http.StatusOK, rsp.StatusCode) + assert.Equal(t, "Testing1", rsp.Header.Get("X-Test1")) + assert.Equal(t, "Testing2", rsp.Header.Get("X-Test2")) + } +} + func doCrossOriginRequest(t *testing.T, method, reqMethod, url string) *http.Response { req, err := http.NewRequest(method, url, nil) require.NoError(t, err) |