Fix incorrect response time at save response file

go-resty · Jul 23, 2020 · 5d3915c · 5d3915c
1 parent ca7661a
commit 5d3915c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -859,10 +859,11 @@ func (c *Client) execute(req *Request) (*Response, error) {
 			return response, err
 		}
 
-		response.setReceivedAt() // after we read the body
 		response.size = int64(len(response.body))
 	}
 
+	response.setReceivedAt() // after we read the body
+
 	// Apply Response middleware
 	for _, f := range c.afterResponse {
 		if err = f(c, response); err != nil {

diff --git a/request_test.go b/request_test.go
@@ -1312,6 +1312,7 @@ func TestOutputFileWithBaseDirAndRelativePath(t *testing.T) {
 
 	assertError(t, err)
 	assertEqual(t, true, resp.Size() != 0)
+	assertEqual(t, true, resp.Time() > 0)
 }
 
 func TestOutputFileWithBaseDirError(t *testing.T) {
@@ -1336,6 +1337,7 @@ func TestOutputPathDirNotExists(t *testing.T) {
 
 	assertError(t, err)
 	assertEqual(t, true, resp.Size() != 0)
+	assertEqual(t, true, resp.Time() > 0)
 }
 
 func TestOutputFileAbsPath(t *testing.T) {