def test_valid_lookup_v3(self):
from txtorcon import util
orig = util.city
try:
util.city = FakeGeoIP(version=3)
nl = util.NetLocation('127.0.0.1')
self.assertTrue(nl.city)
self.assertEqual(nl.city[0], 'City')
self.assertEqual(nl.city[1], 'Region')
finally:
util.ity = orig
def test_city_fails(self):
"make sure we don't fail if the city lookup excepts"
from txtorcon import util
orig = util.city
try:
class Thrower(object):
def record_by_addr(*args, **kw):
raise RuntimeError("testing failure")
util.city = Thrower()
nl = util.NetLocation('127.0.0.1')
self.assertEqual(None, nl.city)
finally:
util.city = orig
def test_no_city_db(self):
"ensure we lookup from country if we have no city"
from txtorcon import util
origcity = util.city
origcountry = util.country
try:
util.city = None
obj = object()
class CountryCoder(object):
def country_code_by_addr(self, ipaddr):
return obj
util.country = CountryCoder()
nl = util.NetLocation('127.0.0.1')
self.assertEqual(obj, nl.countrycode)
finally:
util.city = origcity
util.country = origcountry
def test_no_city_or_country_db(self):
"ensure we lookup from asn if we have no city or country"
from txtorcon import util
origcity = util.city
origcountry = util.country
origasn = util.asn
try:
util.city = None
util.country = None
class Thrower:
def org_by_addr(*args, **kw):
raise RuntimeError("testing failure")
util.asn = Thrower()
nl = util.NetLocation('127.0.0.1')
self.assertEqual('', nl.countrycode)
finally:
util.city = origcity
util.country = origcountry
util.asn = origasn
