test: Use NewTestAlways for a stream waker if the count is 0.

google · Jul 10, 2024 · aba5e89 · aba5e89
1 parent b53a574
commit aba5e89
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/internal/mtail/testing.go b/internal/mtail/testing.go
@@ -63,7 +63,12 @@ func TestMakeServer(tb testing.TB, patternWakers int, streamWakers int, options
 		tb:     tb,
 		cancel: cancel,
 	}
-	ts.streamWaker, ts.AwakenLogStreams = waker.NewTest(ctx, streamWakers, "streams")
+	if streamWakers == 0 {
+		ts.streamWaker = waker.NewTestAlways()
+		ts.AwakenLogStreams = func(int, int) {}
+	} else {
+		ts.streamWaker, ts.AwakenLogStreams = waker.NewTest(ctx, streamWakers, "streams")
+	}
 	ts.patternWaker, ts.AwakenPatternPollers = waker.NewTest(ctx, patternWakers, "patterns")
 	options = append(options,
 		LogstreamPollWaker(ts.streamWaker),