예제 #1
0
    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'),
        )
예제 #2
0
    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)
예제 #3
0
    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'),
        )
예제 #4
0
 def setUp(self):
     self._servicier = _CallbackServicer()