def test_get_statistic__status_500(self, r_mock):
m_resp = mock.Mock(r.Response)
m_resp.status_code = 500
m_resp.url = 'http://url'
m_resp.reason = 'reason 500'
r_mock.return_value = m_resp
with self.assertRaises(ServiceException):
get_statistic(self.test_agency_cd, self.test_site_no, 'site-info',
self.test_service_root)
def test_get_statistic__status_404(self, r_mock):
m_resp = mock.Mock(r.Response)
m_resp.status_code = 404
m_resp.url = self.test_service_root
m_resp.reason = 'reason 404'
r_mock.return_value = m_resp
result = get_statistic(self.test_agency_cd, self.test_site_no,
'site-info', self.test_service_root)
self.assertEqual(False, result['is_fetched'])
self.assertEqual(False, result['is_ranked'])
def test_get_statistic__success(self, r_mock):
m_resp = mock.Mock(r.Response)
m_resp.text = '{"value":"SUCCESS"}'
m_resp.status_code = 200
m_resp.url = self.test_service_root
r_mock.return_value = m_resp
result = get_statistic(self.test_agency_cd, self.test_site_no,
'site-info', self.test_service_root)
self.assertEqual('SUCCESS', result['value'])
self.assertEqual(True, result['is_fetched'])
url = '/'.join([
self.test_service_root, 'ngwmn_cache/direct/json/site-info',
self.test_agency_cd, self.test_site_no
])
r_mock.assert_called_with(url)