Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: tests/test_filters_blocks.py

Issue 30053555: Issue 7471 - Add an API for working with blocks of filters (Closed) Base URL: https://hg.adblockplus.org/python-abp
Patch Set: Refine the API and add documentation Created May 9, 2019, 11:13 a.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 # This file is part of Adblock Plus <https://adblockplus.org/>,
2 # Copyright (C) 2006-present eyeo GmbH
3 #
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
6 # published by the Free Software Foundation.
7 #
8 # Adblock Plus is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # GNU General Public License for more details.
12 #
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/>.
15
16 """Tests for abp.filters.blocks."""
17
18 from __future__ import unicode_literals
19
20 import json
21 import os
22
23 import pytest
24
25 from abp.filters import (parse_line, parse_filterlist, SelectorType,
26 FilterAction, ParseError)
27 from abp.filters.blocks import to_blocks, FiltersBlock
28
29 DATA_PATH = os.path.join(os.path.dirname(__file__), 'data')
30
31
32 @pytest.fixture()
33 def fl_lines():
34 with open(os.path.join(DATA_PATH, 'filterlist.txt')) as f:
35 return list(parse_filterlist(f))
36
37
38 @pytest.fixture()
39 def expected_blocks():
40 with open(os.path.join(DATA_PATH, 'expected_blocks.json')) as f:
41 return json.load(f)
42
43
44 def test_to_blocks(fl_lines):
45 blocks = list(to_blocks(fl_lines))
46 assert len(blocks) == 2
47 block = blocks[0]
48 assert block.foo == 'bar'
49 assert block.baz == 'some_tricky?variable=with&funny=chars#and-stuff'
50 assert block.description == 'Example block 1\nAnother comment'
51 # Don't test the filters thouroughly: filter parsing is tested elsewhere.
52 assert len(block.filters) == 2
53 assert block.filters[0].selector['type'] == SelectorType.URL_PATTERN
54 assert block.filters[1].action == FilterAction.SHOW
55
56
57 def test_to_dict(fl_lines, expected_blocks):
58 blocks = [b.to_dict() for b in to_blocks(fl_lines)]
59 print(json.dumps(blocks, indent=2))
60 assert blocks == expected_blocks
61
62
63 @pytest.mark.parametrize('line', [
64 '!:__dict__=foo',
65 '!:_var=bar',
66 '!:filters=stuff',
67 '!:description=he he',
68 ])
69 def test_illegal_vars(line):
70 """Check that illegal variable names are not allowed."""
71 with pytest.raises(ParseError):
72 FiltersBlock([parse_line(line)], [])
OLDNEW

Powered by Google App Engine
This is Rietveld