From 64c1fa2c91b55a35eceeffab6e0715374a5a7231 Mon Sep 17 00:00:00 2001 From: dm Date: Thu, 17 Feb 2022 10:44:09 +0800 Subject: fix: black lint --- tests/test_redirect.py | 23 ++++++++++++++++------- 1 file changed, 16 insertions(+), 7 deletions(-) (limited to 'tests/test_redirect.py') 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 def test_redirect_worked_as_excepted_with_include_redirect_urls(): # https://github.com/brave/adblock-rust/blob/b7f29af8c0a0d000201d8d769b6a0b25a9dd4e89/src/blocker.rs#L1242 filter_set = adblock.FilterSet() - filter_set.add_filter_list("||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=True) + filter_set.add_filter_list( + "||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=True + ) engine = adblock.Engine(filter_set=filter_set) @@ -18,7 +20,9 @@ def test_redirect_worked_as_excepted_with_include_redirect_urls(): def test_redirect_url_is_not_recognized_without_include_redirect_urls(): # https://github.com/brave/adblock-rust/blob/b7f29af8c0a0d000201d8d769b6a0b25a9dd4e89/src/blocker.rs#L1267 filter_set2 = adblock.FilterSet() - filter_set2.add_filter_list("||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=False) + filter_set2.add_filter_list( + "||foo.com$important,redirect-url=http://xyz.com", include_redirect_urls=False + ) engine2 = adblock.Engine(filter_set=filter_set2) @@ -36,13 +40,15 @@ def test_redirect_url_exception(): ||imdb-video.media-imdb.com$media,redirect-url=http://xyz.com @@||imdb-video.media-imdb.com^$domain=imdb.com """, - include_redirect_urls=True + include_redirect_urls=True, ) engine2 = adblock.Engine(filter_set=filter_set, optimize=False) res = engine2.check_network_urls( - "https://imdb-video.media-imdb.com/kBOeI88k1o23eNAi", "https://www.imdb.com/video/13", "media" + "https://imdb-video.media-imdb.com/kBOeI88k1o23eNAi", + "https://www.imdb.com/video/13", + "media", ) assert res.matched is False assert res.redirect == "http://xyz.com" @@ -55,14 +61,17 @@ def test_redirect_with_custom_resource(): filters.add_filter_list("-advertisement-$redirect=test\n") engine = adblock.Engine(filter_set=filters) - engine.add_resource(name="test", content_type="application/javascript", content="YWxlcnQoMSk=") + engine.add_resource( + name="test", content_type="application/javascript", content="YWxlcnQoMSk=" + ) result = engine.check_network_urls( - url="http://example.com/-advertisement-icon.", source_url="example.com", request_type="image", + url="http://example.com/-advertisement-icon.", + source_url="example.com", + request_type="image", ) assert result.matched assert not result.exception assert not result.important assert result.redirect == "data:application/javascript;base64,YWxlcnQoMSk=" - -- cgit v1.2.3