Restore Response.Body when AutoReadResponse is enabled(#152)

imroc · Aug 14, 2022 · 6e38959 · 6e38959
1 parent 6984f39
commit 6e38959
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -1,6 +1,7 @@
 package req
 
 import (
+	"bytes"
 	"context"
 	"crypto/tls"
 	"crypto/x509"
@@ -1191,6 +1192,8 @@ func (c *Client) roundTrip(r *Request) (resp *Response, err error) {
 		if err != nil {
 			return
 		}
+		// restore body for re-reads
+		resp.Body = ioutil.NopCloser(bytes.NewReader(resp.body))
 	}
 
 	for _, f := range r.client.afterResponse {

diff --git a/request_test.go b/request_test.go
@@ -1010,3 +1010,14 @@ func TestDownloadCallback(t *testing.T) {
 	assertSuccess(t, resp, err)
 	tests.AssertEqual(t, true, n > 0)
 }
+
+func TestRestoreResponseBody(t *testing.T) {
+	c := tc()
+	resp, err := c.R().Get("/")
+	assertSuccess(t, resp, err)
+	tests.AssertNoError(t, err)
+	tests.AssertEqual(t, true, len(resp.Bytes()) > 0)
+	body, err := ioutil.ReadAll(resp.Body)
+	tests.AssertNoError(t, err)
+	tests.AssertEqual(t, true, len(body) > 0)
+}