Cache is_third_party requests in a one-entry cache.

components/subresource_filter/core/common/first_party_origin_unittest.cc

Index: components/subresource_filter/core/common/first_party_origin_unittest.cc
diff --git a/components/subresource_filter/core/common/first_party_origin_unittest.cc b/components/subresource_filter/core/common/first_party_origin_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..02ba3026486bb813bfc9f16a6c7389e2c19c4c0d
--- /dev/null
+++ b/components/subresource_filter/core/common/first_party_origin_unittest.cc
@@ -0,0 +1,82 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "components/subresource_filter/core/common/first_party_origin.h"
+
+#include "base/strings/string_number_conversions.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace subresource_filter {
+
+TEST(FirstPartyOriginTest, AllSameDomain) {
+  const std::string kDomain = "sub.example.co.uk";
+
+  FirstPartyOrigin first_party(url::Origin(GURL("https://" + kDomain)));
+  for (int index = 0; index < 5; ++index) {
+    GURL url("https://" + kDomain + "/path?q=" + base::IntToString(index));
+    EXPECT_FALSE(FirstPartyOrigin::IsThirdParty(url, first_party.origin()));
+    EXPECT_FALSE(first_party.IsThirdParty(url));
+  }
+}
+
+TEST(FirstPartyOriginTest, AllFirstParty) {
+  const std::string kDomain = "example.co.uk";
+
+  FirstPartyOrigin first_party(url::Origin(GURL("https://" + kDomain)));
+  for (int index = 0; index < 5; ++index) {
+    GURL url("https://sub" + base::IntToString(index) + "." + kDomain + "/suf");
+    EXPECT_FALSE(FirstPartyOrigin::IsThirdParty(url, first_party.origin()));
+    EXPECT_FALSE(first_party.IsThirdParty(url));
+  }
+}
+
+TEST(FirstPartyOriginTest, AllThirdParty) {
+  const std::string kDomain = "example.co.uk";
+
+  FirstPartyOrigin first_party(url::Origin(GURL("https://" + kDomain)));
+  for (int index = 0; index < 5; ++index) {
+    GURL url("https://example" + base::IntToString(index) + ".co.uk/path?k=v");
+    EXPECT_TRUE(FirstPartyOrigin::IsThirdParty(url, first_party.origin()));
+    EXPECT_TRUE(first_party.IsThirdParty(url));
+  }
+}
+
+TEST(FirstPartyOriginTest, MixedFirstAndThirdParties) {
+  const struct {
+    const char* url;
+    bool is_third_party;
+  } kTestCases[] = {
+      {"https://sub.example.com", false},
+      {"https://subexample.com", true},
+      {"https://sub.subexample.com", true},
+      {"https://sub.sub.example.com", false},
+      {"https://xample.com", true},
+      {"https://sub.xample.com", true},
+      {"data:text/plain,example.com", true},
+  };
+
+  FirstPartyOrigin first_party(url::Origin(GURL("https://example.com")));
+  for (const auto& test_case : kTestCases) {
+    GURL url(test_case.url);
+    EXPECT_EQ(test_case.is_third_party,
+              FirstPartyOrigin::IsThirdParty(url, first_party.origin()));
+    EXPECT_EQ(test_case.is_third_party, first_party.IsThirdParty(url));
+  }
+}
+
+TEST(FirstPartyOriginTest, EmptyHostUrls) {
+  const char* const kUrls[] = {
+      "data:text/plain,example.com", "data:text/plain,another.example.com",
+      "data:text/plain;base64,ABACABA",
+  };
+
+  FirstPartyOrigin first_party(url::Origin(GURL("https://example.com")));
+  for (const auto& url_string : kUrls) {
+    GURL url(url_string);
+    EXPECT_TRUE(FirstPartyOrigin::IsThirdParty(url, first_party.origin()));
+    EXPECT_TRUE(first_party.IsThirdParty(url));
+  }
+}
+
+}  // namespace subresource_filter