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>')
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())
