Function callToStreaming

server/limits_test.go:67–91

(msgsPerStreamCall int)

}

func callToStreaming(msgsPerStreamCall int) func(ctx context.Context, c FakeServerClient) error {
	return func(ctx context.Context, c FakeServerClient) error {
		rcvd := 0
		s, err := c.StreamSleep(ctx, &emptypb.Empty{})
		if err != nil {
			return err
		}

		for {
			_, err := s.Recv()
			if err != nil {
				if errors.Is(err, io.EOF) {
					break
				}
				return err
			}
			rcvd++
		}

		if rcvd != msgsPerStreamCall {
			return fmt.Errorf("invalid number of received messages: %d", rcvd)
		}
		return nil
	}
}

func TestGrpcLimitCheckUnary(t *testing.T) {
	const msgsPerStreamCall = 5