66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import asyncio
|
|
|
|
from sanic_testing.testing import SanicTestClient
|
|
|
|
from sanic.blueprints import Blueprint
|
|
|
|
|
|
def test_routes_with_host(app):
|
|
@app.route("/")
|
|
@app.route("/", name="hostindex", host="example.com")
|
|
@app.route("/path", name="hostpath", host="path.example.com")
|
|
def index(request):
|
|
pass
|
|
|
|
assert app.url_for("index") == "/"
|
|
assert app.url_for("hostindex") == "/"
|
|
assert app.url_for("hostpath") == "/path"
|
|
assert app.url_for("hostindex", _external=True) == "http://example.com/"
|
|
assert (
|
|
app.url_for("hostpath", _external=True)
|
|
== "http://path.example.com/path"
|
|
)
|
|
|
|
|
|
def test_websocket_bp_route_name(app):
|
|
"""Tests that blueprint websocket route is named."""
|
|
event = asyncio.Event()
|
|
bp = Blueprint("test_bp", url_prefix="/bp")
|
|
|
|
@bp.get("/main")
|
|
async def main(request):
|
|
...
|
|
|
|
@bp.websocket("/route")
|
|
async def test_route(request, ws):
|
|
event.set()
|
|
|
|
@bp.websocket("/route2")
|
|
async def test_route2(request, ws):
|
|
event.set()
|
|
|
|
@bp.websocket("/route3", name="foobar_3")
|
|
async def test_route3(request, ws):
|
|
event.set()
|
|
|
|
app.blueprint(bp)
|
|
|
|
uri = app.url_for("test_bp.main")
|
|
assert uri == "/bp/main"
|
|
|
|
uri = app.url_for("test_bp.test_route")
|
|
assert uri == "/bp/route"
|
|
request, response = SanicTestClient(app).websocket(uri)
|
|
assert response.opened is True
|
|
assert event.is_set()
|
|
|
|
event.clear()
|
|
uri = app.url_for("test_bp.test_route2")
|
|
assert uri == "/bp/route2"
|
|
request, response = SanicTestClient(app).websocket(uri)
|
|
assert response.opened is True
|
|
assert event.is_set()
|
|
|
|
uri = app.url_for("test_bp.foobar_3")
|
|
assert uri == "/bp/route3"
|