test_depend_err_handler() — fastapi Function Reference

Architecture documentation for the test_depend_err_handler() function in test_ws_router.py from the fastapi codebase.

Function python FastAPI Routing calls 1

Entity Profile

FastAPI→ Routing→ test_depend_err_handler() — fastapi Function Reference

Dependency Diagram

graph TD
  430ff898_4571_f930_fe5e_cb2dcad36981["test_depend_err_handler()"]
  b2e4c2fb_8042_93dc_9d79_2ae1a12773d6["test_ws_router.py"]
  430ff898_4571_f930_fe5e_cb2dcad36981 -->|defined in| b2e4c2fb_8042_93dc_9d79_2ae1a12773d6
  afb5e1a2_9bfc_75af_3be4_6639f67befbd["make_app()"]
  430ff898_4571_f930_fe5e_cb2dcad36981 -->|calls| afb5e1a2_9bfc_75af_3be4_6639f67befbd
  style 430ff898_4571_f930_fe5e_cb2dcad36981 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

tests/test_ws_router.py lines 257–271

def test_depend_err_handler():
    """
    Verify that it is possible to write custom WebSocket middleware to catch errors
    """

    async def custom_handler(websocket: WebSocket, exc: CustomError) -> None:
        await websocket.close(1002, "foo")

    myapp = make_app(exception_handlers={CustomError: custom_handler})
    client = TestClient(myapp)
    with pytest.raises(WebSocketDisconnect) as e:
        with client.websocket_connect("/custom_error/"):
            pass  # pragma: no cover
    assert e.value.code == 1002
    assert "foo" in e.value.reason