WIP.

github · Oct 21, 2024 · 113e674 · 113e674
1 parent 9e3bc1c
commit 113e674
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/go/logic/coordinator_test.go b/go/logic/coordinator_test.go
@@ -138,8 +138,9 @@ func TestCoordinator(t *testing.T) {
 	coord.applier = applier
 	coord.InitializeWorkers(8)
 
+	streamCtx, cancelStreaming := context.WithCancel(context.Background())
 	canStopStreaming := func() bool {
-		return false
+		return streamCtx.Err() != nil
 	}
 	go func() {
 		err = coord.StartStreaming(canStopStreaming)
@@ -153,6 +154,7 @@ func TestCoordinator(t *testing.T) {
 
 	for {
 		if ctx.Err() != nil {
+			cancelStreaming()
 			break
 		}