def complex_factory():
simple = simple_factory()
complex = ComplexModel()
complex.simple = simple
complex.complex_text = "ComplexText"
complex.complex_num = 2222
complex.complex_date = datetime(2010, 1, 1)
return complex
def test_to_parent_element_nested(self):
m = Message.produce(
namespace=None,
type_name='myMessage',
members={'p':Person}
)
m.resolve_namespace(m,"m")
p = Person()
p.name = 'steve-o'
p.age = 2
p.addresses = []
for i in range(0, 100):
a = Address()
a.street = '123 happy way'
a.zip = i
a.laditude = '45.22'
a.longitude = '444.234'
p.addresses.append(a)
m_inst = m(p=p)
element=etree.Element('test')
m.to_parent_element(m_inst,m.get_namespace(),element)
element=element[0]
self.assertEquals('{%s}myMessage' % m.get_namespace(), element.tag)
addresses = element[0].find('{%s}addresses' % Person.get_namespace())
self.assertEquals(100, len(addresses))
self.assertEquals('0', addresses[0].find('{%s}zip' % Address.get_namespace()).text)
def test_class_to_parent_element(self):
m = Message.produce(
namespace=None,
type_name='myMessage',
members={'p': Person}
)
m.resolve_namespace(m,"punk")
m_inst = m()
m_inst.p = Person()
m_inst.p.name = 'steve-o'
m_inst.p.age = 2
m_inst.p.addresses = []
element=etree.Element('test')
m.to_parent_element(m_inst,m.get_namespace(),element)
element=element[0]
self.assertEquals(element.tag, '{%s}myMessage' % m.get_namespace())
self.assertEquals(element[0].find('{%s}name' % Person.get_namespace()).text,
'steve-o')
self.assertEquals(element[0].find('{%s}age' % Person.get_namespace()).text, '2')
self.assertEquals(
len(element[0].find('{%s}addresses' % Person.get_namespace())), 0)
p1 = m.from_xml(element)[0]
self.assertEquals(p1.name, m_inst.p.name)
self.assertEquals(p1.age, m_inst.p.age)
self.assertEquals(p1.addresses, [])
def test_to_parent_element_nested(self):
m = Message.produce(namespace=None,
type_name='myMessage',
members={'p': Person})
m.resolve_namespace(m, "m")
p = Person()
p.name = 'steve-o'
p.age = 2
p.addresses = []
for i in range(0, 100):
a = Address()
a.street = '123 happy way'
a.zip = i
a.laditude = '45.22'
a.longitude = '444.234'
p.addresses.append(a)
m_inst = m(p=p)
element = etree.Element('test')
Soap11().to_parent_element(m, m_inst, m.get_namespace(), element)
element = element[0]
self.assertEquals('{%s}myMessage' % m.get_namespace(), element.tag)
addresses = element[0].find('{%s}addresses' % Person.get_namespace())
self.assertEquals(100, len(addresses))
self.assertEquals(
'0', addresses[0].find('{%s}zip' % Address.get_namespace()).text)
def test_class_to_parent_element(self):
m = Message.produce(namespace=None,
type_name='myMessage',
members={'p': Person})
m.resolve_namespace(m, "punk")
m_inst = m()
m_inst.p = Person()
m_inst.p.name = 'steve-o'
m_inst.p.age = 2
m_inst.p.addresses = []
element = etree.Element('test')
Soap11().to_parent_element(m, m_inst, m.get_namespace(), element)
element = element[0]
self.assertEquals(element.tag, '{%s}myMessage' % m.get_namespace())
self.assertEquals(
element[0].find('{%s}name' % Person.get_namespace()).text,
'steve-o')
self.assertEquals(
element[0].find('{%s}age' % Person.get_namespace()).text, '2')
self.assertEquals(
len(element[0].find('{%s}addresses' % Person.get_namespace())), 0)
p1 = Soap11().from_element(m, element)[0]
self.assertEquals(p1.name, m_inst.p.name)
self.assertEquals(p1.age, m_inst.p.age)
self.assertEquals(p1.addresses, [])
def _produce_output_message(f, func_name, kparams):
"""Generate an output message for "rpc"-style API methods.
This message is a wrapper to the declared return type.
"""
_returns = kparams.get('_returns')
_body_style = _validate_body_style(kparams)
_out_message_name = kparams.get('_out_message', '%s%s' %
(func_name, RESPONSE_SUFFIX))
out_params = TypeInfo()
if _returns and _body_style == 'wrapped':
if isinstance(_returns, (list, tuple)):
default_names = ['%s%s%d' % (func_name, RESULT_SUFFIX, i) for i in
range(len(_returns))]
_out_variable_names = kparams.get('_out_variable_names',
default_names)
assert (len(_returns) == len(_out_variable_names))
var_pair = zip(_out_variable_names, _returns)
out_params = TypeInfo(var_pair)
else:
_out_variable_name = kparams.get('_out_variable_name',
'%s%s' % (func_name, RESULT_SUFFIX))
out_params[_out_variable_name] = _returns
ns = DEFAULT_NS
if _out_message_name.startswith("{"):
ns = _out_message_name[1:].partition("}")[0]
if _body_style == 'wrapped':
message = ComplexModel.produce(type_name=_out_message_name,
namespace=ns,
members=out_params)
message.__namespace__ = ns # FIXME: is this necessary?
else:
message = ComplexModel.alias(_out_message_name, ns, _returns)
return message
def _produce_output_message(f, func_name, kparams):
"""Generate an output message for "rpc"-style API methods.
This message is a wrapper to the declared return type.
"""
_returns = kparams.get('_returns')
_body_style = _validate_body_style(kparams)
_out_message_name = kparams.get('_out_message', '%sResponse' % func_name)
out_params = TypeInfo()
if _returns and _body_style == 'wrapped':
if isinstance(_returns, (list, tuple)):
default_names = ['%sResult%d' % (func_name, i) for i in
range(len(_returns))]
_out_variable_names = kparams.get('_out_variable_names',
default_names)
assert (len(_returns) == len(_out_variable_names))
var_pair = zip(_out_variable_names, _returns)
out_params = TypeInfo(var_pair)
else:
_out_variable_name = kparams.get('_out_variable_name',
'%sResult' % func_name)
out_params[_out_variable_name] = _returns
ns = DEFAULT_NS
if _out_message_name.startswith("{"):
ns = _out_message_name[1:].partition("}")[0]
if _body_style == 'wrapped':
message = ComplexModel.produce(type_name=_out_message_name,
namespace=ns,
members=out_params)
message.__namespace__ = ns # FIXME: is this necessary?
else:
message = ComplexModel.alias(_out_message_name, ns, _returns)
return message
def _produce_output_message(f, func_name, params, kparams):
"""Generate an output message for "rpc"-style API methods.
This message is a wrapper to the declared return type.
"""
_returns = kparams.get('_returns')
_body_style = _validate_body_style(kparams)
_out_message = kparams.get('_out_message', '%sResponse' % func_name)
out_params = TypeInfo()
if _returns and _body_style == 'wrapped':
if isinstance(_returns, (list, tuple)):
default_names = ['%sResult%d' % (func_name, i) for i in
range(len(_returns))]
_out_variable_names = kparams.get('_out_variable_names',
default_names)
assert (len(_returns) == len(_out_variable_names))
var_pair = zip(_out_variable_names,_returns)
out_params = TypeInfo(var_pair)
else:
_out_variable_name = kparams.get('_out_variable_name',
'%sResult' % func_name)
out_params[_out_variable_name] = _returns
if _body_style == 'wrapped':
message = ComplexModel.produce(type_name=_out_message,
namespace=DEFAULT_NS,
members=out_params)
message.__namespace__ = DEFAULT_NS # FIXME: is this necessary?
else:
message = ComplexModel.alias(_out_message, DEFAULT_NS, _returns)
return message
def test_namespaces(self):
m = Message.produce(
namespace="some_namespace",
type_name='myMessage',
members={'s': String, 'i': Integer},
)
mi = m()
mi.s = 'a'
e = etree.Element('test')
m.to_parent_element(mi,m.get_namespace(),e)
e=e[0]
self.assertEquals(e.tag, '{some_namespace}myMessage')
def test_namespaces(self):
m = Message.produce(
namespace="some_namespace",
type_name='myMessage',
members={
's': String,
'i': Integer
},
)
mi = m()
mi.s = 'a'
e = etree.Element('test')
Soap11().to_parent_element(m, mi, m.get_namespace(), e)
e = e[0]
self.assertEquals(e.tag, '{some_namespace}myMessage')
def test_simple_message(self):
m = Message.produce(
namespace=None,
type_name='myMessage',
members={'s': String, 'i': Integer}
)
m.resolve_namespace(m,'test')
m_inst = m(s="a", i=43)
e = etree.Element('test')
m.to_parent_element(m_inst,m.get_namespace(),e)
e=e[0]
self.assertEquals(e.tag, '{%s}myMessage' % m.get_namespace())
self.assertEquals(e.find('{%s}s' % m.get_namespace()).text, 'a')
self.assertEquals(e.find('{%s}i' % m.get_namespace()).text, '43')
values = m.from_xml(e)
self.assertEquals('a', values.s)
self.assertEquals(43, values.i)
def test_simple_message(self):
m = Message.produce(namespace=None,
type_name='myMessage',
members={
's': String,
'i': Integer
})
m.resolve_namespace(m, 'test')
m_inst = m(s="a", i=43)
e = etree.Element('test')
Soap11().to_parent_element(m, m_inst, m.get_namespace(), e)
e = e[0]
self.assertEquals(e.tag, '{%s}myMessage' % m.get_namespace())
self.assertEquals(e.find('{%s}s' % m.get_namespace()).text, 'a')
self.assertEquals(e.find('{%s}i' % m.get_namespace()).text, '43')
values = Soap11().from_element(m, e)
self.assertEquals('a', values.s)
self.assertEquals(43, values.i)
try:
for i in range(len(param_names)):
e0 = _in_variable_names.get(param_names[i], param_names[i])
e1 = params[i]
in_params[e0] = e1
except IndexError, e:
raise Exception("The parameter numbers of the %r function and its "
"decorator mismatch." % f.func_name)
ns = DEFAULT_NS
if _in_message_name.startswith("{"):
ns = _in_message_name[1:].partition("}")[0]
message=ComplexModel.produce(type_name=_in_message_name, namespace=ns,
members=in_params)
message.__namespace__ = ns
return message
def _validate_body_style(kparams):
_body_style = kparams.get('_body_style')
_soap_body_style = kparams.get('_soap_body_style')
if _body_style is None:
_body_style = 'wrapped'
elif not (_body_style in ('wrapped', 'bare')):
raise ValueError("body_style must be one of ('wrapped', 'bare')")
elif _soap_body_style == 'document':
_body_style = 'bare'
elif _soap_body_style == 'rpc':
argcount = f.func_code.co_argcount
param_names = f.func_code.co_varnames[arg_start:argcount]
in_params = TypeInfo()
try:
for i in range(len(param_names)):
e0 = _in_variable_names.get(param_names[i], param_names[i])
e1 = params[i]
in_params[e0] = e1
except IndexError, e:
raise Exception("%s has parameter numbers mismatching" % f.func_name)
message=ComplexModel.produce(type_name=_in_message, namespace=DEFAULT_NS,
members=in_params)
message.__namespace__ = DEFAULT_NS
return message
def _validate_body_style(kparams):
_body_style = kparams.get('_body_style')
_soap_body_style = kparams.get('_soap_body_style')
if _body_style is None:
_body_style = 'wrapped'
elif not (_body_style in ('wrapped', 'bare')):
raise ValueError("body_style must be one of ('wrapped', 'bare')")
elif _soap_body_style == 'document':
_body_style = 'bare'
elif _soap_body_style == 'rpc':
