def main(): # pragma: no cover
import sys
parser = argparse.ArgumentParser()
parser.add_argument(
"--port",
dest="port", default=8888, type=int,
)
parser.add_argument(
"--host",
dest="host", default="127.0.0.1"
)
args = parser.parse_args()
client = TChannel('%s:%d' % (args.host, args.port))
# TODO: service=test_as_raw
handler = RequestDispatcher()
client.host(handler)
# TODO: do we need to implement these differently?
handler.register("echo", echo)
handler.register("streaming_echo", echo)
client.listen()
print 'listening on', client.hostport
sys.stdout.flush()
tornado.ioloop.IOLoop.instance().start()
def test_dispatch():
dispatcher = RequestDispatcher()
dispatcher.register(
r"/hello",
dummy_endpoint,
RawSerializer(),
RawSerializer(),
)
endpoint = dispatcher.handlers.get("/hello")[0]
assert endpoint == dummy_endpoint
def test_dispatch():
dispatcher = RequestDispatcher()
dispatcher.register(
r"/hello",
dummy_endpoint
)
@dispatcher.route(r"/")
def dummy_endpoint1(request, response, proxy):
pass
endpoint = dispatcher.endpoints.get("/hello")
assert endpoint == dispatcher.default_broker.handle_call
assert dispatcher.default_broker.endpoint.get('/hello') == dummy_endpoint
endpoint = dispatcher.endpoints.get("/")
assert endpoint == dispatcher.default_broker.handle_call
assert dispatcher.default_broker.endpoint.get('/') == dummy_endpoint1
def dispatcher():
return RequestDispatcher()
