#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")); }