Left: | ||
Right: |
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-2017 eyeo GmbH | 2 # Copyright (C) 2006-2017 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 | |
111 | |
112 | |
113 def test_parse_bad_option(): | |
Vasily Kuznetsov
2017/07/28 18:57:49
Now we can also detect unknown options. Yaaay!
Vasily Kuznetsov
2017/08/02 16:21:17
I remove this test since we've agreed to remove th
| |
114 with pytest.raises(ParseError): | |
115 parse_line('foo.com$script,foo') | |
116 with pytest.raises(ParseError): | |
117 parse_line('foo.com$~foo') | |
33 | 118 |
34 | 119 |
35 def test_parse_comment(): | 120 def test_parse_comment(): |
36 line = parse_line('! Block foo') | 121 line = parse_line('! Block foo') |
37 assert line.type == 'comment' | 122 assert line.type == 'comment' |
38 assert line.text == 'Block foo' | 123 assert line.text == 'Block foo' |
39 | 124 |
40 | 125 |
41 def test_parse_meta(): | 126 def test_parse_meta(): |
42 line = parse_line('! Homepage : http://aaa.com/b') | 127 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(): | 160 def test_parse_filterlist(): |
76 result = parse_filterlist(['! foo', '! Title: bar']) | 161 result = parse_filterlist(['! foo', '! Title: bar']) |
77 assert list(result) == [Comment('foo'), Metadata('Title', 'bar')] | 162 assert list(result) == [Comment('foo'), Metadata('Title', 'bar')] |
78 | 163 |
79 | 164 |
80 def test_exception_timing(): | 165 def test_exception_timing(): |
81 result = parse_filterlist(['! good line', '%bad line%']) | 166 result = parse_filterlist(['! good line', '%bad line%']) |
82 assert next(result) == Comment('good line') | 167 assert next(result) == Comment('good line') |
83 with pytest.raises(ParseError): | 168 with pytest.raises(ParseError): |
84 next(result) | 169 next(result) |
OLD | NEW |