mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
aec45e30cc
Regions in GEOBLOCK refuse the service and everything else is allowed. A hit stops the run with a region error instead of auto-proxying; the check also covers proxy exits and the remote-dl client region.
43 lines
1001 B
Python
43 lines
1001 B
Python
"""Region checks for GEOFENCE and GEOBLOCK on the Service base class."""
|
|
|
|
from unshackle.core.service import Service
|
|
|
|
|
|
class Blocked(Service):
|
|
GEOBLOCK = ("GB",)
|
|
|
|
|
|
class Fenced(Service):
|
|
GEOFENCE = ("us",)
|
|
|
|
|
|
class Both(Service):
|
|
GEOFENCE = ("us", "gb")
|
|
GEOBLOCK = ("GB",)
|
|
|
|
|
|
def test_no_restrictions_allows_everything():
|
|
assert Service.is_region_allowed("xx")
|
|
assert Service.is_region_allowed(None)
|
|
|
|
|
|
def test_geoblock_refuses_listed_region_any_case():
|
|
assert Blocked.is_region_allowed("de")
|
|
assert not Blocked.is_region_allowed("gb")
|
|
assert not Blocked.is_region_allowed("GB")
|
|
|
|
|
|
def test_geofence_allows_only_listed_regions():
|
|
assert Fenced.is_region_allowed("US")
|
|
assert not Fenced.is_region_allowed("gb")
|
|
|
|
|
|
def test_geoblock_wins_over_geofence():
|
|
assert Both.is_region_allowed("us")
|
|
assert not Both.is_region_allowed("gb")
|
|
|
|
|
|
def test_unknown_region_is_allowed():
|
|
assert Blocked.is_region_allowed("")
|
|
assert Blocked.is_region_allowed(None)
|