From 2a617ca67a6b10f38850843c22be4156c4590a3f Mon Sep 17 00:00:00 2001 From: Arvind Bright Date: Wed, 28 Feb 2024 14:37:10 -0800 Subject: [PATCH] experimental: re-split message/service pb imports (#7011) --- experimental/shared_buffer_pool_test.go | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/experimental/shared_buffer_pool_test.go b/experimental/shared_buffer_pool_test.go index 7c4074e18bd5..df8d82be9bb7 100644 --- a/experimental/shared_buffer_pool_test.go +++ b/experimental/shared_buffer_pool_test.go @@ -32,6 +32,7 @@ import ( "google.golang.org/grpc/internal/stubserver" testgrpc "google.golang.org/grpc/interop/grpc_testing" + testpb "google.golang.org/grpc/interop/grpc_testing" ) type s struct { @@ -68,8 +69,8 @@ func (s) TestRecvBufferPoolStream(t *testing.T) { FullDuplexCallF: func(stream testgrpc.TestService_FullDuplexCallServer) error { for i := 0; i < reqCount; i++ { preparedMsg := &grpc.PreparedMsg{} - if err := preparedMsg.Encode(stream, &testgrpc.StreamingOutputCallResponse{ - Payload: &testgrpc.Payload{ + if err := preparedMsg.Encode(stream, &testpb.StreamingOutputCallResponse{ + Payload: &testpb.Payload{ Body: []byte{'0' + uint8(i)}, }, }); err != nil { @@ -148,9 +149,9 @@ func (s) TestRecvBufferPoolUnary(t *testing.T) { const largeSize = 1024 ss := &stubserver.StubServer{ - UnaryCallF: func(ctx context.Context, in *testgrpc.SimpleRequest) (*testgrpc.SimpleResponse, error) { - return &testgrpc.SimpleResponse{ - Payload: &testgrpc.Payload{ + UnaryCallF: func(ctx context.Context, in *testpb.SimpleRequest) (*testpb.SimpleResponse, error) { + return &testpb.SimpleResponse{ + Payload: &testpb.Payload{ Body: make([]byte, largeSize), }, }, nil @@ -172,8 +173,8 @@ func (s) TestRecvBufferPoolUnary(t *testing.T) { for i := 0; i < reqCount; i++ { if _, err := ss.Client.UnaryCall( ctx, - &testgrpc.SimpleRequest{ - Payload: &testgrpc.Payload{ + &testpb.SimpleRequest{ + Payload: &testpb.Payload{ Body: make([]byte, largeSize), }, },