chore: fix TestMeasureLatency/MeasureLatencyRecvTimeout flake

coadler · coadler · commit 9d4e1254a559 · 2024-05-16T17:54:05.000Z
diff --git a/coderd/database/pubsub/pubsub_linux_test.go b/coderd/database/pubsub/pubsub_linux_test.go
@@ -342,12 +342,13 @@ 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.Greater(t, send.Seconds(), 0.0)
+		require.ErrorContains(t, err, context.Canceled.Error())
+		require.Greater(t, send.Nanoseconds(), 0)
 		require.EqualValues(t, recv, time.Duration(-1))
 	})
 
