Close channel in tests

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