comments

Signed-off-by: Sidhant Kohli <[email protected]>
numaproj · Jul 2, 2024 · 2e02e6c · 2e02e6c
1 parent f3cc8d6
commit 2e02e6c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/batchmapper/service_test.go b/pkg/batchmapper/service_test.go
@@ -257,6 +257,13 @@ func TestService_MapFnStream(t *testing.T) {
 
 			wg.Add(1)
 			go func() {
+				defer func() {
+					f := recover()
+					if f != nil {
+						err = fmt.Errorf("got a panic")
+						wg.Done()
+					}
+				}()
 				defer wg.Done()
 				err = fs.BatchMapFn(udfBatchMapFnStream)
 				close(outputCh)