Skip to content

Commit 893b2ef

Browse files
committed
Fix XML test
1 parent 386e28e commit 893b2ef

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

test/postmon_test.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -183,12 +183,12 @@ def get_cep(self, cep):
183183
return response
184184

185185
def test_xml_return(self):
186-
expected = ('<?xml version="1.0" encoding="utf-8"?>\n'
187-
'<result><bairro>Parque S\xc3\xa3o George</bairro>'
188-
'<cidade>Cotia</cidade>'
189-
'<cep>06708070</cep>'
190-
'<logradouro>Avenida Eid Mansur</logradouro>'
191-
'<estado>SP</estado></result>')
192-
186+
import xmltodict
193187
response = self.get_cep('06708070')
194-
self.assertEqual(response.body.strip(), expected.strip())
188+
parsed = xmltodict.parse(response.body)
189+
result = parsed.get('result')
190+
self.assertEqual(result['bairro'], u'Parque S\xe3o George')
191+
self.assertEqual(result['cidade'], u'Cotia')
192+
self.assertEqual(result['cep'], u'06708070')
193+
self.assertEqual(result['estado'], u'SP')
194+
self.assertEqual(result['logradouro'], u'Avenida Eid Mansur')

0 commit comments

Comments
 (0)