lqs
d20566fe07
Some checks failed
Test / lint (push) Has been cancelled
Test / test (pydantic-v1, 3.10) (push) Has been cancelled
Test / test (pydantic-v1, 3.11) (push) Has been cancelled
Test / test (pydantic-v1, 3.12) (push) Has been cancelled
Test / test (pydantic-v1, 3.8) (push) Has been cancelled
Test / test (pydantic-v1, 3.9) (push) Has been cancelled
Test / test (pydantic-v2, 3.10) (push) Has been cancelled
Test / test (pydantic-v2, 3.11) (push) Has been cancelled
Test / test (pydantic-v2, 3.12) (push) Has been cancelled
Test / test (pydantic-v2, 3.8) (push) Has been cancelled
Test / test (pydantic-v2, 3.9) (push) Has been cancelled
Test / coverage-combine (push) Has been cancelled
Test / check (push) Has been cancelled
Issue Manager / issue-manager (push) Has been cancelled
Label Approved / label-approved (push) Has been cancelled
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
import json
|
|
from typing import List
|
|
|
|
from fastapi import APIRouter, Depends, FastAPI, WebSocket
|
|
from fastapi.testclient import TestClient
|
|
from typing_extensions import Annotated
|
|
|
|
|
|
def dependency_list() -> List[str]:
|
|
return []
|
|
|
|
|
|
DepList = Annotated[List[str], Depends(dependency_list)]
|
|
|
|
|
|
def create_dependency(name: str):
|
|
def fun(deps: DepList):
|
|
deps.append(name)
|
|
|
|
return Depends(fun)
|
|
|
|
|
|
router = APIRouter(dependencies=[create_dependency("router")])
|
|
prefix_router = APIRouter(dependencies=[create_dependency("prefix_router")])
|
|
app = FastAPI(dependencies=[create_dependency("app")])
|
|
|
|
|
|
@app.websocket("/", dependencies=[create_dependency("index")])
|
|
async def index(websocket: WebSocket, deps: DepList):
|
|
await websocket.accept()
|
|
await websocket.send_text(json.dumps(deps))
|
|
await websocket.close()
|
|
|
|
|
|
@router.websocket("/router", dependencies=[create_dependency("routerindex")])
|
|
async def routerindex(websocket: WebSocket, deps: DepList):
|
|
await websocket.accept()
|
|
await websocket.send_text(json.dumps(deps))
|
|
await websocket.close()
|
|
|
|
|
|
@prefix_router.websocket("/", dependencies=[create_dependency("routerprefixindex")])
|
|
async def routerprefixindex(websocket: WebSocket, deps: DepList):
|
|
await websocket.accept()
|
|
await websocket.send_text(json.dumps(deps))
|
|
await websocket.close()
|
|
|
|
|
|
app.include_router(router, dependencies=[create_dependency("router2")])
|
|
app.include_router(
|
|
prefix_router, prefix="/prefix", dependencies=[create_dependency("prefix_router2")]
|
|
)
|
|
|
|
|
|
def test_index():
|
|
client = TestClient(app)
|
|
with client.websocket_connect("/") as websocket:
|
|
data = json.loads(websocket.receive_text())
|
|
assert data == ["app", "index"]
|
|
|
|
|
|
def test_routerindex():
|
|
client = TestClient(app)
|
|
with client.websocket_connect("/router") as websocket:
|
|
data = json.loads(websocket.receive_text())
|
|
assert data == ["app", "router2", "router", "routerindex"]
|
|
|
|
|
|
def test_routerprefixindex():
|
|
client = TestClient(app)
|
|
with client.websocket_connect("/prefix/") as websocket:
|
|
data = json.loads(websocket.receive_text())
|
|
assert data == ["app", "prefix_router2", "prefix_router", "routerprefixindex"]
|