Commit 0b964257 by William Tisäter

Improve test coverage

parent 9bec82b1
......@@ -7,6 +7,18 @@ from tests.config import REGION_DB_PATH
class TestGeoIPRegionFunctions(unittest.TestCase):
def setUp(self):
self.se_hostname = 'dn.se'
self.se_region_data = {
'region_code': None,
'country_code': None
}
self.ca_hostname = 'cbc.ca'
self.ca_region_data = {
'region_code': 'ON',
'country_code': 'CA'
}
self.us_code = 'US'
self.us_hostname = 'apple.com'
self.us_ip = '17.172.224.47'
......@@ -32,3 +44,11 @@ class TestGeoIPRegionFunctions(unittest.TestCase):
def testCountryCodeByAddr(self):
us_code = self.gir.country_code_by_addr(self.us_ip)
self.assertEqual(us_code, self.us_code)
def testCanadaRegion(self):
region_name = self.gir.region_by_name(self.ca_hostname)
self.assertEqual(region_name, self.ca_region_data)
def testSwedenRegion(self):
region_name = self.gir.region_by_name(self.se_hostname)
self.assertEqual(region_name, self.se_region_data)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment