reverting unwanted change

razorpay · Aug 31, 2022 · 3ebd7bc · 3ebd7bc
1 parent 2c22795
commit 3ebd7bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/stream/stream_test.go b/internal/stream/stream_test.go
@@ -204,8 +204,6 @@ func getMockSubscriber(ctx context.Context, ctrl *gomock.Controller) *mocks3.Moc
 		counter := 0
 		for {
 			select {
-			case <-cancelChan:
-				return
 			case <-reqCh:
 				if counter < 1 {
 					messages := getMockResponseMessages()
@@ -216,6 +214,8 @@ func getMockSubscriber(ctx context.Context, ctrl *gomock.Controller) *mocks3.Moc
 				} else {
 					resCh <- &metrov1.PullResponse{}
 				}
+			case <-cancelChan:
+				return
 			}
 		}
 	}()