encoding/proto: make sure proto imports are renamed (#6934)

This commit is contained in:
Doug Fawley 2024-01-23 11:18:17 -08:00 committed by GitHub
parent 4d792e5b29
commit d66bc9b79c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 7 deletions

View File

@ -24,7 +24,7 @@ import (
"github.com/golang/protobuf/proto"
"google.golang.org/grpc/encoding"
"google.golang.org/grpc/test/codec_perf"
pb "google.golang.org/grpc/test/codec_perf"
)
func setupBenchmarkProtoCodecInputs(payloadBaseSize uint32) []proto.Message {
@ -40,7 +40,7 @@ func setupBenchmarkProtoCodecInputs(payloadBaseSize uint32) []proto.Message {
protoStructs := make([]proto.Message, 0)
for _, p := range payloadSuffixes {
ps := &codec_perf.Buffer{}
ps := &pb.Buffer{}
ps.Body = append(payloadBase, p...)
protoStructs = append(protoStructs, ps)
}
@ -91,7 +91,7 @@ func fastMarshalAndUnmarshal(codec encoding.Codec, protoStruct proto.Message, b
if err != nil {
b.Errorf("codec.Marshal(_) returned an error")
}
res := codec_perf.Buffer{}
res := pb.Buffer{}
if err := codec.Unmarshal(marshaledBytes, &res); err != nil {
b.Errorf("codec.Unmarshal(_) returned an error")
}

View File

@ -25,11 +25,11 @@ import (
"google.golang.org/grpc/encoding"
"google.golang.org/grpc/internal/grpctest"
"google.golang.org/grpc/test/codec_perf"
pb "google.golang.org/grpc/test/codec_perf"
)
func marshalAndUnmarshal(t *testing.T, codec encoding.Codec, expectedBody []byte) {
p := &codec_perf.Buffer{}
p := &pb.Buffer{}
p.Body = expectedBody
marshalledBytes, err := codec.Marshal(p)
@ -99,8 +99,8 @@ func (s) TestStaggeredMarshalAndUnmarshalUsingSamePool(t *testing.T) {
expectedBody1 := []byte{1, 2, 3}
expectedBody2 := []byte{4, 5, 6}
proto1 := codec_perf.Buffer{Body: expectedBody1}
proto2 := codec_perf.Buffer{Body: expectedBody2}
proto1 := pb.Buffer{Body: expectedBody1}
proto2 := pb.Buffer{Body: expectedBody2}
var m1, m2 []byte
var err error