Issue 3363 - Implement IsDocumentWhitelisted and IsElemhideWhitelisted

test/FilterEngine.cpp

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2015 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
@@ skipping 395 matching lines @@
   filterEngine->SetUpdateAvailableCallback(mockUpdateAvailableCallback);
   filterEngine->ForceUpdateCheck(&NoOpUpdaterCallback);
   AdblockPlus::Sleep(100);
   ASSERT_EQ(1, timesCalled);
 
   filterEngine->RemoveUpdateAvailableCallback();
   filterEngine->ForceUpdateCheck(&NoOpUpdaterCallback);
   AdblockPlus::Sleep(100);
   ASSERT_EQ(1, timesCalled);
 }
+
+TEST_F(FilterEngineTest, DocumentWhitelisting)
+{
+  filterEngine->GetFilter("@@||example.org^$document")->AddToList();
+  filterEngine->GetFilter("@@||example.com^$document,domain=example.de")->AddToList();
+
+  ASSERT_TRUE(filterEngine->IsDocumentWhitelisted(
+      "http://example.org",
+      std::vector<std::string>()));
+
+  ASSERT_FALSE(filterEngine->IsDocumentWhitelisted(
+      "http://example.co.uk",
+      std::vector<std::string>()));
+
+  ASSERT_FALSE(filterEngine->IsDocumentWhitelisted(
+      "http://example.com",
+      std::vector<std::string>()));
+
+  std::vector<std::string> documentUrls1;
+  documentUrls1.push_back("http://example.de");
+
+  ASSERT_TRUE(filterEngine->IsDocumentWhitelisted(
+      "http://example.com",
+      documentUrls1));
+
+  ASSERT_FALSE(filterEngine->IsDocumentWhitelisted(
+      "http://example.co.uk",
+      documentUrls1));
+}
+
+TEST_F(FilterEngineTest, ElemhideWhitelisting)
+{
+  filterEngine->GetFilter("@@||example.org^$elemhide")->AddToList();
+  filterEngine->GetFilter("@@||example.com^$elemhide,domain=example.de")->AddToList();
+
+  ASSERT_TRUE(filterEngine->IsElemhideWhitelisted(
+      "http://example.org",
+      std::vector<std::string>()));
+
+  ASSERT_FALSE(filterEngine->IsElemhideWhitelisted(
+      "http://example.co.uk",
+      std::vector<std::string>()));
+
+  ASSERT_FALSE(filterEngine->IsElemhideWhitelisted(
+      "http://example.com",
+      std::vector<std::string>()));
+
+  std::vector<std::string> documentUrls1;
+  documentUrls1.push_back("http://example.de");
+
+  ASSERT_TRUE(filterEngine->IsElemhideWhitelisted(
+      "http://example.com",
+      documentUrls1));
+
+  ASSERT_FALSE(filterEngine->IsElemhideWhitelisted(
+      "http://example.co.uk",
+      documentUrls1));
+}