Make StringifySnapshot use cupaloy format

gothunder · Sep 29, 2023 · 404ea41 · 404ea41
1 parent 729c57c
commit 404ea41
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/grpc/snapshot.go b/pkg/grpc/snapshot.go
@@ -1,6 +1,10 @@
 package grpc
 
-import "strings"
+import (
+	"fmt"
+	"reflect"
+	"strings"
+)
 
 type ProtoMessage interface {
 	String() string
@@ -9,8 +13,9 @@ type ProtoMessage interface {
 // StringifySnapshot converts a protobuf response to a string and removes all double spaces.
 // This is needed because of a pseudo-random effect https://github.com/golang/protobuf/issues/1121
 func StringifySnapshot(resp ProtoMessage) string {
+	respType := reflect.TypeOf(resp).String()
 	respStr := resp.String()
 	respStr = strings.ReplaceAll(respStr, "  ", " ")
 
-	return respStr
+	return fmt.Sprintf("(%s)(%s)", respType, respStr)
 }