def setUp(self): patch() self._tracer = get_dummy_tracer() Pin.override(grpc, tracer=self._tracer) self._server = grpc.server(logging_pool.pool(2)) self._server.add_insecure_port('[::]:%d' % (GRPC_PORT)) add_HelloServicer_to_server(SendBackDatadogHeaders(), self._server) self._server.start()
def patch_grpc(): patch() try: yield finally: unpatch()
def setUp(self): super(GrpcTestCase, self).setUp() patch() Pin.override(constants.GRPC_PIN_MODULE_SERVER, tracer=self.tracer) Pin.override(constants.GRPC_PIN_MODULE_CLIENT, tracer=self.tracer) self._start_server()