Browse Source

Merge pull request #3216 from yichengq/cancel-err

client: return context canceled error correctly
Xiang Li 10 năm trước cách đây
mục cha
commit
b9c6b64d61
2 tập tin đã thay đổi với 4 bổ sung6 xóa
  1. 2 4
      client/client.go
  2. 2 2
      client/client_test.go

+ 2 - 4
client/client.go

@@ -392,11 +392,9 @@ func (c *simpleHTTPClient) Do(ctx context.Context, act httpAction) (*http.Respon
 
 	select {
 	case <-ctx.Done():
-		err = resp.Body.Close()
+		resp.Body.Close()
 		<-done
-		if err == nil {
-			err = ctx.Err()
-		}
+		return nil, nil, ctx.Err()
 	case <-done:
 	}
 

+ 2 - 2
client/client_test.go

@@ -257,8 +257,8 @@ func TestSimpleHTTPClientDoCancelContextResponseBodyClosedWithBlockingBody(t *te
 	}()
 
 	_, _, err := c.Do(ctx, &fakeAction{})
-	if err == nil {
-		t.Fatalf("expected non-nil error, got nil")
+	if err != context.Canceled {
+		t.Fatalf("expected %+v, got %+v", context.Canceled, err)
 	}
 
 	if !body.closed {