def test_fall_back_to_default_if_country_not_supported(self):
self.request.session = { geo.COUNTRY_SESSION_KEY: 'AE' }
self.assertEqual(geo.get_country_from_request(self.request), 'US')
def test_use_default_if_no_sources_can_detect(self):
self.assertEqual(geo.get_country_from_request(self.request), 'US')
def test_use_http_header_if_other_sources(self):
self.set_country_in_http_header()
self.assertEqual(geo.get_country_from_request(self.request), 'GB')
def test_use_ip_as_tertiary_source(self):
self.set_country_with_ip()
self.set_country_in_http_header()
self.assertEqual(geo.get_country_from_request(self.request), 'CA')
def test_use_cookie_as_secondary_source(self):
self.set_country_in_cookie()
self.set_country_with_ip()
self.set_country_in_http_header()
self.assertEqual(geo.get_country_from_request(self.request), 'AU')
def test_use_session_as_primary_source(self):
self.set_country_in_session()
self.set_country_in_cookie()
self.set_country_with_ip()
self.set_country_in_http_header()
self.assertEqual(geo.get_country_from_request(self.request), 'JP')