gopls/internal/progress: use t.Context in tests

This case was not recognized by modernize tool (probably because ctx is
not being initialized locally, but comes from setup()), so fix this
manually.

Change-Id: I135a3255a903689aee619b16a3fe16699459300a
Reviewed-on: https://go-review.googlesource.com/c/tools/+/703078
Reviewed-by: Alan Donovan <adonovan@google.com>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Auto-Submit: Kirill Kolyshkin <kolyshkin@gmail.com>
Reviewed-by: Mark Freeman <markfreeman@google.com>
Auto-Submit: Alan Donovan <adonovan@google.com>

Author: kolyshkin

gopls/internal/progress/progress_test.go

@@ -63,11 +63,11 @@ func (c *fakeClient) ShowMessage(context.Context, *protocol.ShowMessageParams) e
 	return nil
 }
 
-func setup() (context.Context, *Tracker, *fakeClient) {
+func setup() (*Tracker, *fakeClient) {
 	c := &fakeClient{}
 	tracker := NewTracker(c)
 	tracker.SetSupportsWorkDoneProgress(true)
-	return context.Background(), tracker, c
+	return tracker, c
 }
 
 func TestProgressTracker_Reporting(t *testing.T) {
@@ -108,9 +108,8 @@ func TestProgressTracker_Reporting(t *testing.T) {
 		},
 	} {
 		t.Run(test.name, func(t *testing.T) {
-			ctx, tracker, client := setup()
-			ctx, cancel := context.WithCancel(ctx)
-			defer cancel()
+			tracker, client := setup()
+			ctx := t.Context()
 			tracker.supportsWorkDoneProgress = test.supported
 			work := tracker.Start(ctx, "work", "message", test.token, nil)
 			client.mu.Lock()
@@ -146,10 +145,10 @@ func TestProgressTracker_Reporting(t *testing.T) {
 
 func TestProgressTracker_Cancellation(t *testing.T) {
 	for _, token := range []protocol.ProgressToken{nil, 1, "a"} {
-		ctx, tracker, _ := setup()
+		tracker, _ := setup()
 		var canceled bool
 		cancel := func() { canceled = true }
-		work := tracker.Start(ctx, "work", "message", token, cancel)
+		work := tracker.Start(t.Context(), "work", "message", token, cancel)
 		if err := tracker.Cancel(work.Token()); err != nil {
 			t.Fatal(err)
 		}