add(Tests): check is got sending header values to server tests.

Author: melbahja

download_test.go

@@ -31,6 +31,7 @@ func TestGetInfoAndInit(t *testing.T) {
 	t.Run("getInfoTest", getInfoTest)
 	t.Run("okInitTest", okInitTest)
 	t.Run("errInitTest", errInitTest)
+	t.Run("sendHeadersTest", sendHeadersTest)
 }
 
 func TestDownloading(t *testing.T) {
@@ -68,6 +69,35 @@ func getInfoTest(t *testing.T) {
 	}
 }
 
+func sendHeadersTest(t *testing.T) {
+
+	tmpFile := createTemp()
+	defer clean(tmpFile)
+
+	dl := got.NewDownload(context.Background(), httpt.URL+"/header_values", tmpFile)
+	dl.Header = []got.GotHeader{
+		{
+			Key:   "x-test-header",
+			Value: "foobar",
+		},
+	}
+
+	info, err := dl.GetInfo()
+
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	if info.Rangeable == false {
+		t.Error("rangeable should be true")
+	}
+
+	if info.Size != uint64(okFileStat.Size()) {
+		t.Errorf("Invalid file size, wants %d but got %d", okFileStat.Size(), info.Size)
+	}
+}
+
 func getFilenameTest(t *testing.T) {
 
 	tmpFile := createTemp()

got_test.go

@@ -53,6 +53,15 @@ func NewHttptestServer() *httptest.Server {
 			http.ServeFile(w, r, "go.mod")
 			return
 
+		case "/header_values":
+			if r.Header.Get("x-test-header") == "foobar" {
+				http.ServeFile(w, r, "go.mod")
+				return
+			}
+
+			w.WriteHeader(403)
+			return
+
 		case "/not_found":
 		}