843c6a3d0f
--HG-- rename : wsme/restjson.py => wsme/protocols/restjson.py rename : wsme/restxml.py => wsme/protocols/restxml.py rename : wsme/soap.py => wsme/protocols/soap.py rename : wsme/templates/__init__.py => wsme/protocols/templates/__init__.py rename : wsme/templates/fault.html => wsme/protocols/templates/fault.html rename : wsme/templates/soap.html => wsme/protocols/templates/soap.html rename : wsme/templates/wsdl.html => wsme/protocols/templates/wsdl.html
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
import decimal
|
|
import base64
|
|
|
|
import wsme.tests.protocol
|
|
|
|
try:
|
|
import simplejson as json
|
|
except:
|
|
import json
|
|
|
|
import wsme.protocols.restjson
|
|
from wsme.utils import *
|
|
|
|
def prepare_value(value, datatype):
|
|
if datatype in (datetime.date, datetime.time, datetime.datetime):
|
|
return value.isoformat()
|
|
if datatype == decimal.Decimal:
|
|
return str(value)
|
|
if datatype == wsme.types.binary:
|
|
return base64.encodestring(value)
|
|
return value
|
|
|
|
def prepare_result(value, datatype):
|
|
if datatype == datetime.date:
|
|
return parse_isodate(value)
|
|
if datatype == datetime.time:
|
|
return parse_isotime(value)
|
|
if datatype == datetime.datetime:
|
|
return parse_isodatetime(value)
|
|
if hasattr(datatype, '_wsme_attributes'):
|
|
for name, attr in datatype._wsme_attributes:
|
|
if name not in value:
|
|
continue
|
|
value[name] = prepare_result(value[name], attr.datatype)
|
|
return value
|
|
if datatype == wsme.types.binary:
|
|
return base64.decodestring(value)
|
|
if type(value) != datatype:
|
|
return datatype(value)
|
|
return value
|
|
|
|
class TestRestJson(wsme.tests.protocol.ProtocolTestCase):
|
|
protocol = 'REST+Json'
|
|
|
|
def call(self, fpath, _rt=None, **kw):
|
|
for key in kw:
|
|
if isinstance(kw[key], tuple):
|
|
value, datatype = kw[key]
|
|
else:
|
|
value = kw[key]
|
|
datatype = type(value)
|
|
kw[key] = prepare_value(value, datatype)
|
|
content = json.dumps(kw)
|
|
res = self.app.post(
|
|
'/' + fpath,
|
|
content,
|
|
headers={
|
|
'Content-Type': 'application/json',
|
|
},
|
|
expect_errors=True)
|
|
print "Received:", res.body
|
|
r = json.loads(res.body)
|
|
if 'result' in r:
|
|
r = r['result']
|
|
if _rt and r:
|
|
r = prepare_result(r, _rt)
|
|
return r
|
|
else:
|
|
raise wsme.tests.protocol.CallException(
|
|
r['faultcode'],
|
|
r['faultstring'],
|
|
r.get('debuginfo'))
|
|
|
|
return json.loads(res.body)
|