Add test for sending Git blobs

main_test.go

@@ -555,6 +555,44 @@ func TestArtifactsGetSingleFile(t *testing.T) {
 	}
 }
 
+func TestGetGitBlob(t *testing.T) {
+	blobId := "50b27c6518be44c42c4d87966ae2481ce895624c" // the LICENSE file in the test repository
+	blobLength := 1075
+	ts := testhelper.TestServerWithHandler(regexp.MustCompile(`.`), func(w http.ResponseWriter, r *http.Request) {
+		responseJSON := fmt.Sprintf(`{"RepoPath":"%s","BlobId":"%s"}`, path.Join(testRepoRoot, testRepo), blobId)
+		encodedJSON := base64.StdEncoding.EncodeToString([]byte(responseJSON))
+		w.Header().Set("Gitlab-Workhorse-Send-Data", "git-blob:"+encodedJSON)
+		// Prevent the Go HTTP server from setting the Content-Length to 0.
+		w.Header().Set("Transfer-Encoding", "chunked")
+		if _, err := fmt.Fprintf(w, "GNU General Public License"); err != nil {
+			t.Fatal(err)
+		}
+		return
+	})
+	defer ts.Close()
+	ws := startWorkhorseServer(ts.URL)
+	defer ws.Close()
+	resourcePath := "/something"
+	resp, err := http.Get(ws.URL + resourcePath)
+	if err != nil {
+		t.Error(err)
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != 200 {
+		t.Errorf("GET %q: expected 200, got %d", resourcePath, resp.StatusCode)
+	}
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(body) != blobLength {
+		t.Fatalf("Expected body of %d bytes, got %d", blobLength, len(body))
+	}
+	if !strings.HasPrefix(string(body), "The MIT License (MIT)") {
+		t.Fatalf("Expected MIT license, got %q", body)
+	}
+}
+
 func setupStaticFile(fpath, content string) error {
 	cwd, err := os.Getwd()
 	if err != nil {