def test_get_cloudfront_country_with_no_header(rf): segment = SegmentFactory(name='Test segment') rule = OriginCountryRuleFactory(segment=segment, country='GB') request = rf.get('/') assert 'HTTP_CLOUDFRONT_VIEWER_COUNTRY' not in request.META assert rule.get_cloudfront_country(request) is None
def test_get_cloudfront_country_with_header(rf): segment = SegmentFactory(name='Test segment') rule = OriginCountryRuleFactory(segment=segment, country='GB') request = rf.get('/', HTTP_CLOUDFRONT_VIEWER_COUNTRY='BY') assert rule.get_cloudfront_country(request) == 'by'