def thrift1(request): host_port = request.body.hostport res = yield tchannel.thrift( thrift_service.X.thrift2(), hostport=host_port, ) raise tornado.gen.Return(Response(res.body))
def endpoint(request): assert isinstance(request, Request) assert request.headers == 'req headers' assert request.body == 'req body' return Response('resp body', 'resp headers')
def _call_http(): response = yield http_client.fetch(base_url) if response.code != 200: raise Exception('Downstream http service returned code=%s: %s' % ( response.code, response.body )) raise tornado.gen.Return(Response(body=response.body))
def testStruct(request): assert request.headers == {'req': 'header'} assert request.body.thing.string_thing == 'req string' return Response( server_ttypes.Xtruct(string_thing="resp string"), headers={'resp': 'header'}, )
def handler1(request): host_port = request.body res = yield tchannel.json( service='handler2', hostport=host_port, endpoint="endpoint2", ) raise tornado.gen.Return(Response(res.body))
def handler1(request): hostport = request.headers res = yield tchannel.raw( service='handler2', hostport=hostport, endpoint="endpoint2", ) raise tornado.gen.Return(Response(res.body, "from handler1"))
def testStruct(request): assert isinstance(request, Request) assert request.headers == {'req': 'header'} assert request.body.thing.string_thing == 'req string' return Response( ThriftTest.Xtruct(string_thing="resp string"), headers={'resp': 'header'}, )
def endpoint1(request): yield server.call( scheme=schemes.RAW, service='server', arg1='endpoint2', arg2='req headers', arg3='req body', hostport=server.hostport, ) context[0] = get_current_context() raise gen.Return(Response('resp body', 'resp headers'))
def thrift4(request): print 'thrift4 hit' host_port = request.body.hostport res = tchannel.raw( service='test-service', endpoint='raw2', hostport=host_port, ) res = yield res print 'result', res raise tornado.gen.Return(Response(res.body))
def thrift4(request): print 'thrift4 hit' host_port = request.body.hostport res = tchannel.raw( service='test-service', endpoint='raw2', hostport=host_port, headers='some value that will not be parsed as a dict', ) res = yield res print 'result', res raise tornado.gen.Return(Response(res.body))
def test_advertise_should_take_a_router_file(): host_path = os.path.join( os.path.dirname(os.path.realpath(__file__)), 'data/hosts.json', ) tchannel = TChannel(name='client') with open(host_path, 'r') as json_data: routers = json.load(json_data) with ( patch( 'tchannel.tornado.TChannel.advertise', autospec=True, ) ) as mock_advertise: f = gen.Future() mock_advertise.return_value = f f.set_result(Response()) tchannel.advertise(router_file=host_path) mock_advertise.assert_called_once_with(ANY, routers=routers, name=ANY, timeout=ANY)
def testVoid(request): assert request.headers == {'req': 'header'} return Response(headers={'resp': 'header'})
def json_echo(request): headers = request.headers body = request.body return Response(body, headers)
def handler(request): return Response( status=1, headers=b'\x00\x00', body=b'\x00', )
def endpoint(request): assert request.headers == {'req': 'header'} assert request.body == {'req': 'body'} return Response({'resp': 'body'}, headers={'resp': 'header'})
def endpoint(request): assert request.headers == 'req headers' assert request.body == 'req body' return Response('resp body', headers='resp headers')
def testString(request): assert request.headers == {'req': 'header'} assert request.body.thing == 'req' return Response('resp', headers={'resp': 'header'})
def handle_expected_endpoint(request): response = Response() return expectation.execute(request, response)
def endpoint2(request): context[1] = get_current_context() return Response('resp body', 'resp headers')
def json_trace_handler(request): res = yield _process_request(request.body) raise tornado.gen.Return(Response(body=res))