fix(internal/gensupport): Make SendRequestWithRetry check for canceled contexts twice

internal/gensupport/send.go

@@ -99,6 +99,19 @@ func sendAndRetry(ctx context.Context, client *http.Client, req *http.Request, r
 		case <-time.After(pause):
 		}
 
+		select {
+		case <-ctx.Done():
+			// Check for context cancellation once more. If more than one case in a
+			// select is satisfied at the same time, Go will choose one arbitrarily.
+			// That can cause an operation to go through even if the context was
+			// canceled before.
+			if err == nil {
+				err = ctx.Err()
+			}
+			return resp, err
+		default:
+		}
+
 		resp, err = client.Do(req.WithContext(ctx))
 
 		var status int

internal/gensupport/send_test.go

@@ -6,6 +6,7 @@ package gensupport
 
 import (
 	"context"
+	"errors"
 	"net/http"
 	"testing"
 )
@@ -29,3 +30,24 @@ func TestSendRequestWithRetry(t *testing.T) {
 		t.Error("got nil, want error")
 	}
 }
+
+type brokenRoundTripper struct{}
+
+func (t *brokenRoundTripper) RoundTrip(r *http.Request) (*http.Response, error) {
+	return nil, errors.New("this should not happen")
+}
+
+func TestCanceledContextDoesNotPerformRequest(t *testing.T) {
+	client := http.Client{
+		Transport: &brokenRoundTripper{},
+	}
+	for i := 0; i < 1000; i++ {
+		req, _ := http.NewRequest("GET", "url", nil)
+		ctx, cancel := context.WithCancel(context.Background())
+		cancel()
+		_, err := SendRequestWithRetry(ctx, &client, req)
+		if !errors.Is(err, context.Canceled) {
+			t.Fatalf("got %v, want %v", err, context.Canceled)
+		}
+	}
+}