Support redirectUrl at onHeadersReceived in WebRequest / DWR API

chrome/browser/extensions/api/declarative_webrequest/webrequest_action_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_action.h"
 
 #include "base/files/file_path.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 261 matching lines @@
 #undef CallPerlToRe2Style
 }
 
 TEST_F(WebRequestActionWithThreadsTest, PermissionsToRedirect) {
   const char kAction[] =
       "[{"
       " \"instanceType\": \"declarativeWebRequest.RedirectRequest\","
       " \"redirectUrl\": \"http://www.foobar.com\""
       "}]";
   CheckActionNeedsAllUrls(kAction, ON_BEFORE_REQUEST);
+  CheckActionNeedsAllUrls(kAction, ON_HEADERS_RECEIVED);
 }
 
 TEST_F(WebRequestActionWithThreadsTest, PermissionsToRedirectByRegEx) {
   const char kAction[] =
       "[{"
       " \"instanceType\": \"declarativeWebRequest.RedirectByRegEx\","
       " \"from\": \".*\","
       " \"to\": \"http://www.foobar.com\""
       "}]";
   CheckActionNeedsAllUrls(kAction, ON_BEFORE_REQUEST);
@@ skipping 143 matching lines @@
       "[{"
       " \"instanceType\": \"declarativeWebRequest.RedirectToTransparentImage\""
       "}]";
   scoped_ptr<WebRequestActionSet> action_set(CreateSetOfActions(kAction));
 
   // Redirecting to transparent images works without full host permissions.
   EXPECT_TRUE(ActionWorksOnRequest("http://test.org",
                                    extension_->id(),
                                    action_set.get(),
                                    ON_BEFORE_REQUEST));
+  EXPECT_TRUE(ActionWorksOnRequest("http://test.org",
+                                   extension_->id(),
+                                   action_set.get(),
+                                   ON_HEADERS_RECEIVED));
 }
 
 TEST_F(WebRequestActionWithThreadsTest, PermissionsToRedirectToEmptyDocument) {
   const char kAction[] =
       "[{"
       " \"instanceType\": \"declarativeWebRequest.RedirectToEmptyDocument\""
       "}]";
   scoped_ptr<WebRequestActionSet> action_set(CreateSetOfActions(kAction));
 
   // Redirecting to the empty document works without full host permissions.
   EXPECT_TRUE(ActionWorksOnRequest("http://test.org",
                                    extension_->id(),
                                    action_set.get(),
                                    ON_BEFORE_REQUEST));
+  EXPECT_TRUE(ActionWorksOnRequest("http://test.org",
+                                   extension_->id(),
+                                   action_set.get(),
+                                   ON_HEADERS_RECEIVED));
 }
 
 TEST_F(WebRequestActionWithThreadsTest, PermissionsToIgnore) {
   const char kAction[] =
       "[{"
       " \"instanceType\": \"declarativeWebRequest.IgnoreRules\","
       " \"lowerPriorityThan\": 123,"
       " \"hasTag\": \"some_tag\""
       "}]";
   scoped_ptr<WebRequestActionSet> action_set(CreateSetOfActions(kAction));
@@ skipping 103 matching lines @@
   for (WebRequestActionSet::Actions::const_iterator it =
            action_set->actions().begin();
        it != action_set->actions().end();
        ++it) {
     EXPECT_EQ(kExpectedNames[index], (*it)->GetName());
     ++index;
   }
 }
 
 }  // namespace extensions