pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.9 KiB
59 lines
1.9 KiB
from fastapi import FastAPI, WebSocket, APIRouter
|
|
from fastapi.testclient import TestClient
|
|
|
|
app = FastAPI(ignore_trailing_slash=True)
|
|
router = APIRouter()
|
|
|
|
@app.get("/example")
|
|
async def example_endpoint():
|
|
return {"msg": "Example"}
|
|
|
|
@app.get("/example2/")
|
|
async def example_endpoint():
|
|
return {"msg": "Example 2"}
|
|
|
|
@app.websocket("/websocket")
|
|
async def websocket_endpoint(websocket: WebSocket):
|
|
await websocket.accept()
|
|
await websocket.send_text("Websocket")
|
|
await websocket.close()
|
|
|
|
@app.websocket("/websocket2/")
|
|
async def websocket_endpoint(websocket: WebSocket):
|
|
await websocket.accept()
|
|
await websocket.send_text("Websocket 2")
|
|
await websocket.close()
|
|
|
|
@router.get("/example")
|
|
def route_endpoint():
|
|
return {"msg": "Routing Example"}
|
|
|
|
@router.get("/example2/")
|
|
def route_endpoint():
|
|
return {"msg": "Routing Example 2"}
|
|
|
|
app.include_router(router, prefix="/router")
|
|
|
|
client = TestClient(app)
|
|
|
|
def test_ignoring_trailing_slash():
|
|
response = client.get("/example/", follow_redirects=False)
|
|
assert response.status_code == 200
|
|
assert response.json()["msg"] == "Example"
|
|
response = client.get("/example2", follow_redirects=False)
|
|
assert response.status_code == 200
|
|
assert response.json()["msg"] == "Example 2"
|
|
|
|
def test_ignoring_trailing_shlash_ws():
|
|
with client.websocket_connect("/websocket/") as websocket:
|
|
assert websocket.receive_text() == "Websocket"
|
|
with client.websocket_connect("/websocket2") as websocket:
|
|
assert websocket.receive_text() == "Websocket 2"
|
|
|
|
def test_ignoring_trailing_routing():
|
|
response = client.get("router/example/", follow_redirects=False)
|
|
assert response.status_code == 200
|
|
assert response.json()["msg"] == "Routing Example"
|
|
response = client.get("router/example2", follow_redirects=False)
|
|
assert response.status_code == 200
|
|
assert response.json()["msg"] == "Routing Example 2"
|