From c74367d82c1c7bec393548d2e5014c794333822f Mon Sep 17 00:00:00 2001
From: Aqua-sama <aqua@iserlohn-fortress.net>
Date: Tue, 16 Oct 2018 17:25:40 +0200
Subject: urlfilter: Add FilterTree class

FilterTree is a class that holds filter rules, sorted by the domain they
are to be applied on. The rules are to follow FilterLeaf as interface.
- Add a hostlist rule format to FilterTree.
- Add a test for hostlist format.
---
 test/hostlist/hostlisttest.cpp | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 test/hostlist/hostlisttest.cpp

(limited to 'test/hostlist/hostlisttest.cpp')

diff --git a/test/hostlist/hostlisttest.cpp b/test/hostlist/hostlisttest.cpp
new file mode 100644
index 0000000..46f6a85
--- /dev/null
+++ b/test/hostlist/hostlisttest.cpp
@@ -0,0 +1,42 @@
+#include "hostlisttest.h"
+#include <QtTest/QtTest>
+
+void HostlistTest::parseList()
+{
+    //FilterTree tree;
+
+    // load filters
+    QFile hostlist("hostlist.txt");
+    QCOMPARE(hostlist.open(QIODevice::ReadOnly | QIODevice::Text), true);
+    QCOMPARE(loadHostlist(hostlist, &tree), true);
+
+    QCOMPARE(tree.branches().length(), 1);
+}
+
+void HostlistTest::checkRules()
+{
+    // test block
+    QVector<const FilterLeaf *> block = tree.match(QString(), "blockeddomain.com");
+    QCOMPARE(block.length(), 1);
+    QCOMPARE(block.constFirst()->action(), FilterLeaf::Block);
+
+    // test redirect
+    QVector<const FilterLeaf *> redirectResult = tree.match(QString(), "localhost.localdomain");
+    QCOMPARE(redirectResult.length(), 1);
+    QCOMPARE(redirectResult.at(0)->action(), FilterLeaf::Redirect);
+    QCOMPARE(redirectResult.at(0)->redirect(), "127.0.0.1");
+
+    // two domains on one line
+    QVector<const FilterLeaf *> blockFirst = tree.match(QString(), "blockeddomain.first");
+    QCOMPARE(blockFirst.length(), 1);
+    QCOMPARE(blockFirst.constFirst()->action(), FilterLeaf::Block);
+    QVector<const FilterLeaf *> blockSecond = tree.match(QString(), "blockeddomain.second");
+    QCOMPARE(blockSecond.length(), 1);
+    QCOMPARE(blockSecond.constFirst()->action(), FilterLeaf::Block);
+
+    // domain not on list
+    QVector<const FilterLeaf *> missing = tree.match(QString(), "other.domain");
+    QCOMPARE(missing.length(), 0);
+}
+
+QTEST_GUILESS_MAIN(HostlistTest)
-- 
cgit v1.2.1