class StatusResponseTestCase(TestCase): def setUp(self): self.status_response = StatusResponse(url='http://mock.info', status=201, method='GET', response_type='STATUS') self.text_response = StatusResponse(url='http://mock.info', status=200, method='GET', response_type='TEXT', content='CONTENT EQUAL') self.url_response = StatusResponse(url='http://mock.info', status=200, method='GET', response_type='URL', content='http://other.info') self.regex_response = StatusResponse(url='http://mock.info', status=200, method='GET', response_type='REGEX', content='<quantidade>(.*)</quantidade>') @requests_mock.mock() def test_check_regex(self, mock): xml_test = '<?xml version="1.0" encoding="UTF-8" standalone="yes"?><qtdeFilaProcessar><quantidade>12345</quantidade></qtdeFilaProcessar>' mock.get('http://mock.info', status_code=200, text=xml_test) self.assertEqual(self.regex_response.check(), (True, '12345')) @requests_mock.mock() def test_check_url(self, mock): mock.get('http://mock.info', status_code=200, text='URL WITH CONTENT EQUAL') mock.get('http://other.info', status_code=200, text='URL WITH CONTENT EQUAL') self.assertEqual(self.url_response.check(), (True, 'URL WITH CONTENT EQUAL')) @requests_mock.mock() def test_check_text(self, mock): mock.get('http://mock.info', status_code=200, text='CONTENT EQUAL') self.assertEqual(self.text_response.check(), (True, 'CONTENT EQUAL')) @requests_mock.mock() def test_check_status(self, mock): mock.get('http://mock.info', status_code=201) self.assertEqual(self.status_response.check(), (True, 201)) def test_check_exception(self): self.status_response.method = 'PUT' self.assertRaises(Exception, self.status_response.check()) self.assertEqual((False, 500), self.status_response.check())