24 lines
750 B
Python
24 lines
750 B
Python
|
from sanic import Sanic
|
||
|
|
from sanic.response import json, text
|
||
|
|
from sanic.utils import sanic_endpoint_test
|
||
|
|
|
||
|
|
|
||
|
|
def test_vhosts():
|
||
|
|
app = Sanic('test_text')
|
||
|
|
|
||
|
|
@app.route('/', host="example.com")
|
||
|
|
async def handler(request):
|
||
|
|
return text("You're at example.com!")
|
||
|
|
|
||
|
|
@app.route('/', host="subdomain.example.com")
|
||
|
|
async def handler(request):
|
||
|
|
return text("You're at subdomain.example.com!")
|
||
|
|
|
||
|
|
headers = {"Host": "example.com"}
|
||
|
|
request, response = sanic_endpoint_test(app, headers=headers)
|
||
|
|
assert response.text == "You're at example.com!"
|
||
|
|
|
||
|
|
headers = {"Host": "subdomain.example.com"}
|
||
|
|
request, response = sanic_endpoint_test(app, headers=headers)
|
||
|
|
assert response.text == "You're at subdomain.example.com!"
|