Files
privacore-open-source-searc…/test/unit/UrlBlockListTest.cpp
2017-03-07 12:43:51 +01:00

89 lines
4.0 KiB
C++

#include <gtest/gtest.h>
#include "UrlBlockList.h"
#include "Url.h"
class TestUrlBlockList : public UrlBlockList {
public:
TestUrlBlockList(const char *filename)
: UrlBlockList() {
m_filename = filename;
}
bool isUrlBlocked(const char *urlStr) {
Url url;
url.set(urlStr);
return UrlBlockList::isUrlBlocked(url);
}
using UrlBlockList::load;
};
TEST(UrlBlockListTest, Domain) {
TestUrlBlockList urlBlockList("blocklist/domain.txt");
urlBlockList.load();
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.badsite.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.badsite.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("httpp://www.badsite.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.badsite.com/page.html"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.httponly.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.httponly.com/page.html"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("https://www.httponly.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://subdomain.httponly.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.httpsonly.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.httpsonly.com/page.html"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://www.httpsonly.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("https://subdomain.httpsonly.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.allsubdomain.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://sub1.allsubdomain.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://sub2.allsubdomain.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://sub1.sub2.allsubdomain.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://allsubdomain.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://something.com/sub1.allsubdomain.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://something.com/www.allsubdomain.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://sub1.diffdomain.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.onlyroot.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://www.onlyroot.com/page.html"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://sub1.sub2.example.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://sub1.example.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.sub1.example.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://www.example.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://example.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://specific.host.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://specific.host.com/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("https://www.host.com/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://specific.host.dk/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.host.my/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.host.com.my/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.somesite.com/badpath/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.somesite.com/badpath/me.html"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://www.somesite.com/path/me.html"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://sub.somesite.com/badpath/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://www.itsybitsy.com/spider/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://itsybitsy.com/spider/waterspout.html"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.itsybitsy.com/spider/waterspout.html"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://nursery.itsybitsy.com/spider/waterspout.html"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://rhyme.itsybitsy.com/spider/waterspout.html"));
}
TEST(UrlBlockListTest, Path) {
TestUrlBlockList urlBlockList("blocklist/path.txt");
urlBlockList.load();
EXPECT_TRUE(urlBlockList.isUrlBlocked("http://www.example.com/wp-admin/"));
EXPECT_FALSE(urlBlockList.isUrlBlocked("http://www.example.com/tag/wp-admin/"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.host.com/file1.html"));
EXPECT_TRUE(urlBlockList.isUrlBlocked("https://www.example.com/file1.html"));
}