From b6e0fe8cebbee5114d3b0f17cb917c78018b5425 Mon Sep 17 00:00:00 2001 From: adarsh0728 Date: Fri, 31 Jan 2025 12:37:41 +0530 Subject: [PATCH] fix tests Signed-off-by: adarsh0728 --- pkg/mapper/service_test.go | 6 +++++- pkg/mapstreamer/service_test.go | 6 +++++- pkg/sourcetransformer/service_test.go | 6 +++++- 3 files changed, 15 insertions(+), 3 deletions(-) diff --git a/pkg/mapper/service_test.go b/pkg/mapper/service_test.go index bbdc8c95..9d074686 100644 --- 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 index 1373716e..5629f197 100644 --- 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 index a98bb712..e53c8529 100644 --- 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") }