def test_XmlResponse_class_zero_length_body_strip(self):
self._mock_response.read.return_value = ' '
response = XmlResponse(response=self._mock_response,
connection=self._mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed, '')
def test_XmlResponse_class(self):
self._mock_response.read.return_value = '<foo>bar</foo>'
response = XmlResponse(response=self._mock_response,
connection=self._mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed.tag, 'foo')
self.assertEqual(parsed.text, 'bar')
def test_XmlResponse_class_zero_length_body_strip(self):
with requests_mock.mock() as m:
m.register_uri("GET", "mock://test.com/", text=" ")
response_obj = requests.get("mock://test.com/")
response = XmlResponse(response=response_obj,
connection=self.mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed, "")
def test_XmlResponse_class_zero_length_body_strip(self):
with requests_mock.mock() as m:
m.register_uri('GET', 'mock://test.com/', text=' ')
response_obj = requests.get('mock://test.com/')
response = XmlResponse(response=response_obj,
connection=self.mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed, '')
def test_XmlResponse_class(self):
with requests_mock.mock() as m:
m.register_uri("GET", "mock://test.com/2", text="<foo>bar</foo>")
response_obj = requests.get("mock://test.com/2")
response = XmlResponse(response=response_obj,
connection=self.mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed.tag, "foo")
self.assertEqual(parsed.text, "bar")
def test_XmlResponse_class(self):
with requests_mock.mock() as m:
m.register_uri('GET', 'mock://test.com/2', text='<foo>bar</foo>')
response_obj = requests.get('mock://test.com/2')
response = XmlResponse(response=response_obj,
connection=self.mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed.tag, 'foo')
self.assertEqual(parsed.text, 'bar')
def test_XmlResponse_class(self):
with requests_mock.mock() as m:
m.register_uri('GET', 'mock://test.com/2', text='<foo>bar</foo>')
response_obj = requests.get('mock://test.com/2')
response = XmlResponse(response=response_obj,
connection=self.mock_connection)
parsed = response.parse_body()
self.assertEqual(parsed.tag, 'foo')
self.assertEqual(parsed.text, 'bar')
def test_lxml_from_unicode_with_encoding_declaration(self):
if etree is None:
return
connection = mock.Mock()
response_text = u'<?xml version="1.0" encoding="UTF-8"?>\n<Response>' \
'<Errors><Error><Code>InvalidSnapshot.NotFound</Code><Message>' \
'Snapshot does not exist</Message></Error></Errors><RequestID>' \
'd65ede94-44d9-40c4-992b-23aafd611797</RequestID></Response>'
with requests_mock.Mocker() as m:
m.get('https://127.0.0.1',
text=response_text,
headers={'content-type': 'application/xml'})
response = XmlResponse(requests.get('https://127.0.0.1'),
connection)
body = response.parse_body()
self.assertIsNotNone(body)
self.assertIsInstance(body, etree._Element)
