Browse Source

edit: bson dependency comment for mypy in responses.py & add tests for bson serialization

pull/13760/head
WrldEngine 1 month ago
parent
commit
308c762d6e
  1. 4
      fastapi/responses.py
  2. 21
      tests/test_bson_response_class.py

4
fastapi/responses.py

@ -21,9 +21,9 @@ except ImportError: # pragma: nocover
try:
import bson
import bson # type: ignore[import-untyped]
except ImportError: # pragma: nocover
bson = None # type: ignore
bson = None
class UJSONResponse(JSONResponse):

21
tests/test_bson_response_class.py

@ -0,0 +1,21 @@
import bson
from fastapi import FastAPI
from fastapi.responses import BSONResponse
from fastapi.testclient import TestClient
app = FastAPI(default_response_class=BSONResponse)
@app.get("/bson_keys")
def get_bson_serialized_data():
return {"key": "Hello World", 1: 1}
client = TestClient(app)
def test_bson_serialized_data():
with client:
response = client.get("/bson_keys")
assert response.content == bson.dumps({"key": "Hello World", 1: 1})
Loading…
Cancel
Save