summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authordm <dm@work-pc>2022-02-17 10:44:09 +0800
committerÁrni Dagur <arni@dagur.eu>2022-02-26 22:06:53 +0100
commit64c1fa2c91b55a35eceeffab6e0715374a5a7231 (patch)
treebbbad7873246febaed3c9512024ac2ba6bec9584 /tests
parent3bbed50af769e2310c0bb366ab12a1d22c115082 (diff)
fix: black lint
Diffstat (limited to 'tests')
-rw-r--r--tests/test_exceptions.py3
-rw-r--r--tests/test_redirect.py23
2 files changed, 18 insertions, 8 deletions
diff --git a/tests/test_exceptions.py b/tests/test_exceptions.py
index 3e3dc9f..2efe2b2 100644
--- a/tests/test_exceptions.py
+++ b/tests/test_exceptions.py
@@ -1,6 +1,7 @@
1import adblock 1import adblock
2import pytest 2import pytest
3 3
4
4def test_correct_baseclasses(): 5def test_correct_baseclasses():
5 assert issubclass(adblock.AdblockException, Exception) 6 assert issubclass(adblock.AdblockException, Exception)
6 assert issubclass(adblock.BlockerException, adblock.AdblockException) 7 assert issubclass(adblock.BlockerException, adblock.AdblockException)
@@ -17,6 +18,6 @@ def test_add_resource_error():
17 engine = adblock.Engine(filter_set=filter_set) 18 engine = adblock.Engine(filter_set=filter_set)
18 19
19 with pytest.raises(adblock.AddResourceError) as exc: 20 with pytest.raises(adblock.AddResourceError) as exc:
20 engine.add_resource(name='aa', content_type="image/jpeg", content="111") 21 engine.add_resource(name="aa", content_type="image/jpeg", content="111")
21 22
22 assert "invalid base64 content" in str(exc.value) 23 assert "invalid base64 content" in str(exc.value)
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