Browse Source

🎨 Auto format

pull/15937/head
pre-commit-ci-lite[bot] 2 weeks ago
committed by GitHub
parent
commit
7ab5435f46
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
  1. 15
      tests/test_stream_status_code.py

15
tests/test_stream_status_code.py

@ -54,8 +54,6 @@ async def jsonl_default() -> AsyncIterable[dict[str, str]]:
yield {"message": "ok"} yield {"message": "ok"}
@app.post("/sse-created-override", response_class=EventSourceResponse, status_code=201) @app.post("/sse-created-override", response_class=EventSourceResponse, status_code=201)
async def sse_created_override( async def sse_created_override(
accepted: None = Depends(set_accepted), accepted: None = Depends(set_accepted),
@ -144,25 +142,23 @@ def test_jsonl_stream_default_status_code_stays_200() -> None:
assert get_jsonl_data(response.text) == [{"message": "ok"}] assert get_jsonl_data(response.text) == [{"message": "ok"}]
def test_sse_stream_dependency_overrides_declared_status_code() -> None: def test_sse_stream_dependency_overrides_declared_status_code() -> None:
response = client.post("/sse-created-override") response = client.post("/sse-created-override")
assert response.status_code == 202 assert response.status_code == 202
assert get_sse_data(response.text) == [{"message": "overridden"}] assert get_sse_data(response.text) == [{"message": "overridden"}]
def test_jsonl_stream_dependency_overrides_declared_status_code() -> None: def test_jsonl_stream_dependency_overrides_declared_status_code() -> None:
response = client.post("/jsonl-created-override") response = client.post("/jsonl-created-override")
assert response.status_code == 202 assert response.status_code == 202
assert get_jsonl_data(response.text) == [{"message": "overridden"}] assert get_jsonl_data(response.text) == [{"message": "overridden"}]
def test_raw_stream_dependency_overrides_declared_status_code() -> None: def test_raw_stream_dependency_overrides_declared_status_code() -> None:
response = client.post("/raw-created-override") response = client.post("/raw-created-override")
assert response.status_code == 202 assert response.status_code == 202
assert response.text == "overridden" assert response.text == "overridden"
@ -174,8 +170,7 @@ def test_stream_status_codes_match_openapi() -> None:
assert response_status_codes(schema, "/jsonl-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, "/sse-default", "get") == ["200"]
assert response_status_codes(schema, "/jsonl-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, "/sse-created-override", "post") == ["201"]
assert response_status_codes(schema, "/jsonl-created-override", "post") == ["201"] assert response_status_codes(schema, "/jsonl-created-override", "post") == ["201"]
assert response_status_codes(schema, "/raw-created-override", "post") == ["201"] assert response_status_codes(schema, "/raw-created-override", "post") == ["201"]
@ -186,4 +181,4 @@ def response_status_codes(schema: dict[str, Any], path: str, method: str) -> lis
route = cast(dict[str, Any], paths[path]) route = cast(dict[str, Any], paths[path])
operation = cast(dict[str, Any], route[method]) operation = cast(dict[str, Any], route[method])
responses = cast(dict[str, Any], operation["responses"]) responses = cast(dict[str, Any], operation["responses"])
return sorted(responses) return sorted(responses)

Loading…
Cancel
Save