Tests for sending diffs

main_test.go

@@ -621,6 +621,45 @@ func TestGetGitBlob(t *testing.T) {
 	}
 }
 
+func TestGetGitDiff(t *testing.T) {
+	fromSha := "be93687618e4b132087f430a4d8fc3a609c9b77c"
+	toSha := "54fcc214b94e78d7a41a9a8fe6d87a5e59500e51"
+	headerKey := http.CanonicalHeaderKey("Gitlab-Workhorse-Send-Data")
+
+	ts := testhelper.TestServerWithHandler(regexp.MustCompile(`.`), func(w http.ResponseWriter, r *http.Request) {
+		responseJSON := fmt.Sprintf(`{"RepoPath":"%s","ShaFrom":"%s","ShaTo":"%s"}`, path.Join(testRepoRoot, testRepo), fromSha, toSha)
+		encodedJSON := base64.StdEncoding.EncodeToString([]byte(responseJSON))
+		w.Header().Set(headerKey, "git-diff:"+encodedJSON)
+		return
+	})
+
+	defer ts.Close()
+	ws := startWorkhorseServer(ts.URL)
+	defer ts.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)
+	}
+	if len(resp.Header[headerKey]) != 0 {
+		t.Fatalf("Unexpected response header: %s: %q", headerKey, resp.Header.Get(headerKey))
+	}
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if !strings.HasPrefix(string(body), "diff --git a/README b/README") {
+		t.Fatalf("diff --git a/README b/README, got %q", body)
+	}
+}
+
 func setupStaticFile(fpath, content string) error {
 	cwd, err := os.Getwd()
 	if err != nil {