fix: add timeout to the test (#618)

Signed-off-by: Vigith Maurice <vigith@gmail.com>
Signed-off-by: Yashash H L <yashashhl25@gmail.com>
Signed-off-by: jyu6 <juanlu_yu@intuit.com>
Co-authored-by: Yashash H L <yashashhl25@gmail.com>
Co-authored-by: jyu6 <juanlu_yu@intuit.com>

Makefile

@@ -81,7 +81,7 @@ dist/$(BINARY_NAME)-%:
 
 .PHONY: test
 test:
-	go test $(shell go list ./... | grep -v /vendor/ | grep -v /numaflow/test/) -race -short -v
+	go test $(shell go list ./... | grep -v /vendor/ | grep -v /numaflow/test/) -race -short -v -timeout 60s
 
 .PHONY: test-coverage
 test-coverage:
@@ -100,7 +100,7 @@ test-coverage-with-isb:
 
 .PHONY: test-code
 test-code:
-	go test -tags=isb_redis -race -v $(shell go list ./... | grep -v /vendor/ | grep -v /numaflow/test/)
+	go test -tags=isb_redis -race -v $(shell go list ./... | grep -v /vendor/ | grep -v /numaflow/test/) -timeout 120s
 
 test-e2e:
 test-kafka-e2e:

pkg/isb/stores/simplebuffer/buffer.go

@@ -180,6 +180,9 @@ func (b *InMemoryBuffer) Read(ctx context.Context, count int64) ([]*isb.ReadMess
 			if errors.Is(err, context.Canceled) {
 				return readMessages, nil
 			}
+			if errors.Is(err, context.DeadlineExceeded) {
+				return readMessages, nil
+			}
 			return readMessages, isb.BufferReadErr{Name: b.name, Empty: true, Message: err.Error()}
 		}
 		// access buffer via lock