| Index: tests/test_stats/test_filtehits_loader.py |
| diff --git a/tests/test_stats/test_filtehits_loader.py b/tests/test_stats/test_filtehits_loader.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..3ecfc3fc3398377aed59cba1ac74a273fdbbfca5 |
| --- /dev/null |
| +++ b/tests/test_stats/test_filtehits_loader.py |
| @@ -0,0 +1,71 @@ |
| +# This file is part of Adblock Plus <https://adblockplus.org/>, |
| +# Copyright (C) 2006-present eyeo GmbH |
| +# |
| +# Adblock Plus is free software: you can redistribute it and/or modify |
| +# it under the terms of the GNU General Public License version 3 as |
| +# published by the Free Software Foundation. |
| +# |
| +# Adblock Plus is distributed in the hope that it will be useful, |
| +# but WITHOUT ANY WARRANTY; without even the implied warranty of |
| +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| +# GNU General Public License for more details. |
| +# |
| +# You should have received a copy of the GNU General Public License |
| +# along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| + |
| +import pytest |
| +import py |
| + |
| +from abp.stats.filterhits import load_filterhit_statistics |
| + |
| + |
| +@pytest.fixture |
| +def filterhits_file(): |
| + return py.path.local(__file__).dirpath('data').join('filterhits.csv') |
| + |
| + |
| +@pytest.fixture |
| +def filterhits_file_missing_columns(): |
| + return py.path.local(__file__).dirpath('data').join( |
| + 'filterhits_missing_columns.csv', |
| + ) |
| + |
| + |
| +def test_filterhits_load_no_filtering(filterhits_file): |
| + entries = load_filterhit_statistics(str(filterhits_file)) |
| + |
| + count = 0 |
| + |
| + for entry in entries: |
| + count += 1 |
| + assert isinstance(entry['hits'], int) |
| + assert isinstance(entry['onehour_sessions'], int) |
| + assert isinstance(entry['domains'], int) |
| + assert isinstance(entry['rootdomains'], int) |
| + |
| + assert count == 2 |
| + |
| + |
| +@pytest.mark.parametrize('sources,exp_count', [ |
| + (['www.exceptionlist.com'], 1), |
| + (['www.exceptionlist.com', 'www.blocklist.com'], 2), |
| + (['inexistent_source', 'foo', 'bar'], 0), |
| +]) |
| +def test_filterhits_load_with_filtering(sources, exp_count, filterhits_file): |
| + entries = load_filterhit_statistics(str(filterhits_file), sources) |
| + |
| + count = 0 |
| + for _ in entries: |
| + count += 1 |
| + |
| + assert count == exp_count |
| + |
| + |
| +def test_filterhits_load_missing_columns(filterhits_file_missing_columns): |
| + entries = load_filterhit_statistics(str(filterhits_file_missing_columns)) |
| + |
| + count = 0 |
| + for _ in entries: |
| + count += 1 |
| + |
| + assert count == 2 |