def setUp(self):
self.geoip = GeoIP()
self.geoip.enabled = True
self.mock_maxminddb = mock.patch.object(self.geoip, '_db').start()
self.mock_download_db = mock.patch(__name__ + '.GeoIP._download_db',
asynctest.CoroutineMock()).start()
self.mock_load = mock.patch(__name__ + '.GeoIP.load',
asynctest.CoroutineMock()).start()
def setUp(self):
self.geoip = GeoIP()
self.geoip.enabled = True
self.mock_open_database = mock.patch('maxminddb.open_database').start()
self.mock_download_db = mock.patch(__name__ + '.GeoIP._download_db',
asynctest.CoroutineMock()).start()
class TestGeoIP_prune_lookup_cache(asynctest.TestCase):
def setUp(self):
self.geoip = GeoIP()
self.mock_time = mock.patch('time.time').start()
def tearDown(self):
mock.patch.stopall()
def test_lookup_cache_is_pruned_correctly(self):
import random
self.geoip._lookup_cache = {
'1.2.3.%d' % i: ('AB', i)
for i in random.sample(range(10), 10)
}
self.geoip.max_cache_size = 6
self.mock_time.return_value = 10
self.geoip._prune_lookup_cache()
self.assertEqual(self.geoip._lookup_cache, {
'1.2.3.7': ('AB', 7),
'1.2.3.8': ('AB', 8),
'1.2.3.9': ('AB', 9)
})
def setUp(self):
self.geoip = GeoIP()
self.mock_response = asynctest.MagicMock()
self.mock_response.content.read = asynctest.CoroutineMock()
self.mock_get = asynctest.patch('aiohttp.ClientSession.get').start()
self.mock_get.return_value.__aenter__.return_value = self.mock_response
self.mock_exists = mock.patch('os.path.exists').start()
self.mock_getmtime = mock.patch('os.path.getmtime').start()
self.mock_time = mock.patch('time.time').start()
self.mock_maybe_replace_db = mock.patch(
__name__ + '.GeoIP._maybe_replace_db').start()
class Test_country_code(asynctest.TestCase):
def setUp(self):
self.geoip = GeoIP()
self.geoip.enabled = True
self.mock_maxminddb = mock.patch.object(self.geoip, '_db').start()
self.mock_download_db = mock.patch(__name__ + '.GeoIP._download_db',
asynctest.CoroutineMock()).start()
self.mock_load = mock.patch(__name__ + '.GeoIP.load',
asynctest.CoroutineMock()).start()
def tearDown(self):
mock.patch.stopall()
def test_not_enabled(self):
self.geoip.enabled = False
self.assertIs(self.geoip.country_code('asdf'), None)
def test_cache(self):
self.geoip._lookup_cache = {'1.2.3.4': ('AB', time.time())}
self.assertEqual(self.geoip.country_code('1.2.3.4'), 'AB')
self.mock_maxminddb.get.assert_not_called()
self.mock_load.assert_not_called()
def test_maxminddb_raises_InvalidDatabaseError(self):
self.mock_maxminddb.get.side_effect = maxminddb.InvalidDatabaseError()
self.assertEqual(self.geoip.country_code('1.2.3.4'), None)
self.mock_maxminddb.get.assert_called_once_with('1.2.3.4')
self.mock_load.assert_called_once_with(force_update=True,
ignore_errors=True)
def test_maxminddb_raises_UnicodeDecodeError(self):
self.mock_maxminddb.get.side_effect = UnicodeDecodeError(
'mockcodec', b'', 0, 0, 'Oh no!')
self.assertEqual(self.geoip.country_code('1.2.3.4'), None)
self.mock_maxminddb.get.assert_called_once_with('1.2.3.4')
self.mock_load.assert_called_once_with(force_update=True,
ignore_errors=True)
def test_maxminddb_returns_non_dictinary(self):
self.mock_maxminddb.get.return_value = 'not a dict'
self.assertEqual(self.geoip.country_code('1.2.3.4'), None)
self.mock_maxminddb.get.assert_called_once_with('1.2.3.4')
self.mock_load.assert_called_once_with(force_update=True,
ignore_errors=True)
def test_country_key_maps_to_non_dictionary(self):
self.mock_maxminddb.get.return_value = {'country': 'not a dict'}
self.assertEqual(self.geoip.country_code('1.2.3.4'), None)
self.mock_maxminddb.get.assert_called_once_with('1.2.3.4')
self.mock_load.assert_called_once_with(force_update=True,
ignore_errors=True)
class TestGeoIPBase(asynctest.TestCase):
def setUp(self):
class GeoIP(GeoIPBase):
pass
self.geoip = GeoIP()
def test_cachedir(self):
self.assertEqual(self.geoip.cachedir, '')
self.geoip.cachedir = '/foo/bar/baz/'
self.assertEqual(self.geoip.cachedir, '/foo/bar/baz/')
def test_enabled(self):
self.assertEqual(self.geoip.enabled, False)
self.geoip.enabled = 'yes'
self.assertEqual(self.geoip.enabled, True)
async def test_load(self):
self.assertEqual(await self.geoip.load(), None)
def test_country_code(self):
self.assertEqual(self.geoip.country_code('1.2.3.4'), None)
def setUp(self):
self.geoip = GeoIP()
def setUp(self):
self.geoip = GeoIP()
self.mock_time = mock.patch('time.time').start()
def setUp(self):
class GeoIP(GeoIPBase):
pass
self.geoip = GeoIP()