def test_query(self):
query_test = PhishNetAPI(api_key='foo')
self.assertRaises(NotImplementedError, query_test._query, 'PUT', 'foo')
with requests_mock.mock() as m:
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{
'text': '{"success": "0"}',
'status_code': 500
},
])
self.assertRaises(PhishNetAPIError, query_test._query, 'GET',
'foo', None, 0)
with requests_mock.mock() as m:
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{
'text': '{"success": "0"}',
'status_code': 500
},
])
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{
'text': '{"success": "0"}',
'status_code': 500
},
])
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{
'text': '{"success": "0"}',
'status_code': 200
},
])
self.assertEqual(
200,
query_test._query('GET', 'foo', data=None,
retry=3).status_code)
def test_query(self):
query_test = PhishNetAPI(api_key='foo')
self.assertRaises(NotImplementedError, query_test._query, 'PUT', 'foo')
with requests_mock.mock() as m:
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{'text': '{"success": "0"}', 'status_code': 500},
])
self.assertRaises(PhishNetAPIError, query_test._query, 'GET', 'foo', None, 0)
with requests_mock.mock() as m:
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{'text': '{"success": "0"}', 'status_code': 500},
])
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{'text': '{"success": "0"}', 'status_code': 500},
])
m._adapter.register_uri('GET', 'https://api.phish.net/foo', [
{'text': '{"success": "0"}', 'status_code': 200},
])
self.assertEqual(200, query_test._query('GET', 'foo', data=None, retry=3).status_code)
