From da42e808643a1fc6ce913b2ce8bcb353e47c6544 Mon Sep 17 00:00:00 2001 From: adarsh0728 Date: Mon, 27 Jan 2025 10:06:24 +0530 Subject: [PATCH] fix: unit test -- TestService_SourceTransformFn_Panic Signed-off-by: adarsh0728 --- pkg/sourcetransformer/service_test.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pkg/sourcetransformer/service_test.go b/pkg/sourcetransformer/service_test.go index 79449c9a..a98bb712 100644 --- a/pkg/sourcetransformer/service_test.go +++ b/pkg/sourcetransformer/service_test.go @@ -259,9 +259,10 @@ func TestService_SourceTransformFn_Multiple_Messages(t *testing.T) { } func TestService_SourceTransformFn_Panic(t *testing.T) { + panicMssg := "transformer panicked" svc := &Service{ Transformer: SourceTransformFunc(func(ctx context.Context, keys []string, datum Datum) Messages { - panic("transformer panicked") + panic(panicMssg) }), // panic in the transformer causes the server to send a shutdown signal to shutdownCh channel. // The function that errgroup runs in a goroutine will be blocked until this shutdown signal is received somewhere else. @@ -293,6 +294,6 @@ 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) - expectedStatus := status.Convert(status.Errorf(codes.Internal, errTransformerPanic.Error())) + expectedStatus := status.Convert(status.Errorf(codes.Internal, "%s: %v", errTransformerPanic, panicMssg)) require.Equal(t, expectedStatus, gotStatus) }