Issue 1564-Fix FilterEngine::Matches for allowing request which is whitelisted in the ascendant node

test/ReferrerMapping.cpp

 /*
  * This file is part of Adblock Plus <http://adblockplus.org/>,
  * Copyright (C) 2006-2014 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/>.
  */
 
 #include <AdblockPlus.h>
 #include <gtest/gtest.h>
 
-TEST(ReferrerMappingTest, EmptyReferrerChain)
+typedef AdblockPlus::FilterEngine::ContentType ContentType;
+
+TEST(ReferrerMappingTest, BuildReferrerChainShouldIncludeUrlItselfWhenThereIsNoMappingEntryForIt)
 {
   AdblockPlus::ReferrerMapping referrerMapping;
   std::vector<std::string> referrerChain =
     referrerMapping.BuildReferrerChain("first");
   ASSERT_EQ(1, referrerChain.size());
   ASSERT_EQ("first", referrerChain[0]);
 }
 
-TEST(ReferrerMappingTest, TwoElementReferrerChain)
+TEST(ReferrerMappingTest, BuildReferrerChainShouldNotIncludeUrlItselfWhenItsTypeIsNotSubdocument)
 {
   AdblockPlus::ReferrerMapping referrerMapping;
-  referrerMapping.Add("second", "first");
+  referrerMapping.Add("first", "", ContentType::CONTENT_TYPE_IMAGE);
+  std::vector<std::string> referrerChain =
+    referrerMapping.BuildReferrerChain("first");
+  ASSERT_EQ(0, referrerChain.size());
+}
+
+TEST(ReferrerMappingTest, BuildReferrerChainShouldIncludeReferrerIfThereIsNoInfoAboutIt)
+{
+  AdblockPlus::ReferrerMapping referrerMapping;
+  referrerMapping.Add("second", "first", ContentType::CONTENT_TYPE_SUBDOCUMENT);
   std::vector<std::string> referrerChain =
     referrerMapping.BuildReferrerChain("second");
   ASSERT_EQ(2, referrerChain.size());
   ASSERT_EQ("first", referrerChain[0]);
   ASSERT_EQ("second", referrerChain[1]);
 }
 
 TEST(ReferrerMappingTest, TenElementReferrerChain)
 {
   AdblockPlus::ReferrerMapping referrerMapping;
-  referrerMapping.Add("second", "first");
-  referrerMapping.Add("third", "second");
-  referrerMapping.Add("fourth", "third");
-  referrerMapping.Add("fifth", "fourth");
-  referrerMapping.Add("sixth", "fifth");
-  referrerMapping.Add("seventh", "sixth");
-  referrerMapping.Add("eighth", "seventh");
-  referrerMapping.Add("ninth", "eighth");
-  referrerMapping.Add("tenth", "ninth");
+  referrerMapping.Add("second", "first", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("third", "second", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("fourth", "third", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("fifth", "fourth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("sixth", "fifth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("seventh", "sixth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("eighth", "seventh", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("ninth", "eighth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("tenth", "ninth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
   std::vector<std::string> referrerChain =
     referrerMapping.BuildReferrerChain("tenth");
   ASSERT_EQ(10, referrerChain.size());
   ASSERT_EQ("first", referrerChain[0]);
   ASSERT_EQ("second", referrerChain[1]);
   ASSERT_EQ("third", referrerChain[2]);
   ASSERT_EQ("fourth", referrerChain[3]);
   ASSERT_EQ("fifth", referrerChain[4]);
   ASSERT_EQ("sixth", referrerChain[5]);
   ASSERT_EQ("seventh", referrerChain[6]);
   ASSERT_EQ("eighth", referrerChain[7]);
   ASSERT_EQ("ninth", referrerChain[8]);
   ASSERT_EQ("tenth", referrerChain[9]);
 }
 
 TEST(ReferrerMappingTest, CacheOnlyFiveUrls)
 {
   AdblockPlus::ReferrerMapping referrerMapping(5);
-  referrerMapping.Add("second", "first");
-  referrerMapping.Add("third", "second");
-  referrerMapping.Add("fourth", "third");
-  referrerMapping.Add("fifth", "fourth");
-  referrerMapping.Add("sixth", "fifth");
-  referrerMapping.Add("seventh", "sixth");
+  std::string outOfCacheUrl = "first";
+  referrerMapping.Add("second", outOfCacheUrl, ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("third", "second", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("fourth", "third", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("fifth", "fourth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("sixth", "fifth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("seventh", "sixth", ContentType::CONTENT_TYPE_SUBDOCUMENT);
   std::vector<std::string> referrerChain =
     referrerMapping.BuildReferrerChain("seventh");
   ASSERT_EQ(6, referrerChain.size());
   ASSERT_EQ("second", referrerChain[0]);
   ASSERT_EQ("third", referrerChain[1]);
   ASSERT_EQ("fourth", referrerChain[2]);
   ASSERT_EQ("fifth", referrerChain[3]);
   ASSERT_EQ("sixth", referrerChain[4]);
   ASSERT_EQ("seventh", referrerChain[5]);
 }
+
+TEST(ReferrerMappingTest, BuildReferrerChainShouldIncludeOnlySubdocumentReferrers)
+{
+  AdblockPlus::ReferrerMapping referrerMapping;
+  referrerMapping.Add("iframe1", "root-page", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("other", "iframe1", ContentType::CONTENT_TYPE_OTHER);
+  referrerMapping.Add("script", "other", ContentType::CONTENT_TYPE_SCRIPT);
+  referrerMapping.Add("image", "script", ContentType::CONTENT_TYPE_IMAGE);
+  referrerMapping.Add("stylesheet", "image", ContentType::CONTENT_TYPE_STYLESHEET);
+  referrerMapping.Add("object", "stylesheet", ContentType::CONTENT_TYPE_OBJECT);
+  referrerMapping.Add("iframe2", "object", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerMapping.Add("document", "iframe2", ContentType::CONTENT_TYPE_DOCUMENT);
+  referrerMapping.Add("xmlhttprequest", "document", ContentType::CONTENT_TYPE_XMLHTTPREQUEST);
+  referrerMapping.Add("object subrequest", "xmlhttprequest", ContentType::CONTENT_TYPE_OBJECT_SUBREQUEST);
+  referrerMapping.Add("font", "object subrequest", ContentType::CONTENT_TYPE_FONT);
+  referrerMapping.Add("media", "font", ContentType::CONTENT_TYPE_MEDIA);
+
+  std::vector<std::string> referrerChain = referrerMapping.BuildReferrerChain("media");
+  ASSERT_EQ(3, referrerChain.size());
+  EXPECT_EQ("root-page", referrerChain[0]);
+  EXPECT_EQ("iframe1", referrerChain[1]);
+  EXPECT_EQ("iframe2", referrerChain[2]);
+
+  referrerMapping.Add("iframe3", "media", ContentType::CONTENT_TYPE_SUBDOCUMENT);
+  referrerChain = referrerMapping.BuildReferrerChain("iframe3");
+  ASSERT_EQ(4, referrerChain.size());
+  EXPECT_EQ("root-page", referrerChain[0]);
+  EXPECT_EQ("iframe1", referrerChain[1]);
+  EXPECT_EQ("iframe2", referrerChain[2]);
+  EXPECT_EQ("iframe3", referrerChain[3]);
+}