mirror of https://github.com/encode/starlette.git
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
from starlette.applications import Starlette
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
from starlette.responses import JSONResponse
|
|
from starlette.testclient import TestClient
|
|
|
|
|
|
def view_session(request):
|
|
return JSONResponse({"session": request.session})
|
|
|
|
|
|
async def update_session(request):
|
|
data = await request.json()
|
|
request.session.update(data)
|
|
return JSONResponse({"session": request.session})
|
|
|
|
|
|
async def clear_session(request):
|
|
request.session.clear()
|
|
return JSONResponse({"session": request.session})
|
|
|
|
|
|
def create_app():
|
|
app = Starlette()
|
|
app.add_route("/view_session", view_session)
|
|
app.add_route("/update_session", update_session, methods=["POST"])
|
|
app.add_route("/clear_session", clear_session, methods=["POST"])
|
|
return app
|
|
|
|
|
|
def test_session():
|
|
app = create_app()
|
|
app.add_middleware(SessionMiddleware, secret_key="example")
|
|
client = TestClient(app)
|
|
|
|
response = client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
response = client.post("/update_session", json={"some": "data"})
|
|
assert response.json() == {"session": {"some": "data"}}
|
|
|
|
response = client.get("/view_session")
|
|
assert response.json() == {"session": {"some": "data"}}
|
|
|
|
response = client.post("/clear_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
response = client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
|
|
def test_session_expires():
|
|
app = create_app()
|
|
app.add_middleware(SessionMiddleware, secret_key="example", max_age=-1)
|
|
client = TestClient(app)
|
|
|
|
response = client.post("/update_session", json={"some": "data"})
|
|
assert response.json() == {"session": {"some": "data"}}
|
|
|
|
response = client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
|
|
def test_secure_session():
|
|
app = create_app()
|
|
app.add_middleware(SessionMiddleware, secret_key="example", https_only=True)
|
|
secure_client = TestClient(app, base_url="https://testserver")
|
|
unsecure_client = TestClient(app, base_url="http://testserver")
|
|
|
|
response = unsecure_client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
response = unsecure_client.post("/update_session", json={"some": "data"})
|
|
assert response.json() == {"session": {"some": "data"}}
|
|
|
|
response = unsecure_client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
response = secure_client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
response = secure_client.post("/update_session", json={"some": "data"})
|
|
assert response.json() == {"session": {"some": "data"}}
|
|
|
|
response = secure_client.get("/view_session")
|
|
assert response.json() == {"session": {"some": "data"}}
|
|
|
|
response = secure_client.post("/clear_session")
|
|
assert response.json() == {"session": {}}
|
|
|
|
response = secure_client.get("/view_session")
|
|
assert response.json() == {"session": {}}
|