• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• grpclib-master
  • .github
    • workflows
      • release.yaml
  • Makefile
  • pi.yaml
  • examples
    • helloworld
      • helloworld.proto
      • client.py
      • server.py
      • helloworld_grpc.py
      • helloworld_pb2_grpc.py
      • __init__.py
      • helloworld_pb2.py
    • reflection
      • server.py
      • __init__.py
    • streaming
      • helloworld.proto
      • client.py
      • server.py
      • helloworld_grpc.py
      • __init__.py
      • helloworld_pb2.py
    • README.rst
    • mtls
      • keys
        • Makefile
      • client.py
      • server.py
      • __init__.py
    • multiproc
      • client.py
      • server.py
      • primes_grpc.py
      • primes.proto
      • __init__.py
      • primes_pb2.py
    • tracing
      • client.py
      • server.py
      • __init__.py
    • _reference
      • client.py
      • server.py
      • __init__.py
  • setup.txt
  • .readthedocs.yml
  • README.rst
  • setup.py
  • grpclib
    • testing.py
    • health
      • v1
        • health_grpc.py
        • health_pb2.py
        • __init__.py
        • health.proto
      • check.py
      • __init__.py
      • service.py
    • reflection
      • v1
        • reflection.proto
        • reflection_pb2.py
        • reflection_grpc.py
        • __init__.py
      • v1alpha
        • reflection.proto
        • reflection_pb2.py
        • reflection_grpc.py
        • __init__.py
      • __init__.py
      • _deprecated.py
      • service.py
    • exceptions.py
    • channelz
      • v1
        • channelz_grpc.py
        • channelz.proto
        • __init__.py
        • channelz_pb2.py
      • __init__.py
      • service.py
    • config.py
    • plugin
      • __init__.py
      • main.py
    • py.typed
    • client.py
    • server.py
    • events.py
    • metadata.py
    • _compat.py
    • __init__.py
    • utils.py
    • const.py
    • _typing.py
    • encoding
      • proto.py
      • __init__.py
      • base.py
    • _registry.py
    • stream.py
    • protocol.py
  • setup.cfg
  • .travis.yml
  • tests
    • dummy.proto
    • dummy_pb2.py
    • test_server_handler.py
    • test_utils.py
    • test_memory.py
    • test_metadata.py
    • test_reflection.py
    • test_compat.py
    • test_events.py
    • test_status_details_codec.py
    • test_protocol.py
    • test_health_service.py
    • test_client_stream.py
    • test_ping.py
    • test_testing.py
    • test_config.py
    • test_functional.py
    • test_codec.py
    • test_health_check.py
    • test_client_events.py
    • dummy_grpc.py
    • conn.py
    • test_server_events.py
    • test_client_methods.py
    • test_stats.py
    • stubs.py
    • test_client_channel.py
    • conftest.py
    • test_server_stream.py
  • scripts
    • release_check.sh
    • bench.py
  • .gitignore
  • docs
    • client.rst
    • errors.rst
    • server.rst
    • overview.rst
    • config.rst
    • reflection.rst
    • changelog
      • index.rst
    • diagram.txt
    • health.rst
    • metadata.rst
    • testing.rst
    • index.rst
    • conf.py
    • encoding.rst
    • _static
      • diagram.html
      • style.css
      • diagram.css
    • events.rst
  • LICENSE.txt
  • requirements
    • release.txt
    • check.in
    • test.txt
    • runtime.txt
    • lint.txt
    • docs.txt
    • docs.in
    • lint.in
    • release.in
    • runtime.in
    • test.in
    • check.txt
  • tox.ini

import gc
import sys
import asyncio.tasks

import pytest

from grpclib.const import Status
from grpclib.testing import ChannelFor
from grpclib.exceptions import GRPCError

from conn import ClientServer
from dummy_pb2 import DummyRequest, DummyReply
from dummy_grpc import DummyServiceBase, DummyServiceStub


class DummyService(DummyServiceBase):

    async def UnaryUnary(self, stream):
        request = await stream.recv_message()
        assert request == DummyRequest(value='ping')
        await stream.send_message(DummyReply(value='pong'))

    async def UnaryStream(self, stream):
        raise GRPCError(Status.UNIMPLEMENTED)

    async def StreamUnary(self, stream):
        raise GRPCError(Status.UNIMPLEMENTED)

    async def StreamStream(self, stream):
        raise GRPCError(Status.UNIMPLEMENTED)


def collect():
    objects = gc.get_objects()
    return {id(obj): obj for obj in objects}


def _check(type_name):
    """Utility function to debug references"""
    import objgraph

    objects = objgraph.by_type(type_name)
    if objects:
        obj = objects[0]
        objgraph.show_backrefs(obj, max_depth=3, filename='graph.png')


@pytest.mark.skipif(sys.version_info < (3, 7), reason='Python < 3.7')
def test_connection():
    loop = asyncio.new_event_loop()

    async def example():
        async with ChannelFor([DummyService()]) as channel:
            stub = DummyServiceStub(channel)
            await stub.UnaryUnary(DummyRequest(value='ping'))

    # warm up
    loop.run_until_complete(example())

    gc.collect()
    gc.disable()
    try:
        pre = set(collect())
        loop.run_until_complete(example())
        loop.stop()
        loop.close()
        post = collect()

        diff = set(post).difference(pre)
        diff.discard(id(pre))
        diff.discard(id(asyncio.tasks._current_tasks))
        if diff:
            for i in diff:
                try:
                    print(post[i])
                except Exception:
                    print('...')
            raise AssertionError('Memory leak detected')
    finally:
        gc.enable()


@pytest.mark.asyncio
@pytest.mark.skipif(sys.version_info < (3, 7), reason='Python < 3.7')
async def test_stream():
    cs = ClientServer(DummyService, DummyServiceStub)
    async with cs as (_, stub):
        await stub.UnaryUnary(DummyRequest(value='ping'))
        handler = next(iter(cs.server._handlers))
        handler.__gc_collect__()
        gc.collect()
        gc.disable()
        try:
            pre = set(collect())
            await stub.UnaryUnary(DummyRequest(value='ping'))
            handler.__gc_collect__()
            post = collect()

            diff = set(post).difference(pre)
            diff.discard(id(pre))
            for i in diff:
                try:
                    print(repr(post[i])[:120])
                except Exception:
                    print('...')
                else:
                    if 'grpclib.' in repr(post[i]):
                        raise AssertionError('Memory leak detected')
        finally:
            gc.enable()