(t *testing.T)
| 105 | } |
| 106 | |
| 107 | func TestHTTPServiceHostHeaderOverride(t *testing.T) { |
| 108 | cfg := OriginRequestConfig{ |
| 109 | HTTPHostHeader: t.Name(), |
| 110 | } |
| 111 | handler := func(w http.ResponseWriter, r *http.Request) { |
| 112 | require.Equal(t, r.Host, t.Name()) |
| 113 | if websocket.IsWebSocketUpgrade(r) { |
| 114 | respHeaders := websocket.NewResponseHeader(r) |
| 115 | for k, v := range respHeaders { |
| 116 | w.Header().Set(k, v[0]) |
| 117 | } |
| 118 | w.WriteHeader(http.StatusSwitchingProtocols) |
| 119 | return |
| 120 | } |
| 121 | // return the X-Forwarded-Host header for assertions |
| 122 | // as the httptest Server URL isn't available here yet |
| 123 | w.Write([]byte(r.Header.Get("X-Forwarded-Host"))) |
| 124 | } |
| 125 | origin := httptest.NewServer(http.HandlerFunc(handler)) |
| 126 | defer origin.Close() |
| 127 | |
| 128 | originURL, err := url.Parse(origin.URL) |
| 129 | require.NoError(t, err) |
| 130 | |
| 131 | httpService := &httpService{ |
| 132 | url: originURL, |
| 133 | } |
| 134 | shutdownC := make(chan struct{}) |
| 135 | require.NoError(t, httpService.start(TestLogger, shutdownC, cfg)) |
| 136 | |
| 137 | req, err := http.NewRequest(http.MethodGet, originURL.String(), nil) |
| 138 | require.NoError(t, err) |
| 139 | |
| 140 | resp, err := httpService.RoundTrip(req) |
| 141 | require.NoError(t, err) |
| 142 | require.Equal(t, http.StatusOK, resp.StatusCode) |
| 143 | |
| 144 | respBody, err := io.ReadAll(resp.Body) |
| 145 | require.NoError(t, err) |
| 146 | require.Equal(t, respBody, []byte(originURL.Host)) |
| 147 | } |
| 148 | |
| 149 | // TestHTTPServiceUsesIngressRuleScheme makes sure httpService uses scheme defined in ingress rule and not by eyeball request |
| 150 | func TestHTTPServiceUsesIngressRuleScheme(t *testing.T) { |
nothing calls this directly
no test coverage detected