halfapi/tests/test_lib_responses.py

import json
import decimal
import datetime

from halfapi.lib.responses import ORJSONResponse


def test_orjson():
    test_obj = {
        "ok": "ko",
        "dec": decimal.Decimal(42),
        "set": {0, 4, 2},
        "date": datetime.date(1,1,1),
        "datetime": datetime.datetime(1,1,1),
    }

    resp = ORJSONResponse(test_obj)
    body = resp.body.decode()
    test_obj_dec = json.loads(body)
    print(test_obj_dec)
    assert 'ok' in test_obj_dec.keys()
    assert isinstance(test_obj_dec['ok'], str)
    assert isinstance(test_obj_dec['dec'], str)
    assert isinstance(test_obj_dec['set'], list)
    assert isinstance(test_obj_dec['date'], str)
    assert test_obj_dec['date'] == '0001-01-01'
    assert test_obj_dec['datetime'] == '0001-01-01T00:00:00'