diff options
| author | dm <dm@work-pc> | 2022-02-17 10:44:09 +0800 |
|---|---|---|
| committer | Árni Dagur <arni@dagur.eu> | 2022-02-26 22:06:53 +0100 |
| commit | 64c1fa2c91b55a35eceeffab6e0715374a5a7231 (patch) | |
| tree | bbbad7873246febaed3c9512024ac2ba6bec9584 /tests | |
| parent | 3bbed50af769e2310c0bb366ab12a1d22c115082 (diff) | |
fix: black lint
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_exceptions.py | 3 | ||||
| -rw-r--r-- | tests/test_redirect.py | 23 |
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 @@ | |||
| 1 | import adblock | 1 | import adblock |
| 2 | import pytest | 2 | import pytest |
| 3 | 3 | ||
| 4 | |||
| 4 | def test_correct_baseclasses(): | 5 | def 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 | |||
| 4 | def test_redirect_worked_as_excepted_with_include_redirect_urls(): | 4 | def 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(): | |||
| 18 | def test_redirect_url_is_not_recognized_without_include_redirect_urls(): | 20 | def 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 | |||
