def test_build_envelope_class_with_style_rpc(
self,
mock_get_or_create_inner_class,
mock_map_binding_message_parts,
mock_map_port_type_message,
):
mock_get_or_create_inner_class.side_effect = mock_create_inner
mock_map_binding_message_parts.side_effect = mock_create_attr
mock_map_port_type_message.side_effect = mock_create_attr
name = "some_operation_bindings"
style = "rpc"
namespace = "xsdata"
definitions = Definitions(location="foo.wsdl", target_namespace="bar")
port_type_message = PortTypeMessage(message="some_operation")
binding_message = BindingMessage(extended=[
AnyElement(qname="body", attributes={"namespace": "bodyns"}),
AnyElement(qname="header"),
AnyElement(qname="header"),
])
binding_message.ns_map["foo"] = "bar"
result = DefinitionsMapper.build_envelope_class(
definitions, binding_message, port_type_message, name, style,
namespace)
expected = Class(
qname=build_qname("bar", name),
meta_name="Envelope",
tag=Tag.BINDING_MESSAGE,
module="foo",
ns_map={"foo": "bar"},
namespace="xsdata",
)
self.assertEqual(2, len(result.inner))
self.assertEqual(1, len(result.inner[0].attrs))
self.assertEqual(2, len(result.inner[1].attrs))
self.maxDiff = None
mock_map_port_type_message.assert_called_once_with(
port_type_message, "bodyns")
mock_map_binding_message_parts.assert_has_calls([
mock.call(
definitions,
port_type_message.message,
binding_message.extended[1],
result.inner[1].ns_map,
),
mock.call(
definitions,
port_type_message.message,
binding_message.extended[2],
result.inner[1].ns_map,
),
])
mock_get_or_create_inner_class.assert_has_calls([
mock.call(mock.ANY, "Body"),
mock.call(mock.ANY, "Header"),
mock.call(mock.ANY, "Header"),
])
result.inner.clear()
self.assertEqual(expected, result)
def test_build_envelope_class(
self,
mock_get_or_create_inner_class,
mock_map_binding_message_parts,
mock_map_port_type_message,
):
mock_get_or_create_inner_class.side_effect = mock_create_inner
mock_map_binding_message_parts.side_effect = mock_create_attr
name = "some_operation_bindings"
style = "document"
namespace = "xsdata"
definitions = Definitions(location="foo.wsdl", target_namespace="bar")
port_type_message = PortTypeMessage(message="some_operation")
binding_message = BindingMessage(
extended=[
AnyElement(qname="body"),
AnyElement(qname="header"),
AnyElement(qname="header"),
]
)
binding_message.ns_map["foo"] = "bar"
result = DefinitionsMapper.build_envelope_class(
definitions, binding_message, port_type_message, name, style, namespace
)
expected = Class(
qname=build_qname("bar", name),
meta_name="Envelope",
type=BindingMessage,
module="foo",
ns_map={"foo": "bar"},
namespace="xsdata",
)
self.assertEqual(2, len(result.inner))
self.assertEqual(1, len(result.inner[0].attrs))
self.assertEqual(2, len(result.inner[1].attrs))
self.assertEqual(0, mock_map_port_type_message.call_count)
mock_map_binding_message_parts.assert_has_calls(
[
mock.call(
definitions,
port_type_message.message,
binding_message.extended[0],
result.inner[0].ns_map,
),
mock.call(
definitions,
port_type_message.message,
binding_message.extended[1],
result.inner[1].ns_map,
),
mock.call(
definitions,
port_type_message.message,
binding_message.extended[2],
result.inner[1].ns_map,
),
]
)
mock_get_or_create_inner_class.assert_has_calls(
[
mock.call(mock.ANY, "Body"),
mock.call(mock.ANY, "Header"),
mock.call(mock.ANY, "Header"),
]
)
result.inner.clear()
self.assertEqual(expected, result)
