Close channel in tests

Khan · Sep 27, 2024 · fa3054c · fa3054c
1 parent 4466fc1
commit fa3054c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/integration/server/server.go b/internal/integration/server/server.go
@@ -161,6 +161,10 @@ func (s *subscriptionResolver) Count(ctx context.Context) (<-chan int, error) {
 	go func(respChan chan int) {
 		counter := 0
 		for {
+			if counter == 3 {
+				close(respChan)
+				return
+			}
 			respChan <- counter
 			counter++
 			time.Sleep(time.Second)