starlette/tests/test_routing.py

94 lines
2.4 KiB
Python
Raw Normal View History

2018-07-12 13:09:05 +00:00
from starlette import Response, TestClient
from starlette.routing import Path, PathPrefix, Router, ProtocolRouter
from starlette.websockets import WebSocketSession
2018-06-25 21:23:40 +00:00
def homepage(scope):
2018-06-25 21:26:01 +00:00
return Response("Hello, world", media_type="text/plain")
2018-06-25 21:23:40 +00:00
def users(scope):
2018-06-25 21:26:01 +00:00
return Response("All users", media_type="text/plain")
2018-06-25 21:23:40 +00:00
def user(scope):
2018-06-25 21:26:01 +00:00
content = "User " + scope["kwargs"]["username"]
return Response(content, media_type="text/plain")
2018-06-25 21:23:40 +00:00
2018-06-26 09:09:35 +00:00
def staticfiles(scope):
return Response("xxxxx", media_type="image/png")
2018-06-25 21:26:01 +00:00
app = Router(
[
2018-06-26 09:09:59 +00:00
Path("/", app=homepage, methods=["GET"]),
2018-06-25 21:26:01 +00:00
PathPrefix(
"/users", app=Router([Path("", app=users), Path("/{username}", app=user)])
),
2018-06-26 09:09:59 +00:00
PathPrefix("/static", app=staticfiles, methods=["GET"]),
2018-06-25 21:26:01 +00:00
]
)
2018-06-25 21:23:40 +00:00
def test_router():
client = TestClient(app)
2018-06-25 21:26:01 +00:00
response = client.get("/")
2018-06-25 21:23:40 +00:00
assert response.status_code == 200
2018-06-25 21:26:01 +00:00
assert response.text == "Hello, world"
2018-06-25 21:23:40 +00:00
2018-06-26 09:09:35 +00:00
response = client.post("/")
assert response.status_code == 406
assert response.text == "Method not allowed"
2018-06-25 21:26:01 +00:00
response = client.get("/foo")
2018-06-25 21:23:40 +00:00
assert response.status_code == 404
2018-06-25 21:26:01 +00:00
assert response.text == "Not found"
2018-06-25 21:23:40 +00:00
2018-06-25 21:26:01 +00:00
response = client.get("/users")
2018-06-25 21:23:40 +00:00
assert response.status_code == 200
2018-06-25 21:26:01 +00:00
assert response.text == "All users"
2018-06-25 21:23:40 +00:00
2018-06-25 21:26:01 +00:00
response = client.get("/users/tomchristie")
2018-06-25 21:23:40 +00:00
assert response.status_code == 200
2018-06-25 21:26:01 +00:00
assert response.text == "User tomchristie"
2018-06-26 09:09:35 +00:00
response = client.get("/static/123")
assert response.status_code == 200
assert response.text == "xxxxx"
response = client.post("/static/123")
assert response.status_code == 406
assert response.text == "Method not allowed"
def http_endpoint(scope):
return Response("Hello, world", media_type="text/plain")
def websocket_endpoint(scope):
async def asgi(receive, send):
session = WebSocketSession(scope, receive, send)
await session.accept()
await session.send_json({"hello": "world"})
await session.close()
return asgi
mixed_protocol_app = ProtocolRouter(
{"http": http_endpoint, "websocket": websocket_endpoint}
)
def test_protocol_switch():
client = TestClient(mixed_protocol_app)
response = client.get("/")
assert response.status_code == 200
assert response.text == "Hello, world"
with client.wsconnect("/") as session:
assert session.receive_json() == {"hello": "world"}