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'