Add a function to add extra headers from NavigationThrottle

content/browser/loader/navigation_url_loader_unittest.cc

 // Copyright 2014 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 "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/run_loop.h"
 #include "content/browser/frame_host/navigation_request_info.h"
@@ skipping 19 matching lines @@
 #include "net/url_request/redirect_info.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_job_factory_impl.h"
 #include "net/url_request/url_request_test_job.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
 
+using ExtraHeadersList = std::vector<std::pair<std::string, std::string> >;
+
 namespace {
 
 class StreamProtocolHandler
     : public net::URLRequestJobFactory::ProtocolHandler {
  public:
   StreamProtocolHandler(StreamRegistry* registry) : registry_(registry) {}
 
   // net::URLRequestJobFactory::ProtocolHandler implementation.
   net::URLRequestJob* MaybeCreateJob(
       net::URLRequest* request,
@@ skipping 208 matching lines @@
   delegate.WaitForRequestRedirected();
   EXPECT_EQ(net::URLRequestTestJob::test_url_2(),
             delegate.redirect_info().new_url);
   EXPECT_EQ("GET", delegate.redirect_info().new_method);
   EXPECT_EQ(net::URLRequestTestJob::test_url_2(),
             delegate.redirect_info().new_first_party_for_cookies);
   EXPECT_EQ(302, delegate.redirect_response()->head.headers->response_code());
   EXPECT_EQ(1, delegate.on_request_handled_counter());
 
   // Wait for the response to complete.
-  loader->FollowRedirect();
+  loader->FollowRedirect(ExtraHeadersList());
   delegate.WaitForResponseStarted();
 
   // Check the response is correct.
   EXPECT_EQ("text/html", delegate.response()->head.mime_type);
   EXPECT_EQ(200, delegate.response()->head.headers->response_code());
 
   // Release the body and check it is correct.
   EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage());
   EXPECT_EQ(net::URLRequestTestJob::test_data_2(),
             FetchURL(delegate.body()->GetURL()));
@@ skipping 26 matching lines @@
   TestNavigationURLLoaderDelegate delegate;
   scoped_ptr<NavigationURLLoader> loader =
       MakeTestLoader(net::URLRequestTestJob::test_url_redirect_to_url_2(),
                      &delegate);
 
   // Wait for the request to redirect.
   delegate.WaitForRequestRedirected();
 
   // In the same event loop iteration, follow the redirect (allowing the
   // response to go through) and destroy the loader.
-  loader->FollowRedirect();
+  loader->FollowRedirect(ExtraHeadersList());
   loader.reset();
 
   // Verify the URLRequestTestJob no longer has anything paused and that no
   // response body was received.
   base::RunLoop().RunUntilIdle();
   EXPECT_FALSE(net::URLRequestTestJob::ProcessOnePendingMessage());
   EXPECT_FALSE(delegate.body());
 }
 
 // Tests that the loader may be canceled by context.
@@ skipping 69 matching lines @@
 
   // Release the body.
   delegate.ReleaseBody();
   base::RunLoop().RunUntilIdle();
 
   // Verify that URLRequestTestJob no longer has anything paused.
   EXPECT_FALSE(net::URLRequestTestJob::ProcessOnePendingMessage());
 }
 
 }  // namespace content