def test_can_reject_invalid_root_tag(self):
     soap_message = ('<ns0:invalid xmlns:ns0="invalid"/>')
     request_message = self._wrap_with_soap_envelope(soap_message)
     request = SOAPRequest(dict(REQUEST_METHOD='POST'), request_message)
     dispatcher = SOAPDispatcher(echo_service())
     response = dispatcher.dispatch(request)
     self.assert_is_soap_fault(response, partial_fault_string="DocumentInvalid")

 def test_can_dispatch_soap_request_with_plain_wsgi(self):
     dispatcher = SOAPDispatcher(echo_service())
     app = WsgiSoapApplication(dispatcher)
     start_response = self._response_mock()
     soap_message = (
         b'<?xml version="1.0" encoding="utf-8"?>'
         b'<senv:Envelope xmlns:senv="http://schemas.xmlsoap.org/soap/envelope/" xmlns:tns="http://soap.example/echo/types">'
         b'<senv:Body>'
         b'<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
         b'<value>foobar</value>'
         b'</ns1:echoRequest>'
         b'</senv:Body>'
         b'</senv:Envelope>'
     )
     response = app(self._wsgi_env(soap_message), start_response)
     assert_equals('200 OK', start_response.code)
     assert_equals('text/xml', dict(start_response.headers)['Content-Type'])
     expected_xml = (
         b'<ns0:Envelope xmlns:ns0="http://schemas.xmlsoap.org/soap/envelope/">'
         b'<ns0:Body>'
         b'<ns0:echoResponse xmlns:ns0="http://soap.example/echo/types">'
         b'<value>foobar</value>'
         b'</ns0:echoResponse>'
         b'</ns0:Body>'
         b'</ns0:Envelope>'
     )
     assert_equals(expected_xml, b''.join(response))

 def test_can_reject_malformed_xml_soap_message(self):
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), 'garbage')
     dispatcher = SOAPDispatcher(echo_service())
     response = dispatcher.dispatch(request)
     assert_equals(500, response.http_status_code)
     assert_equals('text/xml', response.http_headers['Content-Type'])
     self.assert_is_soap_fault(response, partial_fault_string=u"Start tag expected, '<' not found")

 def test_can_include_imported_schemas_during_validation(self):
     # In case the SOAPDispatcher would not use imported schemas for
     # validation it would fail because the 'code' tag is only defined in
     # the imported schema
     handler, handler_state = echo_handler()
     service = echo_service(handler)
     class CodeType(xsd.String):
         pattern = r'[0-9]{5}'
     class Container(xsd.ComplexType):
         value = xsd.Element(CodeType)
     code_schema = xsd.Schema('http://soap.example/included',
         location='http://soap.example/included',
         elementFormDefault=xsd.ElementFormDefault.UNQUALIFIED,
         simpleTypes=[CodeType],
         complexTypes=[Container],
         elements={'foo': xsd.Element(Container)},
     )
     service.methods[0].input = 'foo'
     service.schema.imports = (code_schema, )
     # The setup is a bit simplistic because the <code> tag is not parsed
     # into a soapfish model element for the handler but this was enough
     # to trigger the bug
     dispatcher = SOAPDispatcher(service)
     wsgi_environ = dict(SOAPACTION='echo', REQUEST_METHOD='POST')
     soap_message = '<ns0:foo xmlns:ns0="http://soap.example/included"><value>12345</value></ns0:foo>'
     request = SOAPRequest(wsgi_environ, self._wrap_with_soap_envelope(soap_message))
     response = dispatcher.dispatch(request)
     self.assert_is_successful_response(response, handler_state)
     assert_equals('12345', handler_state.input_.value)

    def test_can_dispatch_good_soap_message(self):
        handler, handler_state = echo_handler()
        dispatcher = SOAPDispatcher(echo_service(handler))
        soap_message = ('<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
            '<value>foobar</value>'
        '</ns1:echoRequest>')
        request_message = self._wrap_with_soap_envelope(soap_message)
        request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)

        response = dispatcher.dispatch(request)
        self.assert_is_successful_response(response, handler_state)
        assert_equals('foobar', handler_state.input_.value)

        response_document = etree.fromstring(response.http_content)
        response_xml = etree.tostring(response_document, pretty_print=True)
        expected_xml = (
            b'<ns0:Envelope xmlns:ns0="http://schemas.xmlsoap.org/soap/envelope/">\n'
            b'  <ns0:Body>\n'
            b'    <ns0:echoResponse xmlns:ns0="http://soap.example/echo/types">\n'
            b'      <value>foobar</value>\n'
            b'    </ns0:echoResponse>\n'
            b'  </ns0:Body>\n'
            b'</ns0:Envelope>\n'
        )
        assert_equals(expected_xml, response_xml)

 def test_can_reject_invalid_action(self):
     soap_message = ('<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
         '<value>foobar</value>'
         '</ns1:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message)
     request = SOAPRequest(dict(SOAPACTION='invalid', REQUEST_METHOD='POST'), request_message)
     dispatcher = SOAPDispatcher(echo_service())
     response = dispatcher.dispatch(request)
     self.assert_is_soap_fault(response, partial_fault_string=u"Invalid soap action 'invalid'")

    def test_can_reject_non_soap_xml_body(self):
        request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), '<some>xml</some>')
        dispatcher = SOAPDispatcher(echo_service())

        # previously this raised an AttributeError due to an unhandled exception
        response = dispatcher.dispatch(request)
        assert_equals(500, response.http_status_code)
        assert_equals('text/xml', response.http_headers['Content-Type'])
        self.assert_is_soap_fault(response, partial_fault_string=u'Missing SOAP body')

 def test_can_dispatch_requests_based_on_soap_body(self):
     handler, handler_state = echo_handler()
     dispatcher = SOAPDispatcher(echo_service(handler))
     soap_message = ('<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
         '<value>foobar</value>'
     '</ns1:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message)
     request = SOAPRequest(dict(SOAPACTION='""', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     self.assert_is_successful_response(response, handler_state)

 def test_can_handle_empty_output_header(self):
     handler, handler_state = echo_handler()
     dispatcher = SOAPDispatcher(echo_service(handler, output_header=EchoOutputHeader))
     soap_message = ('<tns:echoRequest xmlns:tns="http://soap.example/echo/types">'
         '<value>foobar</value>'
     '</tns:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message)
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     self.assert_is_successful_response(response, handler_state)

 def test_evaluate_service_location(self):
     handler, _ = echo_handler()
     service = echo_service(handler)
     service.location = '{scheme}://{host}/ws'
     dispatcher = SOAPDispatcher(service)
     request = SOAPRequest(dict(REQUEST_METHOD='GET', QUERY_STRING='wsdl',
                                HTTP_HOST='soap.example'), '')
     response = dispatcher.dispatch(request)
     self.assert_is_successful_response(response)
     assert_not_contains('{scheme}', response.http_content.decode())
     assert_not_contains('{http}', response.http_content.decode())

 def test_can_validate_soap_header(self):
     handler, handler_state = echo_handler()
     dispatcher = SOAPDispatcher(echo_service(handler, input_header=EchoInputHeader))
     soap_header = ('<tns:invalid>42</tns:invalid>')
     soap_message = ('<tns:echoRequest>'
         '<value>foobar</value>'
     '</tns:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message, header=soap_header)
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     self.assert_is_soap_fault(response, partial_fault_string="DocumentInvalid")

 def test_can_validate_soap_message(self):
     handler, handler_state = echo_handler()
     dispatcher = SOAPDispatcher(echo_service(handler))
     soap_message = ('<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
         '<invalid>foobar</invalid>'
         '</ns1:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message)
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     assert_false(handler_state.was_called)
     self.assert_is_soap_fault(response,
         partial_fault_string=u"Element 'invalid': This element is not expected. Expected is ( value ).")

 def setUp(self):
     if django is None:
         raise SkipTest('django not installed')
     super(DjangoDispatchTest, self).setUp()
     self.service = echo_service()
     if django.conf.settings.__dict__['_wrapped'] is empty:
         django.conf.settings.configure(self.django_settings())
         django.conf.settings.update(AttrDict(
             ROOT_URLCONF=AttrDict(
                 urlpatterns = patterns('', (r"^ws/$", django_dispatcher(self.service))),
             ),
         ))
     self.client = Client()

 def test_can_propagate_custom_input_header(self):
     handler, handler_state = echo_handler()
     dispatcher = SOAPDispatcher(echo_service(handler, input_header=EchoInputHeader))
     soap_header = ('<tns:InputVersion>42</tns:InputVersion>')
     soap_message = ('<tns:echoRequest>'
         '<value>foobar</value>'
     '</tns:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message, header=soap_header)
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     self.assert_is_successful_response(response, handler_state)
     assert_not_none(handler_state.input_header)
     assert_equals('42', handler_state.input_header.InputVersion)

    def test_can_handle_xsd_element_as_return_value_from_handler(self):
        handler = lambda request, input_: input_
        dispatcher = SOAPDispatcher(echo_service(handler))
        soap_message = ('<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
            '<value>hello</value>'
        '</ns1:echoRequest>')
        request_message = self._wrap_with_soap_envelope(soap_message)
        request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)

        response = dispatcher.dispatch(request)
        body_text = response.http_content
        if not isinstance(body_text, basestring):
            body_text = body_text.decode('utf-8')
        assert_contains('<value>hello</value>', body_text)

 def test_can_propagate_custom_output_header(self):
     handler, handler_state = echo_handler()
     def _handler(request, _input):
         resp = handler(request, _input)
         resp.soap_header = EchoOutputHeader(OutputVersion='42')
         return resp
     dispatcher = SOAPDispatcher(echo_service(_handler, output_header=EchoOutputHeader))
     soap_header = ('<tns:InputVersion>42</tns:InputVersion>')
     soap_message = ('<tns:echoRequest xmlns:tns="http://soap.example/echo/types">'
         '<value>foobar</value>'
     '</tns:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message, header=soap_header)
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     self.assert_is_successful_response(response, handler_state)
     assert_contains(b'<ns0:OutputVersion>42</ns0:OutputVersion>', response.http_content)

    def test_can_use_soap_error_from_handler(self):
        soap_error = SOAPError('code', 'internal data error', 'actor')
        faulty_handler = lambda request, input_: SOAPResponse(soap_error)
        dispatcher = SOAPDispatcher(echo_service(handler=faulty_handler))
        soap_message = ('<ns1:echoRequest xmlns:ns1="http://soap.example/echo/types">'
            '<value>foobar</value>'
        '</ns1:echoRequest>')
        request_message = self._wrap_with_soap_envelope(soap_message)
        request = SOAPRequest(dict(REQUEST_METHOD='POST'), request_message)

        response = dispatcher.dispatch(request)
        assert_equals('text/xml', response.http_headers['Content-Type'])
        assert_equals(500, response.http_status_code)
        self.assert_is_soap_fault(response,
            fault_code='code',
            partial_fault_string=u'internal data error'
        )

 def test_return_soap_fault_on_exception(self):
     def handler(request, _input):
         raise Exception('unexpected exception')
     service = echo_service(handler)
     dispatcher = SOAPDispatcher(service, [ExceptionToSoapFault()])
     soap_message = ('<tns:echoRequest xmlns:tns="http://soap.example/echo/types">'
         '<value>foobar</value>'
     '</tns:echoRequest>')
     request_message = self._wrap_with_soap_envelope(soap_message)
     request = SOAPRequest(dict(SOAPACTION='echo', REQUEST_METHOD='POST'), request_message)
     response = dispatcher.dispatch(request)
     self.assert_is_soap_fault(response,
         fault_code=service.version.Code.SERVER,
         partial_fault_string=u'Internal Error',
     )
     assert_equals('text/xml', response.http_headers['Content-Type'])
     assert_equals(500, response.http_status_code)

    def test_hook_soap_response(self):
        message = (
            '<tns:echoRequest xmlns:tns="http://soap.example/echo/types">'
            '<value>Cast a hook to catch a soapfish.</value>'
            '</tns:echoRequest>'
        )
        request = SOAPRequest(
            {'REQUEST_METHOD': 'POST', 'SOAPACTION':'echo'},
            self._wrap_with_soap_envelope(message),
        )

        def hook(dispatcher, request, response):
            response.http_status_code = 999
            return response

        dispatcher = SOAPDispatcher(echo_service(), hooks={'soap-response': hook})
        response = dispatcher.dispatch(request)
        self.assertEqual(response.http_status_code, 999)

    def test_hook_soap_request(self):
        message = (
            '<tns:echoRequest xmlns:tns="http://soap.example/echo/types">'
            '<value>Cast a hook to catch a soapfish.</value>'
            '</tns:echoRequest>'
        )
        request = SOAPRequest(
            {'REQUEST_METHOD': 'POST', 'SOAPACTION':'echo'},
            self._wrap_with_soap_envelope(message),
        )

        def hook(dispatcher, request):
            request.http_content = request.http_content.replace(b'catch', b'snare')
            return request

        dispatcher = SOAPDispatcher(echo_service(), hooks={'soap-request': hook})
        response = dispatcher.dispatch(request)
        self.assertIn(b'Cast a hook to snare a soapfish.', response.http_content)