import json

import bsonjs
from fastapi import FastAPI
from fastapi.responses import BSONJSResponse
from fastapi.testclient import TestClient

app = FastAPI(default_response_class=BSONJSResponse)


@app.get("/bsonjs_keys")
def get_bsonjs_serialized_data():
    return {"key": "Hello World", 1: 1}


client = TestClient(app)


def test_bsonjs_serialized_data():
    with client:
        response = client.get("/bsonjs_keys")

    assert response.content == bsonjs.loads(json.dumps({"key": "Hello World", 1: 1}))