github.com/bartle-stripe/trillian@v1.2.1/testonly/matchers/proto.go (about) 1 // Copyright 2017 Google Inc. All Rights Reserved. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package matchers 16 17 import ( 18 "fmt" 19 20 "github.com/golang/mock/gomock" 21 "github.com/golang/protobuf/proto" 22 ) 23 24 type protoEqual struct { 25 msg proto.Message 26 } 27 28 // ProtoEqual returns a matcher that compares protobuf messages. 29 func ProtoEqual(m proto.Message) gomock.Matcher { 30 return protoEqual{msg: m} 31 } 32 33 func (pe protoEqual) Matches(msg interface{}) bool { 34 m, ok := msg.(proto.Message) 35 if !ok { 36 return false 37 } 38 return proto.Equal(m, pe.msg) 39 } 40 41 func (pe protoEqual) String() string { 42 return fmt.Sprintf("is equal to %s", pe.msg) 43 }