|
|
|
@ -44,16 +44,6 @@ async def raw_dependency( |
|
|
|
yield "accepted" |
|
|
|
|
|
|
|
|
|
|
|
@app.get("/sse-default", response_class=EventSourceResponse) |
|
|
|
async def sse_default() -> AsyncIterable[dict[str, str]]: |
|
|
|
yield {"message": "ok"} |
|
|
|
|
|
|
|
|
|
|
|
@app.get("/jsonl-default") |
|
|
|
async def jsonl_default() -> AsyncIterable[dict[str, str]]: |
|
|
|
yield {"message": "ok"} |
|
|
|
|
|
|
|
|
|
|
|
@app.post("/sse-created-override", response_class=EventSourceResponse, status_code=201) |
|
|
|
async def sse_created_override( |
|
|
|
accepted: None = Depends(set_accepted), |
|
|
|
@ -128,20 +118,6 @@ def test_raw_stream_still_honors_dependency_status_code() -> None: |
|
|
|
assert response.text == "accepted" |
|
|
|
|
|
|
|
|
|
|
|
def test_sse_stream_default_status_code_stays_200() -> None: |
|
|
|
response = client.get("/sse-default") |
|
|
|
|
|
|
|
assert response.status_code == 200 |
|
|
|
assert get_sse_data(response.text) == [{"message": "ok"}] |
|
|
|
|
|
|
|
|
|
|
|
def test_jsonl_stream_default_status_code_stays_200() -> None: |
|
|
|
response = client.get("/jsonl-default") |
|
|
|
|
|
|
|
assert response.status_code == 200 |
|
|
|
assert get_jsonl_data(response.text) == [{"message": "ok"}] |
|
|
|
|
|
|
|
|
|
|
|
def test_sse_stream_dependency_overrides_declared_status_code() -> None: |
|
|
|
response = client.post("/sse-created-override") |
|
|
|
|
|
|
|
@ -168,8 +144,6 @@ def test_stream_status_codes_match_openapi() -> None: |
|
|
|
|
|
|
|
assert response_status_codes(schema, "/sse-created", "post") == ["201"] |
|
|
|
assert response_status_codes(schema, "/jsonl-created", "post") == ["201"] |
|
|
|
assert response_status_codes(schema, "/sse-default", "get") == ["200"] |
|
|
|
assert response_status_codes(schema, "/jsonl-default", "get") == ["200"] |
|
|
|
|
|
|
|
assert response_status_codes(schema, "/sse-created-override", "post") == ["201"] |
|
|
|
assert response_status_codes(schema, "/jsonl-created-override", "post") == ["201"] |
|
|
|
|