Add condition attribute for MIME media types from Content-Type header.

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/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"
 
 namespace {
 const char kUnknownConditionName[] = "unknownType";
 }  // namespace
 
 namespace extensions {
 
 namespace keys = declarative_webrequest_constants;
 
@@ skipping 14 matching lines @@
   EXPECT_FALSE(error.empty());
   EXPECT_FALSE(result.get());
 
   // Test wrong data type passed
   error.clear();
   result = WebRequestConditionAttribute::Create(
       keys::kResourceTypeKey, &string_value, &error);
   EXPECT_FALSE(error.empty());
   EXPECT_FALSE(result.get());
 
+  // TODO(yoz): another one for content type

[battre, 2012/08/03 11:37:04]

want to add that to this CL? should not be a lot of effort.

[Yoyo Zhou, 2012/08/03 11:48:44]

Done; I think I meant to.

+
   // Test success
   error.clear();
   result = WebRequestConditionAttribute::Create(
       keys::kResourceTypeKey, &resource_types, &error);
   EXPECT_EQ("", error);
   ASSERT_TRUE(result.get());
   EXPECT_EQ(WebRequestConditionAttribute::CONDITION_RESOURCE_TYPE,
             result->GetType());
 }
 
@@ skipping 19 matching lines @@
       WebRequestRule::RequestData(&url_request_ok, ON_BEFORE_REQUEST)));
 
   TestURLRequest url_request_fail(
       GURL("http://www.example.com"), NULL, &context);
   content::ResourceRequestInfo::AllocateForTesting(&url_request_ok,
       ResourceType::SUB_FRAME, NULL, -1, -1);
   EXPECT_FALSE(attribute->IsFulfilled(
       WebRequestRule::RequestData(&url_request_fail, ON_BEFORE_REQUEST)));
 }
 
+TEST(WebRequestConditionAttributeTest, TestContentType) {
+  // Necessary for TestURLRequest.
+  MessageLoop message_loop(MessageLoop::TYPE_IO);
+
+  std::string error;
+  scoped_ptr<WebRequestConditionAttribute> result;
+
+  net::TestServer test_server(
+      net::TestServer::TYPE_HTTP,
+      net::TestServer::kLocalhost,
+      FilePath(FILE_PATH_LITERAL(
+          "chrome/test/data/extensions/api_test/webrequest/declarative")));
+  ASSERT_TRUE(test_server.Start());
+
+  TestURLRequestContext context;
+  TestDelegate delegate;
+  TestURLRequest url_request(test_server.GetURL("headers.html"),
+                                                &delegate, &context);
+  url_request.Start();
+  MessageLoop::current()->Run();
+
+  ListValue content_types;
+  content_types.Append(Value::CreateStringValue("text/html"));
+  scoped_ptr<WebRequestConditionAttribute> attribute_ok =
+      WebRequestConditionAttribute::Create(
+          keys::kContentTypeKey, &content_types, &error);
+  EXPECT_EQ("", error);
+  ASSERT_TRUE(attribute_ok.get());
+
+  EXPECT_FALSE(attribute_ok->IsFulfilled(
+      WebRequestRule::RequestData(&url_request, ON_BEFORE_REQUEST,
+                                  url_request.response_headers())));
+  EXPECT_TRUE(attribute_ok->IsFulfilled(
+      WebRequestRule::RequestData(&url_request, ON_HEADERS_RECEIVED,
+                                  url_request.response_headers())));
+
+  content_types.Clear();
+  content_types.Append(Value::CreateStringValue("something/invalid"));
+  scoped_ptr<WebRequestConditionAttribute> attribute_fail =
+      WebRequestConditionAttribute::Create(
+          keys::kContentTypeKey, &content_types, &error);
+  EXPECT_EQ("", error);
+  ASSERT_TRUE(attribute_fail.get());
+
+  EXPECT_FALSE(attribute_fail->IsFulfilled(
+      WebRequestRule::RequestData(&url_request, ON_HEADERS_RECEIVED,
+                                  url_request.response_headers())));
+}
+
 }  // namespace extensions