Class RpcServerWrapper

tests/unittest/executor/test_rpc.py:13–26

A helper class to wrap the RPCServer and manage its lifecycle.

class RpcServerWrapper(RPCServer):
    """ A helper class to wrap the RPCServer and manage its lifecycle. """

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.addr = get_unique_ipc_addr()

    def __enter__(self):
        self.bind(self.addr)
        self.start()
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        self.shutdown()


class TestRpcBasics: