Function newStatsTestStubServer

stats/stats_test.go:118–196

()

}

func newStatsTestStubServer() *stubserver.StubServer {
	return &stubserver.StubServer{
		UnaryCallF: func(ctx context.Context, in *testpb.SimpleRequest) (*testpb.SimpleResponse, error) {
			if err := grpc.SendHeader(ctx, testHeaderMetadata); err != nil {
				return nil, status.Errorf(status.Code(err), "grpc.SendHeader(_, %v) = %v, want <nil>", testHeaderMetadata, err)
			}
			if err := grpc.SetTrailer(ctx, testTrailerMetadata); err != nil {
				return nil, status.Errorf(status.Code(err), "grpc.SetTrailer(_, %v) = %v, want <nil>", testTrailerMetadata, err)
			}

			if id := payloadToID(in.Payload); id == errorID {
				return nil, fmt.Errorf("got error id: %v", id)
			}

			return &testpb.SimpleResponse{Payload: in.Payload}, nil
		},
		FullDuplexCallF: func(stream testgrpc.TestService_FullDuplexCallServer) error {
			if err := stream.SendHeader(testHeaderMetadata); err != nil {
				return status.Errorf(status.Code(err), "%v.SendHeader(%v) = %v, want %v", stream, testHeaderMetadata, err, nil)
			}
			stream.SetTrailer(testTrailerMetadata)
			for {
				in, err := stream.Recv()
				if err == io.EOF {
					// read done.
					return nil
				}
				if err != nil {
					return err
				}

				if id := payloadToID(in.Payload); id == errorID {
					return fmt.Errorf("got error id: %v", id)
				}

				if err := stream.Send(&testpb.StreamingOutputCallResponse{Payload: in.Payload}); err != nil {
					return err
				}
			}
		},
		StreamingInputCallF: func(stream testgrpc.TestService_StreamingInputCallServer) error {
			if err := stream.SendHeader(testHeaderMetadata); err != nil {
				return status.Errorf(status.Code(err), "%v.SendHeader(%v) = %v, want %v", stream, testHeaderMetadata, err, nil)
			}
			stream.SetTrailer(testTrailerMetadata)
			for {
				in, err := stream.Recv()
				if err == io.EOF {
					// read done.
					return stream.SendAndClose(&testpb.StreamingInputCallResponse{AggregatedPayloadSize: 0})
				}
				if err != nil {
					return err
				}

				if id := payloadToID(in.Payload); id == errorID {
					return fmt.Errorf("got error id: %v", id)
				}