@@ -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
3738type 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