fix tests

Signed-off-by: adarsh0728 <[email protected]>
numaproj · Jan 31, 2025 · b6e0fe8 · b6e0fe8
1 parent f3fc30e
commit b6e0fe8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/pkg/mapper/service_test.go b/pkg/mapper/service_test.go
@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"net"
+	"strings"
 	"testing"
 	"time"
 
@@ -289,6 +290,9 @@ func TestService_MapFn_Panic(t *testing.T) {
 	_, err = stream.Recv()
 	require.Error(t, err, "Expected error while receiving message from the stream")
 	gotStatus, _ := status.FromError(err)
+	gotMessage := gotStatus.Message()
 	expectedStatus := status.Convert(status.Errorf(codes.Internal, "%s: %v", errMapHandlerPanic, panicMssg))
-	require.Equal(t, expectedStatus, gotStatus)
+	expectedMessage := expectedStatus.Message()
+	require.Equal(t, expectedStatus.Code(), gotStatus.Code(), "Expected error codes to be equal")
+	require.True(t, strings.HasPrefix(gotMessage, expectedMessage), "Expected error message to start with the expected message")
 }
diff --git a/pkg/mapstreamer/service_test.go b/pkg/mapstreamer/service_test.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 	"net"
+	"strings"
 	"testing"
 	"time"
 
@@ -358,8 +359,11 @@ func TestService_MapFn_Panic(t *testing.T) {
 	_, err = stream.Recv()
 	require.Error(t, err, "Expected error while receiving message from the stream")
 	gotStatus, _ := status.FromError(err)
+	gotMessage := gotStatus.Message()
 	expectedStatus := status.Convert(status.Errorf(codes.Internal, "%s: %v", errMapStreamHandlerPanic, panicMssg))
-	require.Equal(t, expectedStatus, gotStatus)
+	expectedMessage := expectedStatus.Message()
+	require.Equal(t, expectedStatus.Code(), gotStatus.Code(), "Expected error codes to be equal")
+	require.True(t, strings.HasPrefix(gotMessage, expectedMessage), "Expected error message to start with the expected message")
 }
 
 func TestService_MapFn_MultipleRequestsAndResponses(t *testing.T) {

diff --git a/pkg/sourcetransformer/service_test.go b/pkg/sourcetransformer/service_test.go
@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"net"
+	"strings"
 	"testing"
 	"time"
 
@@ -294,6 +295,9 @@ func TestService_SourceTransformFn_Panic(t *testing.T) {
 	_, err = stream.Recv()
 	require.Error(t, err, "Expected error while receiving message from the stream")
 	gotStatus, _ := status.FromError(err)
+	gotMessage := gotStatus.Message()
 	expectedStatus := status.Convert(status.Errorf(codes.Internal, "%s: %v", errTransformerPanic, panicMssg))
-	require.Equal(t, expectedStatus, gotStatus)
+	expectedMessage := expectedStatus.Message()
+	require.Equal(t, expectedStatus.Code(), gotStatus.Code(), "Expected error codes to be equal")
+	require.True(t, strings.HasPrefix(gotMessage, expectedMessage), "Expected error message to start with the expected message")
 }