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))