4efd450b32
* Add tests for remove_route() * Add tests for sanic/router.py * Add tests for sanic/cookies.py * Disable reset logging in test_logging.py * Add tests for sanic/request.py * Add tests for ContentRangeHandler * Add tests for exception at response middleware * Fix cached_handlers for ErrorHandler.lookup() * Add test for websocket request timeout * Add tests for getting cookies of StreamResponse, Remove some unused variables in tests/test_cookies.py * Add tests for nested error handle
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from sanic.response import text
|
|
|
|
|
|
def test_vhosts(app):
|
|
|
|
@app.route('/', host="example.com")
|
|
async def handler1(request):
|
|
return text("You're at example.com!")
|
|
|
|
@app.route('/', host="subdomain.example.com")
|
|
async def handler2(request):
|
|
return text("You're at subdomain.example.com!")
|
|
|
|
headers = {"Host": "example.com"}
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.text == "You're at example.com!"
|
|
|
|
headers = {"Host": "subdomain.example.com"}
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.text == "You're at subdomain.example.com!"
|
|
|
|
|
|
def test_vhosts_with_list(app):
|
|
|
|
@app.route('/', host=["hello.com", "world.com"])
|
|
async def handler(request):
|
|
return text("Hello, world!")
|
|
|
|
headers = {"Host": "hello.com"}
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.text == "Hello, world!"
|
|
|
|
headers = {"Host": "world.com"}
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.text == "Hello, world!"
|
|
|
|
|
|
def test_vhosts_with_defaults(app):
|
|
|
|
@app.route('/', host="hello.com")
|
|
async def handler1(request):
|
|
return text("Hello, world!")
|
|
|
|
@app.route('/')
|
|
async def handler2(request):
|
|
return text("default")
|
|
|
|
headers = {"Host": "hello.com"}
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.text == "Hello, world!"
|
|
|
|
request, response = app.test_client.get('/')
|
|
assert response.text == "default"
|
|
|
|
|
|
def test_remove_vhost_route(app):
|
|
|
|
@app.route('/', host="example.com")
|
|
async def handler1(request):
|
|
return text("You're at example.com!")
|
|
|
|
headers = {"Host": "example.com"}
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.status == 200
|
|
|
|
app.remove_route('/', host="example.com")
|
|
request, response = app.test_client.get('/', headers=headers)
|
|
assert response.status == 404
|