def test_application_as_handler(evloop, mock_socket):
app = App(name="testapp")
@app.route("/")
class Echo(BaseHandler):
@asyncio.coroutine
def handle(self, ws, **kwargs):
from_client = yield from ws.recv()
yield from ws.send(from_client)
handler = app.as_handler(loop=evloop)
@asyncio.coroutine
def run_handler():
yield from handler(mock_socket, "/")
assert mock_socket.recv.called
assert mock_socket.send.called
mock_socket.send.assert_called_with("hello")
evloop.run_until_complete(run_handler())
def test_application_as_handler(evloop, mock_socket):
app = App(name="testapp")
@app.route("/")
class Echo(BaseHandler):
@asyncio.coroutine
def handle(self, ws, **kwargs):
from_client = yield from ws.recv()
yield from ws.send(from_client)
handler = app.as_handler(loop=evloop)
@asyncio.coroutine
def run_handler():
yield from handler(mock_socket, "/")
assert mock_socket.recv.called
assert mock_socket.send.called
mock_socket.send.assert_called_with("hello")
evloop.run_until_complete(run_handler())
def test_application_as_handler_unknown_route(evloop, mock_socket):
app = App(name="testapp")
@app.route("/")
class Echo(BaseHandler):
@asyncio.coroutine
def handle(self, ws, **kwargs):
from_client = yield from ws.recv()
yield from ws.send(from_client)
handler = app.as_handler(loop=evloop)
@asyncio.coroutine
def run_handler():
yield from handler(mock_socket, "/echo")
assert mock_socket.close.called
error = "could not resolve the path /echo tried "\
"{}".format(list(app.router.routes.keys()))
mock_socket.close.assert_called_with(code=1011, reason=error)
evloop.run_until_complete(run_handler())
def test_application_as_handler_unknown_route(evloop, mock_socket):
app = App(name="testapp")
@app.route("/")
class Echo(BaseHandler):
@asyncio.coroutine
def handle(self, ws, **kwargs):
from_client = yield from ws.recv()
yield from ws.send(from_client)
handler = app.as_handler(loop=evloop)
@asyncio.coroutine
def run_handler():
yield from handler(mock_socket, "/echo")
assert mock_socket.close.called
error = "could not resolve the path /echo tried "\
"{}".format(list(app.router.routes.keys()))
mock_socket.close.assert_called_with(code=1011, reason=error)
evloop.run_until_complete(run_handler())
