21 lines
920 B
Python
21 lines
920 B
Python
from popup_api.schema import Response
|
|
|
|
|
|
def test_Response():
|
|
|
|
resp1 = Response(code=200, message="success", data=[{"foo": 1}])
|
|
assert resp1.dict() == dict(code=200, message="success", data=[{"foo": 1}])
|
|
assert resp1.json() == '{"code": 200, "message": "success", "data": [{"foo": 1}]}'
|
|
|
|
resp2 = Response(code=200, message="success", data=[])
|
|
assert resp2.dict() == dict(code=200, message="success", data=[])
|
|
assert resp2.json() == '{"code": 200, "message": "success", "data": []}'
|
|
|
|
resp3 = Response(code=200, message="success", data="foo")
|
|
assert resp3.dict() == dict(code=200, message="success", data="foo")
|
|
assert resp3.json() == '{"code": 200, "message": "success", "data": "foo"}'
|
|
|
|
resp4 = Response(code=400, message="failed")
|
|
assert resp4.dict() == dict(code=400, message="failed", data=None)
|
|
assert resp4.json() == '{"code": 400, "message": "failed", "data": null}'
|