class TestGeolocationTaskTestCase(unittest.TestCase):
"""Test GeolocationMixin.get_country_xxx() methods."""
def setUp(self):
self.task = GeolocationMixin()
self.task.geoip = FakeGeoLocation()
def test_country_name(self):
name = self.task.get_country_name(FakeGeoLocation.ip_address_1)
self.assertEquals(name, FakeGeoLocation.country_name_1)
def test_country_code(self):
code = self.task.get_country_code(FakeGeoLocation.ip_address_1)
self.assertEquals(code, FakeGeoLocation.country_code_1)
def test_country_name_exception(self):
self.task.geoip.country_name_by_addr = Mock(side_effect=Exception)
name = self.task.get_country_name(FakeGeoLocation.ip_address_1)
self.assertEquals(name, UNKNOWN_COUNTRY)
def test_country_code_exception(self):
self.task.geoip.country_code_by_addr = Mock(side_effect=Exception)
code = self.task.get_country_code(FakeGeoLocation.ip_address_1)
self.assertEquals(code, UNKNOWN_CODE)
def test_missing_country_name(self):
self.task.geoip.country_name_by_addr = Mock(return_value=None)
name = self.task.get_country_name(FakeGeoLocation.ip_address_1)
self.assertEquals(name, UNKNOWN_COUNTRY)
def test_missing_country_code(self):
self.task.geoip.country_code_by_addr = Mock(return_value=None)
code = self.task.get_country_code(FakeGeoLocation.ip_address_1)
self.assertEquals(code, UNKNOWN_CODE)
def test_empty_country_name(self):
self.task.geoip.country_name_by_addr = Mock(return_value=" ")
name = self.task.get_country_name(FakeGeoLocation.ip_address_1)
self.assertEquals(name, UNKNOWN_COUNTRY)
def test_empty_country_code(self):
self.task.geoip.country_code_by_addr = Mock(return_value=" ")
code = self.task.get_country_code(FakeGeoLocation.ip_address_1)
self.assertEquals(code, UNKNOWN_CODE)
def setUp(self):
self.task = GeolocationMixin()
self.task.geoip = FakeGeoLocation()