OLD | NEW |
1 # This file is part of Adblock Plus <https://adblockplus.org/>, | 1 # This file is part of Adblock Plus <https://adblockplus.org/>, |
2 # Copyright (C) 2006-present eyeo GmbH | 2 # Copyright (C) 2006-present eyeo GmbH |
3 # | 3 # |
4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
7 # | 7 # |
8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
12 # | 12 # |
13 # You should have received a copy of the GNU General Public License | 13 # You should have received a copy of the GNU General Public License |
14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
15 | 15 |
16 from __future__ import unicode_literals | 16 from __future__ import unicode_literals |
17 | 17 |
18 import pytest | 18 import pytest |
19 | 19 |
20 from abp.filters import parse_line, parse_filterlist, ParseError | 20 from abp.filters import ( |
| 21 parse_line, parse_filterlist, ParseError, |
| 22 SELECTOR_TYPE as ST, FILTER_ACTION as FA, FILTER_OPTION as OPT, |
| 23 ) |
21 from abp.filters.parser import Comment, Metadata | 24 from abp.filters.parser import Comment, Metadata |
22 | 25 |
23 | 26 |
24 def test_parse_empty(): | 27 def test_parse_empty(): |
25 line = parse_line(' ') | 28 line = parse_line(' ') |
26 assert line.type == 'emptyline' | 29 assert line.type == 'emptyline' |
27 | 30 |
28 | 31 |
29 def test_parse_filter(): | 32 @pytest.mark.parametrize('filter_text, expected', { |
30 line = parse_line('||example.com/banner.gif') | 33 # Blocking filters with patterns and regexps and blocking exceptions. |
31 assert line.type == 'filter' | 34 '*asdf*d**dd*': { |
32 assert line.expression == '||example.com/banner.gif' | 35 'selector': {'type': ST.URL_PATTERN, 'value': '*asdf*d**dd*'}, |
| 36 'action': FA.BLOCK, |
| 37 }, |
| 38 '@@|*asd|f*d**dd*|': { |
| 39 'selector': {'type': ST.URL_PATTERN, 'value': '|*asd|f*d**dd*|'}, |
| 40 'action': FA.ALLOW, |
| 41 }, |
| 42 '/ddd|f?a[s]d/': { |
| 43 'selector': {'type': ST.URL_REGEXP, 'value': 'ddd|f?a[s]d'}, |
| 44 'action': FA.BLOCK, |
| 45 }, |
| 46 '@@/ddd|f?a[s]d/': { |
| 47 'selector': {'type': ST.URL_REGEXP, 'value': 'ddd|f?a[s]d'}, |
| 48 'action': FA.ALLOW, |
| 49 }, |
| 50 # Blocking filters with some options. |
| 51 'bla$match-case,~script,domain=foo.com|~bar.com,sitekey=foo': { |
| 52 'selector': {'type': ST.URL_PATTERN, 'value': 'bla'}, |
| 53 'action': FA.BLOCK, |
| 54 'options': [ |
| 55 (OPT.MATCH_CASE, True), |
| 56 (OPT.SCRIPT, False), |
| 57 (OPT.DOMAIN, [('foo.com', True), ('bar.com', False)]), |
| 58 (OPT.SITEKEY, ['foo']), |
| 59 ], |
| 60 }, |
| 61 '@@http://bla$~script,~other,sitekey=foo|bar': { |
| 62 'selector': {'type': ST.URL_PATTERN, 'value': 'http://bla'}, |
| 63 'action': FA.ALLOW, |
| 64 'options': [ |
| 65 (OPT.SCRIPT, False), |
| 66 (OPT.OTHER, False), |
| 67 (OPT.SITEKEY, ['foo', 'bar']), |
| 68 ], |
| 69 }, |
| 70 # Element hiding filters and exceptions. |
| 71 '##ddd': { |
| 72 'selector': {'type': ST.CSS, 'value': 'ddd'}, |
| 73 'action': FA.HIDE, |
| 74 'options': [], |
| 75 }, |
| 76 '#@#body > div:first-child': { |
| 77 'selector': {'type': ST.CSS, 'value': 'body > div:first-child'}, |
| 78 'action': FA.SHOW, |
| 79 'options': [], |
| 80 }, |
| 81 'foo,~bar##ddd': { |
| 82 'options': [(OPT.DOMAIN, [('foo', True), ('bar', False)])], |
| 83 }, |
| 84 # Element hiding emulation filters (extended CSS). |
| 85 'foo,~bar#?#:-abp-properties(abc)': { |
| 86 'selector': {'type': ST.XCSS, 'value': ':-abp-properties(abc)'}, |
| 87 'action': FA.HIDE, |
| 88 'options': [(OPT.DOMAIN, [('foo', True), ('bar', False)])], |
| 89 }, |
| 90 'foo.com#?#aaa :-abp-properties(abc) bbb': { |
| 91 'selector': { |
| 92 'type': ST.XCSS, |
| 93 'value': 'aaa :-abp-properties(abc) bbb' |
| 94 }, |
| 95 }, |
| 96 '#?#:-abp-properties(|background-image: url(data:*))': { |
| 97 'selector': { |
| 98 'type': ST.XCSS, |
| 99 'value': ':-abp-properties(|background-image: url(data:*))' |
| 100 }, |
| 101 'options': [], |
| 102 }, |
| 103 }.items()) |
| 104 def test_parse_filters(filter_text, expected): |
| 105 """Parametric test for filter parsing""" |
| 106 parsed = parse_line(filter_text) |
| 107 assert parsed.type == 'filter' |
| 108 assert parsed.text == filter_text |
| 109 for attribute, expected_value in expected.items(): |
| 110 assert getattr(parsed, attribute) == expected_value |
33 | 111 |
34 | 112 |
35 def test_parse_comment(): | 113 def test_parse_comment(): |
36 line = parse_line('! Block foo') | 114 line = parse_line('! Block foo') |
37 assert line.type == 'comment' | 115 assert line.type == 'comment' |
38 assert line.text == 'Block foo' | 116 assert line.text == 'Block foo' |
39 | 117 |
40 | 118 |
41 def test_parse_meta(): | 119 def test_parse_meta(): |
42 line = parse_line('! Homepage : http://aaa.com/b') | 120 line = parse_line('! Homepage : http://aaa.com/b') |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
75 def test_parse_filterlist(): | 153 def test_parse_filterlist(): |
76 result = parse_filterlist(['! foo', '! Title: bar']) | 154 result = parse_filterlist(['! foo', '! Title: bar']) |
77 assert list(result) == [Comment('foo'), Metadata('Title', 'bar')] | 155 assert list(result) == [Comment('foo'), Metadata('Title', 'bar')] |
78 | 156 |
79 | 157 |
80 def test_exception_timing(): | 158 def test_exception_timing(): |
81 result = parse_filterlist(['! good line', '%bad line%']) | 159 result = parse_filterlist(['! good line', '%bad line%']) |
82 assert next(result) == Comment('good line') | 160 assert next(result) == Comment('good line') |
83 with pytest.raises(ParseError): | 161 with pytest.raises(ParseError): |
84 next(result) | 162 next(result) |
OLD | NEW |