def test_default_usename(self):
username_test = PhishNetAPI(api_key='foo')
with requests_mock.mock() as m:
m._adapter.register_uri('GET', 'https://api.phish.net/api.json', [
{'text': '{"success": "1"}', 'status_code': 200},
{'text': '{"success": "1", "authkey": "232342342342"}', 'status_code': 200}
])
username_test.authorize('wilson')
self.assertEqual(username_test._default_username('suzy'), 'suzy')
self.assertEqual(username_test._default_username(None), 'wilson')
username_test = PhishNetAPI(api_key='foo')
self.assertEqual(username_test._default_username('suzy'), 'suzy')
self.assertRaises(TypeError, username_test._default_username, None)
def test_default_usename(self):
username_test = PhishNetAPI(api_key='foo')
with requests_mock.mock() as m:
m._adapter.register_uri(
'GET', 'https://api.phish.net/api.json', [{
'text': '{"success": "1"}',
'status_code': 200
}, {
'text': '{"success": "1", "authkey": "232342342342"}',
'status_code': 200
}])
username_test.authorize('wilson')
self.assertEqual(username_test._default_username('suzy'), 'suzy')
self.assertEqual(username_test._default_username(None), 'wilson')
username_test = PhishNetAPI(api_key='foo')
self.assertEqual(username_test._default_username('suzy'), 'suzy')
self.assertRaises(TypeError, username_test._default_username, None)
