def _(req):
msubtype = req.query.t.values().item()
return rsp().json(message=(req.body.a
.files()
.mimetype('text', (msubtype, ))
.values()
.item()))
def test_basics():
class StartResponse(object):
def __call__(self, status, headerlist):
self.status = status
self.headerlist = headerlist
start_response = StartResponse()
response = rsp().header('test', 1).code(404).body('abc')
assert response(start_response) == [compat.non_unicode_str('abc')]
assert start_response.status == '404 Not Found'
assert start_response.headerlist == [('test', 1)]
def test_json():
response = rsp().json(datetime=datetime(1990, 1, 1, 1, 1, 1),
date=date(1990, 12, 31),
bytes=b'abcde',
set=set([1, 2, 3]))
data = json.loads(response._body)
assert response._headers['Content-Type'] == ['application/json']
assert data['datetime'] == '1990/01/01 01:01:01'
assert data['date'] == '1990/12/31'
assert data['bytes'] == 'abcde'
assert data['set'] == [1, 2, 3]
def _(req):
d = {k + '-many': req.body.many(k) for k in req.body}
d.update({k + '-one': req.body.one(k) for k in req.body})
return rsp().json(**d)
def _(req):
'a' in req.query
d = {k + '-many': req.query.many(k) for k in req.query}
d.update({k + '-one': req.query.one(k) for k in req.query})
return rsp().json(**d)
def _(e):
traceback.print_exc()
return rsp().code(500)
def _(req):
return rsp().json(**req.params)
def test_location():
response = rsp().location('abc')
assert response._code == 301
assert response._headers == {'Location': ['abc']}
def _(req):
return rsp().clear_cookie('cookie')
def _(req):
return rsp().json(cookie=req.cookie)
def _(req):
return rsp()
def _(req):
return rsp().json(**{'f1': req.data('f1')[0],
'f2': req.data('f2')[0]})
def _(req):
return rsp().json(**req.extra)
def _(req):
req.data
req.body
return rsp()
def _(req):
return rsp().json(**req.data)
def _(req):
d = {k: req.body[k].files().items() for k in req.body}
d = {k: [e.filename for e in v] for k, v in d.items()}
return rsp().json(**d)
def test_error():
response = rsp()
pytest.raises(rsp, response.error)
def _(req):
return rsp().cookie('cookie', 100)
def test_html():
response = rsp().html('abc')
assert response._headers['Content-Type'] == ['text/html']
assert response._body == 'abc'
def _(req):
d = {k: req.body[k].values().of_int().items() for k in req.body}
return rsp().json(**d)
def test_json_error():
pytest.raises(TypeError, rsp().json, key=object())
def _(req):
return rsp().json(message=(req.body.a
.files()
.mimetype('text')
.values()
.item()))
def rsp(self):
return rsp()
def _(e):
return rsp().code(401).json(message='unauthorized')
