34 lines
1.4 KiB
Python
34 lines
1.4 KiB
Python
import pytest
|
|
|
|
from proxydoctor.media import Page, classify
|
|
|
|
|
|
@pytest.mark.parametrize('service,message',[
|
|
('Netflix','Netflix is not available in your country'),
|
|
('Disney+','Disney+ is not available in your region'),
|
|
('YouTube Premium','Premium is not available in your country'),
|
|
])
|
|
def test_explicit_region_restriction(service,message):
|
|
assert classify(service,Page(200,'<p>'+message+'</p>'))['state']=='restricted'
|
|
|
|
|
|
@pytest.mark.parametrize('status',[200,301,403,404,429,503,None])
|
|
def test_status_alone_never_means_playback_or_geoblock(status):
|
|
assert classify('Netflix',Page(status,'<h1>Hello</h1>'))['state']=='unknown'
|
|
|
|
|
|
def test_translations_in_scripts_are_not_geoblock_evidence():
|
|
page=Page(200,'<script>"Netflix is not available in your country"</script><p>Join now</p>')
|
|
assert classify('Netflix',page)['state']=='unknown'
|
|
|
|
|
|
def test_partial_response_does_not_support_classification():
|
|
page=Page(200,'Netflix is not available in your country',reason='response_too_large')
|
|
assert classify('Netflix',page)['state']=='unknown'
|
|
|
|
|
|
@pytest.mark.parametrize('attribute',['hidden','aria-hidden="true"','style="display: none"'])
|
|
def test_explicitly_hidden_region_messages_are_not_evidence(attribute):
|
|
page=Page(200,f'<div {attribute}>Netflix is not available in your country</div><p>Welcome</p>')
|
|
assert classify('Netflix',page)['state']=='unknown'
|