masq
/
halfapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[ORJSONResponse] Handle sets.

This commit is contained in:
Joël Maïzi 2021-01-25 10:44:07 +01:00
parent 54e215b6ff
commit f36a2d8e06
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
halfapi/lib/responses.py
View File

@ -62,7 +62,7 @@ class ORJSONResponse(JSONResponse):
decimal.Decimal decimal.Decimal
} }
list_types = { list_types = {
type(set) set
} }
if type(x) in str_types: if type(x) in str_types:
@ -70,7 +70,7 @@ class ORJSONResponse(JSONResponse):
if type(x) in list_types: if type(x) in list_types:
return list(x) return list(x)
raise TypeError(f'Type {type(x)} is not handled in ORJSONResponse') raise TypeError(f'Type {type(x)} is not handled by ORJSONResponse')
class HJSONResponse(ORJSONResponse): class HJSONResponse(ORJSONResponse):

4
tests/test_lib_responses.py
View File

@ -8,7 +8,7 @@ def test_orjson():
test_obj = { test_obj = {
"ok": "ko", "ok": "ko",
"dec": decimal.Decimal(42), "dec": decimal.Decimal(42),
"set": set([0,4,2]) "set": {0, 4, 2}
} }
resp = ORJSONResponse(test_obj) resp = ORJSONResponse(test_obj)
@ -18,4 +18,4 @@ def test_orjson():
assert 'ok' in test_obj_dec.keys() assert 'ok' in test_obj_dec.keys()
assert isinstance(test_obj_dec['ok'], str) assert isinstance(test_obj_dec['ok'], str)
assert isinstance(test_obj_dec['dec'], str) assert isinstance(test_obj_dec['dec'], str)
assert isinstance(test_obj_dec['set'], liststr) assert isinstance(test_obj_dec['set'], list)