def setUp(self): self._servicier = _CallbackServicer() self._binding1_method = Mock() self._binding2_method = Mock() self._servicier.register_binding('binding1', self._binding1_method) self._servicier.register_binding('binding2', self._binding2_method) # fake context self.fake_context = MagicMock() self.fake_context.invocation_metadata.return_value = ( ('key1', 'value1'), ('key2', 'value1'), )
def __init__(self, **kwargs): """Inits App object and creates gRPC server. Args: kwargs: arguments to grpc.server() """ self._servicer = _CallbackServicer() if not kwargs: self._server = grpc.server( futures.ThreadPoolExecutor(max_workers=10)) # type: ignore else: self._server = grpc.server(**kwargs) # type: ignore appcallback_service_v1.add_AppCallbackServicer_to_server( self._servicer, self._server)
def setUp(self): self._servicier = _CallbackServicer() self._topic1_method = Mock() self._topic2_method = Mock() self._topic3_method = Mock() self._servicier.register_topic('pubsub1', 'topic1', self._topic1_method, {'session': 'key'}) self._servicier.register_topic('pubsub1', 'topic3', self._topic3_method, {'session': 'key'}) self._servicier.register_topic('pubsub2', 'topic2', self._topic2_method, {'session': 'key'}) self._servicier.register_topic('pubsub2', 'topic3', self._topic3_method, {'session': 'key'}) # fake context self.fake_context = MagicMock() self.fake_context.invocation_metadata.return_value = ( ('key1', 'value1'), ('key2', 'value1'), )
def setUp(self): self._servicier = _CallbackServicer()