chore: fix TestMeasureLatency/MeasureLatencyRecvTimeout flake

coadler · coadler · commit 6df19593740b · 2024-05-16T17:51:27.000Z
diff --git a/coderd/database/pubsub/pubsub_linux_test.go b/coderd/database/pubsub/pubsub_linux_test.go
@@ -342,11 +342,12 @@ func TestMeasureLatency(t *testing.T) {
 		ps, done := newPubsub()
 		defer done()
 
-		ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(-time.Hour))
-		defer cancel()
+		ctx, cancel := context.WithCancel(context.Background())
+		cancel()
 
+		<-ctx.Done()
 		send, recv, err := pubsub.NewLatencyMeasurer(logger).Measure(ctx, ps)
-		require.ErrorContains(t, err, context.DeadlineExceeded.Error())
+		require.ErrorContains(t, err, context.Canceled.Error())
 		require.Greater(t, send.Seconds(), 0.0)
 		require.EqualValues(t, recv, time.Duration(-1))
 	})
