Skip to content

Commit 2a617ca

Browse files
authored
experimental: re-split message/service pb imports (#7011)
1 parent 99ded5c commit 2a617ca

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

experimental/shared_buffer_pool_test.go

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ import (
3232
"google.golang.org/grpc/internal/stubserver"
3333

3434
testgrpc "google.golang.org/grpc/interop/grpc_testing"
35+
testpb "google.golang.org/grpc/interop/grpc_testing"
3536
)
3637

3738
type s struct {
@@ -68,8 +69,8 @@ func (s) TestRecvBufferPoolStream(t *testing.T) {
6869
FullDuplexCallF: func(stream testgrpc.TestService_FullDuplexCallServer) error {
6970
for i := 0; i < reqCount; i++ {
7071
preparedMsg := &grpc.PreparedMsg{}
71-
if err := preparedMsg.Encode(stream, &testgrpc.StreamingOutputCallResponse{
72-
Payload: &testgrpc.Payload{
72+
if err := preparedMsg.Encode(stream, &testpb.StreamingOutputCallResponse{
73+
Payload: &testpb.Payload{
7374
Body: []byte{'0' + uint8(i)},
7475
},
7576
}); err != nil {
@@ -148,9 +149,9 @@ func (s) TestRecvBufferPoolUnary(t *testing.T) {
148149
const largeSize = 1024
149150

150151
ss := &stubserver.StubServer{
151-
UnaryCallF: func(ctx context.Context, in *testgrpc.SimpleRequest) (*testgrpc.SimpleResponse, error) {
152-
return &testgrpc.SimpleResponse{
153-
Payload: &testgrpc.Payload{
152+
UnaryCallF: func(ctx context.Context, in *testpb.SimpleRequest) (*testpb.SimpleResponse, error) {
153+
return &testpb.SimpleResponse{
154+
Payload: &testpb.Payload{
154155
Body: make([]byte, largeSize),
155156
},
156157
}, nil
@@ -172,8 +173,8 @@ func (s) TestRecvBufferPoolUnary(t *testing.T) {
172173
for i := 0; i < reqCount; i++ {
173174
if _, err := ss.Client.UnaryCall(
174175
ctx,
175-
&testgrpc.SimpleRequest{
176-
Payload: &testgrpc.Payload{
176+
&testpb.SimpleRequest{
177+
Payload: &testpb.Payload{
177178
Body: make([]byte, largeSize),
178179
},
179180
},

0 commit comments

Comments
 (0)