mirror of https://github.com/encode/starlette.git
34 lines
960 B
Python
34 lines
960 B
Python
|
from starlette.applications import Starlette
|
||
|
|
from starlette.middleware.sessions import SessionMiddleware
|
||
|
|
from starlette.responses import JSONResponse
|
||
|
|
from starlette.testclient import TestClient
|
||
|
|
|
||
|
|
app = Starlette()
|
||
|
|
app.add_middleware(SessionMiddleware, secret_key="example")
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/view_session")
|
||
|
|
def view_session(request):
|
||
|
|
return JSONResponse({"session": request.session})
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/update_session", methods=["POST"])
|
||
|
|
async def update_session(request):
|
||
|
|
data = await request.json()
|
||
|
|
request.session.update(data)
|
||
|
|
return JSONResponse({"session": request.session})
|
||
|
|
|
||
|
|
|
||
|
|
client = TestClient(app)
|
||
|
|
|
||
|
|
|
||
|
|
def test_session():
|
||
|
|
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"}}
|