def test_custom_transport_adapter(self):
""" Check if the transport adapter is the one expected (especially if we set it) """
class TestHttpAdapter(HTTPAdapter):
def __init__(self, *args, **kwargs):
super(TestHttpAdapter, self).__init__(*args, **kwargs)
class FakeHttpAdapter(object):
def __init__(self, *args, **kwargs):
pass
eve = EVE()
self.assertTrue(isinstance(eve._session.get_adapter('http://'), HTTPAdapter))
self.assertTrue(isinstance(eve._session.get_adapter('https://'), HTTPAdapter))
self.assertFalse(isinstance(eve._session.get_adapter('http://'), TestHttpAdapter))
self.assertFalse(isinstance(eve._session.get_adapter('https://'), TestHttpAdapter))
eve = EVE(transport_adapter=TestHttpAdapter())
self.assertTrue(isinstance(eve._session.get_adapter('http://'), TestHttpAdapter))
self.assertTrue(isinstance(eve._session.get_adapter('https://'), TestHttpAdapter))
# check that the wrong httpadapter is not used
eve = EVE(transport_adapter=FakeHttpAdapter())
self.assertTrue(isinstance(eve._session.get_adapter('http://'), HTTPAdapter))
self.assertFalse(isinstance(eve._session.get_adapter('http://'), FakeHttpAdapter))
eve = EVE(transport_adapter='')
self.assertTrue(isinstance(eve._session.get_adapter('http://'), HTTPAdapter))
def test_headers(self):
# Check default header
@httmock.all_requests
def check_default_headers(url, request):
self.assertNotIn('PyCrest-Testing', request.headers)
with httmock.HTTMock(check_default_headers):
EVE()
# Check custom header
def check_custom_headers(url, request):
self.assertIn('PyCrest-Testing', request.headers)
with httmock.HTTMock(check_custom_headers):
EVE(additional_headers={'PyCrest-Testing': True})
def test_endpoint_testing(self):
api = EVE(testing=True)
self.assertEqual(api._endpoint, 'https://api-sisi.testeveonline.com/')
# imageserver. is given an 302 redirect to image. on testeveonline.com
# we might just as well keep using the old URL for now
self.assertEqual(api._image_server, 'https://image.testeveonline.com/')
self.assertEqual(api._oauth_endpoint,
'https://sisilogin.testeveonline.com/oauth')
def test_user_agent(self):
@httmock.all_requests
def default_user_agent(url, request):
user_agent = request.headers.get('User-Agent', None)
self.assertEqual(
user_agent, 'PyCrest/{0} +https://github.com/pycrest/PyCrest'
.format(pycrest.version))
with httmock.HTTMock(default_user_agent):
EVE()
@httmock.all_requests
def customer_user_agent(url, request):
user_agent = request.headers.get('User-Agent', None)
self.assertEqual(
user_agent,
'PyCrest-Testing/{0} +https://github.com/pycrest/PyCrest'
.format(pycrest.version))
with httmock.HTTMock(customer_user_agent):
EVE(user_agent='PyCrest-Testing/{0} +https://github.com/pycrest/P'
'yCrest'.format(pycrest.version))
def setUp(self):
self.api = EVE()
with httmock.HTTMock(*all_httmocks):
self.api()
def test_file_cache(self, mkdir_function, isdir_function):
file_cache = FileCache(path=TestFileCache.DIR)
eve = EVE(cache=file_cache)
self.assertEqual(file_cache.path, TestFileCache.DIR)
self.assertTrue(isinstance(eve.cache, FileCache))
def test_apicache(self):
eve = EVE(cache=DictCache())
self.assertTrue(isinstance(eve.cache, DictCache))
def test_implements_apiobject(self):
class CustomCache(object):
pass
with self.assertRaises(ValueError):
eve = EVE(cache=CustomCache)
def test_no_cache(self):
eve = EVE(cache=None)
self.assertTrue(isinstance(eve.cache, DummyCache))
def setUp(self):
self.api = EVE()
def setUp(self):
with httmock.HTTMock(*all_httmocks):
self.api = EVE()
with httmock.HTTMock(*all_httmocks):
self.authed = self.api.authorize(code='code')
def setUp(self):
self.api = EVE(
client_id=1,
redirect_uri='http://localhost:8000/complete/eveonline/')
def test_cache_dir(self, mkdir_function, isdir_function):
eve = EVE(cache_dir=TestFileCache.DIR)
self.assertEqual(eve.cache_dir, TestFileCache.DIR)
self.assertTrue(isinstance(eve.cache, FileCache))
def test_callable_cache(self):
class CustomCache(object):
pass
eve = EVE(cache=CustomCache)
self.assertTrue(isinstance(eve.cache, CustomCache))
