Merge pull request #553 from AndrewSirenko/AddProtobufMatcher

Add Protobuf Matcher utility

Author: k8s-ci-robot

go.mod

@@ -10,6 +10,7 @@ require (
 	github.com/onsi/ginkgo/v2 v2.13.1
 	github.com/onsi/gomega v1.30.0
 	google.golang.org/grpc v1.65.0
+	google.golang.org/protobuf v1.34.1
 	gopkg.in/yaml.v2 v2.4.0
 	k8s.io/klog/v2 v2.130.1
 )
@@ -24,6 +25,5 @@ require (
 	golang.org/x/text v0.15.0 // indirect
 	golang.org/x/tools v0.14.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240528184218-531527333157 // indirect
-	google.golang.org/protobuf v1.34.1 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

utils/protobuf_matcher.go

@@ -0,0 +1,33 @@
+package utils
+
+import (
+	"github.com/golang/mock/gomock"
+	"google.golang.org/protobuf/encoding/prototext"
+	"google.golang.org/protobuf/proto"
+)
+
+// Protobuf returns a Matcher that relies upon proto.Equal to compare Protobuf messages
+// Example usage with mocked request:
+//
+// example.EXPECT().ExampleRequest(Protobuf(requestMsg)).Return(responseMsg, nil).AnyTimes()
+func Protobuf(msg proto.Message) gomock.Matcher {
+	return &ProtobufMatcher{msg}
+}
+
+type ProtobufMatcher struct {
+	msg proto.Message
+}
+
+var _ gomock.Matcher = &ProtobufMatcher{}
+
+func (p *ProtobufMatcher) Matches(x interface{}) bool {
+	otherMsg, ok := x.(proto.Message)
+	if !ok {
+		return false
+	}
+	return proto.Equal(p.msg, otherMsg)
+}
+
+func (p *ProtobufMatcher) String() string {
+	return prototext.Format(p.msg)
+}