Adding thirdParty property to RequestMatcher

chrome/browser/extensions/api/declarative_webrequest/webrequest_condition_attribute_unittest.cc

 // Copyright (c) 2012 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 "chrome/browser/extensions/api/declarative_webrequest/webrequest_condition_attribute.h"
 
 #include "base/basictypes.h"
 #include "base/file_path.h"
 #include "base/message_loop.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_constants.h"
 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rule.h"
 #include "content/public/browser/resource_request_info.h"
 #include "net/url_request/url_request_test_util.h"
 #include "net/test/test_server.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::DictionaryValue;
+using base::FundamentalValue;
 using base::ListValue;
 using base::StringValue;
 using base::Value;
 
 namespace {
 const char kUnknownConditionName[] = "unknownType";
 }  // namespace
 
 namespace extensions {
 
@@ skipping 126 matching lines @@
   scoped_ptr<WebRequestConditionAttribute> attribute_unexcluded =
       WebRequestConditionAttribute::Create(
           keys::kExcludeContentTypeKey, &content_types, &error);
   EXPECT_EQ("", error);
   ASSERT_TRUE(attribute_unexcluded.get());
   EXPECT_TRUE(attribute_unexcluded->IsFulfilled(
       WebRequestRule::RequestData(&url_request, ON_HEADERS_RECEIVED,
                                   url_request.response_headers())));
 }
 
+// Testing WebRequestConditionAttributeThirdParty.
+TEST(WebRequestConditionAttributeTest, ThirdParty) {
+  // Necessary for TestURLRequest.
+  MessageLoop message_loop(MessageLoop::TYPE_IO);
+
+  std::string error;
+  const FundamentalValue value_true(true);
+  // This attribute matches only third party requests.
+  scoped_ptr<WebRequestConditionAttribute> third_party_attribute =
+      WebRequestConditionAttribute::Create(keys::kThirdPartyKey,
+                                           &value_true,
+                                           &error);
+  ASSERT_EQ("", error);
+  ASSERT_TRUE(third_party_attribute.get() != NULL);
+  const FundamentalValue value_false(false);
+  // This attribute matches only first party requests.
+  scoped_ptr<WebRequestConditionAttribute> first_party_attribute =
+      WebRequestConditionAttribute::Create(keys::kThirdPartyKey,
+                                           &value_false,
+                                           &error);
+  ASSERT_EQ("", error);
+  ASSERT_TRUE(first_party_attribute.get() != NULL);
+
+  const GURL url_empty;
+  const GURL url_a("http://a.com");
+  const GURL url_b("http://b.com");
+  TestURLRequestContext context;
+  TestDelegate delegate;
+  TestURLRequest url_request(url_a, &delegate, &context);
+
+  static const RequestStage request_stages[] = {
+    ON_BEFORE_REQUEST,
+    ON_BEFORE_SEND_HEADERS,
+    ON_SEND_HEADERS,
+    ON_HEADERS_RECEIVED,
+    ON_AUTH_REQUIRED,
+    ON_BEFORE_REDIRECT,
+    ON_RESPONSE_STARTED,
+    ON_COMPLETED,
+    ON_ERROR
+  };
+
+  for (size_t i = 0; i < arraysize(request_stages); ++i) {
+    url_request.set_first_party_for_cookies(url_empty);
+    EXPECT_FALSE(third_party_attribute->IsFulfilled(WebRequestRule::RequestData(
+        &url_request, request_stages[i])));
+    EXPECT_TRUE(first_party_attribute->IsFulfilled(WebRequestRule::RequestData(
+        &url_request, request_stages[i])));
+
+    url_request.set_first_party_for_cookies(url_b);
+    EXPECT_TRUE(third_party_attribute->IsFulfilled(WebRequestRule::RequestData(
+        &url_request, request_stages[i])));
+    EXPECT_FALSE(first_party_attribute->IsFulfilled(WebRequestRule::RequestData(
+        &url_request, request_stages[i])));
+
+    url_request.set_first_party_for_cookies(url_a);
+    EXPECT_FALSE(third_party_attribute->IsFulfilled(WebRequestRule::RequestData(
+        &url_request, request_stages[i])));
+    EXPECT_TRUE(first_party_attribute->IsFulfilled(WebRequestRule::RequestData(
+        &url_request, request_stages[i])));
+  }
+}
+
 namespace {
 
 // Builds a vector of vectors of string pointers from an array of strings.
 // |array| is in fact a sequence of arrays. The array |sizes| captures the sizes
 // of all parts of |array|, and |size| is the length of |sizes| itself.
 // Example (this is pseudo-code, not C++):
 // array = { "a", "b", "c", "d", "e", "f" }
 // sizes = { 2, 0, 4 }
 // size = 3
 // results in out == { {&"a", &"b"}, {}, {&"c", &"d", &"e", &"f"} }
@@ skipping 256 matching lines @@
     keys::kNameEqualsKey, "custom-header-b",
     keys::kValueEqualsKey, "valueB"
   };
   const size_t kExistingSize[] = { arraysize(kExisting) };
   GetArrayAsVector(kExisting, kExistingSize, 1u, &tests);
   MatchAndCheck(tests, keys::kExcludeResponseHeadersKey, &url_request, &result);
   EXPECT_FALSE(result);
 }
 
 }  // namespace extensions