Add Request.SetBodyRaw

Fixes https://github.com/valyala/fasthttp/issues/891

1 files changed, 40 insertions, 0 deletions

diff --git a/http_test.go b/http_test.go
index bcac9dc..c67f520 100644
--- a/http_test.go
+++ b/http_test.go
@@ -2081,6 +2081,18 @@ func TestResponseRawBodySet(t *testing.T) {
 	testBodyWriteTo(t, &resp, expectedS, true)
 }
 
+func TestRequestRawBodySet(t *testing.T) {
+	t.Parallel()
+
+	var r Request
+
+	expectedS := "test"
+	body := []byte(expectedS)
+	r.SetBodyRaw(body)
+
+	testBodyWriteTo(t, &r, expectedS, true)
+}
+
 func TestResponseRawBodyReset(t *testing.T) {
 	t.Parallel()
 
@@ -2093,6 +2105,18 @@ func TestResponseRawBodyReset(t *testing.T) {
 	testBodyWriteTo(t, &resp, "", true)
 }
 
+func TestRequestRawBodyReset(t *testing.T) {
+	t.Parallel()
+
+	var r Request
+
+	body := []byte("test")
+	r.SetBodyRaw(body)
+	r.ResetBody()
+
+	testBodyWriteTo(t, &r, "", true)
+}
+
 func TestResponseRawBodyCopyTo(t *testing.T) {
 	t.Parallel()
 
@@ -2105,6 +2129,22 @@ func TestResponseRawBodyCopyTo(t *testing.T) {
 	testResponseCopyTo(t, &resp)
 }
 
+func TestRequestRawBodyCopyTo(t *testing.T) {
+	t.Parallel()
+
+	var a Request
+
+	body := []byte("test")
+	a.SetBodyRaw(body)
+
+	var b Request
+
+	a.CopyTo(&b)
+
+	testBodyWriteTo(t, &a, "test", true)
+	testBodyWriteTo(t, &b, "test", true)
+}
+
 type testReader struct {
 	read chan (int)
 	cb   chan (struct{})