def test_get_cloudflare_country_with_no_header(rf): segment = SegmentFactory(name='Test segment') rule = OriginCountryRuleFactory(segment=segment, country='GB') request = rf.get('/') assert 'HTTP_CF_IPCOUNTRY' not in request.META assert rule.get_cloudflare_country(request) is None
def test_get_cloudflare_country_with_header(rf): segment = SegmentFactory(name='Test segment') rule = OriginCountryRuleFactory(segment=segment, country='GB') request = rf.get('/', HTTP_CF_IPCOUNTRY='PL') assert rule.get_cloudflare_country(request) == 'pl'