diff options
Diffstat (limited to 'workhorse/internal/dependencyproxy/dependencyproxy_test.go')
| -rw-r--r-- | workhorse/internal/dependencyproxy/dependencyproxy_test.go | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/workhorse/internal/dependencyproxy/dependencyproxy_test.go b/workhorse/internal/dependencyproxy/dependencyproxy_test.go index 37e54c0b756..d9169b2b4ce 100644 --- a/workhorse/internal/dependencyproxy/dependencyproxy_test.go +++ b/workhorse/internal/dependencyproxy/dependencyproxy_test.go @@ -33,7 +33,7 @@ func (f *fakeUploadHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { type errWriter struct{ writes int } -func (w *errWriter) Header() http.Header { return nil } +func (w *errWriter) Header() http.Header { return make(http.Header) } func (w *errWriter) WriteHeader(h int) {} // First call of Write function succeeds while all the subsequent ones fail @@ -112,8 +112,15 @@ func TestInject(t *testing.T) { func TestSuccessfullRequest(t *testing.T) { content := []byte("result") + contentLength := strconv.Itoa(len(content)) + contentType := "foo" + dockerContentDigest := "sha256:asdf1234" + overriddenHeader := "originResourceServer" originResourceServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - w.Header().Set("Content-Length", strconv.Itoa(len(content))) + w.Header().Set("Content-Length", contentLength) + w.Header().Set("Content-Type", contentType) + w.Header().Set("Docker-Content-Digest", dockerContentDigest) + w.Header().Set("Overridden-Header", overriddenHeader) w.Write(content) })) @@ -130,11 +137,16 @@ func TestSuccessfullRequest(t *testing.T) { require.Equal(t, "/target/upload", uploadHandler.request.URL.Path) require.Equal(t, int64(6), uploadHandler.request.ContentLength) + require.Equal(t, contentType, uploadHandler.request.Header.Get("Workhorse-Proxy-Content-Type")) + require.Equal(t, dockerContentDigest, uploadHandler.request.Header.Get("Docker-Content-Digest")) + require.Equal(t, overriddenHeader, uploadHandler.request.Header.Get("Overridden-Header")) require.Equal(t, content, uploadHandler.body) require.Equal(t, 200, response.Code) require.Equal(t, string(content), response.Body.String()) + require.Equal(t, contentLength, response.Header().Get("Content-Length")) + require.Equal(t, dockerContentDigest, response.Header().Get("Docker-Content-Digest")) } func TestIncorrectSendData(t *testing.T) { @@ -175,6 +187,7 @@ func TestFailedOriginServer(t *testing.T) { func makeRequest(injector *Injector, data string) *httptest.ResponseRecorder { w := httptest.NewRecorder() r := httptest.NewRequest("GET", "/target", nil) + r.Header.Set("Overridden-Header", "request") sendData := base64.StdEncoding.EncodeToString([]byte(data)) injector.Inject(w, r, sendData) |