summaryrefslogtreecommitdiff
path: root/tests/test_redirect.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_redirect.py')
-rw-r--r--tests/test_redirect.py23
1 files changed, 16 insertions, 7 deletions
diff --git a/tests/test_redirect.py b/tests/test_redirect.py
index a99a728..cca3a91 100644
--- a/tests/test_redirect.py
+++ b/tests/test_redirect.py
@@ -4,7 +4,9 @@ import adblock
4def test_redirect_worked_as_excepted_with_include_redirect_urls(): 4def test_redirect_worked_as_excepted_with_include_redirect_urls():
5 # https://github.com/brave/adblock-rust/blob/b7f29af8c0a0d000201d8d769b6a0b25a9dd4e89/src/blocker.rs#L1242 5 # https://github.com/brave/adblock-rust/blob/b7f29af8c0a0d000201d8d769b6a0b25a9dd4e89/src/blocker.rs#L1242
6 filter_set = adblock.FilterSet() 6 filter_set = adblock.FilterSet()
7 filter_set.add_filter_list("||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=True) 7 filter_set.add_filter_list(
8 "||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=True
9 )
8 10
9 engine = adblock.Engine(filter_set=filter_set) 11 engine = adblock.Engine(filter_set=filter_set)
10 12
@@ -18,7 +20,9 @@ def test_redirect_worked_as_excepted_with_include_redirect_urls():
18def test_redirect_url_is_not_recognized_without_include_redirect_urls(): 20def test_redirect_url_is_not_recognized_without_include_redirect_urls():
19 # https://github.com/brave/adblock-rust/blob/b7f29af8c0a0d000201d8d769b6a0b25a9dd4e89/src/blocker.rs#L1267 21 # https://github.com/brave/adblock-rust/blob/b7f29af8c0a0d000201d8d769b6a0b25a9dd4e89/src/blocker.rs#L1267
20 filter_set2 = adblock.FilterSet() 22 filter_set2 = adblock.FilterSet()
21 filter_set2.add_filter_list("||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=False) 23 filter_set2.add_filter_list(
24 "||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=False
25 )
22 26
23 engine2 = adblock.Engine(filter_set=filter_set2) 27 engine2 = adblock.Engine(filter_set=filter_set2)
24 28
@@ -36,13 +40,15 @@ def test_redirect_url_exception():
36 ||imdb-video.media-imdb.com$media,redirect-url=http://xyz.com 40 ||imdb-video.media-imdb.com$media,redirect-url=http://xyz.com
37 @@||imdb-video.media-imdb.com^$domain=imdb.com 41 @@||imdb-video.media-imdb.com^$domain=imdb.com
38 """, 42 """,
39 include_redirect_urls=True 43 include_redirect_urls=True,
40 ) 44 )
41 45
42 engine2 = adblock.Engine(filter_set=filter_set, optimize=False) 46 engine2 = adblock.Engine(filter_set=filter_set, optimize=False)
43 47
44 res = engine2.check_network_urls( 48 res = engine2.check_network_urls(
45 "https://imdb-video.media-imdb.com/kBOeI88k1o23eNAi", "https://www.imdb.com/video/13", "media" 49 "https://imdb-video.media-imdb.com/kBOeI88k1o23eNAi",
50 "https://www.imdb.com/video/13",
51 "media",
46 ) 52 )
47 assert res.matched is False 53 assert res.matched is False
48 assert res.redirect == "http://xyz.com" 54 assert res.redirect == "http://xyz.com"
@@ -55,14 +61,17 @@ def test_redirect_with_custom_resource():
55 filters.add_filter_list("-advertisement-$redirect=test\n") 61 filters.add_filter_list("-advertisement-$redirect=test\n")
56 62
57 engine = adblock.Engine(filter_set=filters) 63 engine = adblock.Engine(filter_set=filters)
58 engine.add_resource(name="test", content_type="application/javascript", content="YWxlcnQoMSk=") 64 engine.add_resource(
65 name="test", content_type="application/javascript", content="YWxlcnQoMSk="
66 )
59 67
60 result = engine.check_network_urls( 68 result = engine.check_network_urls(
61 url="http://example.com/-advertisement-icon.", source_url="example.com", request_type="image", 69 url="http://example.com/-advertisement-icon.",
70 source_url="example.com",
71 request_type="image",
62 ) 72 )
63 73
64 assert result.matched 74 assert result.matched
65 assert not result.exception 75 assert not result.exception
66 assert not result.important 76 assert not result.important
67 assert result.redirect == "data:application/javascript;base64,YWxlcnQoMSk=" 77 assert result.redirect == "data:application/javascript;base64,YWxlcnQoMSk="
68