Add MaybeInterceptRedirect/Response to URLRequestInterceptor

chrome/browser/errorpage_browsertest.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 "base/bind.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
@@ skipping 141 matching lines @@
 // all requests after that, keeping count of failures and successes.
 class FailFirstNRequestsInterceptor : public net::URLRequestInterceptor {
  public:
   explicit FailFirstNRequestsInterceptor(int requests_to_fail)
       : requests_(0), failures_(0), requests_to_fail_(requests_to_fail) {}
   ~FailFirstNRequestsInterceptor() override {}
 
   // net::URLRequestInterceptor implementation
   net::URLRequestJob* MaybeInterceptRequest(
       net::URLRequest* request,
-      net::NetworkDelegate* network_delegate) const override {
+      net::NetworkDelegate* network_delegate) override {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
     requests_++;
     if (failures_ < requests_to_fail_) {
       failures_++;
       // Note: net::ERR_CONNECTION_RESET does not summon the Link Doctor; see
       // NetErrorHelperCore::GetErrorPageURL.
       return new URLRequestFailedJob(request,
                                      network_delegate,
                                      net::ERR_CONNECTION_RESET);
     } else {
       return new URLRequestTestJob(request, network_delegate,
                                    URLRequestTestJob::test_headers(),
                                    URLRequestTestJob::test_data_1(),
                                    true);
     }
   }
 
+  net::URLRequestJob* MaybeInterceptRedirect(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate,
+      const GURL& location) override {
+    return NULL;
+  }
+
+  net::URLRequestJob* MaybeInterceptResponse(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate) override {
+    return NULL;
+  }
+
   int requests() const { return requests_; }
   int failures() const { return failures_; }
 
  private:
   // These are mutable because MaybeCreateJob is const but we want this state
   // for testing.
   mutable int requests_;
   mutable int failures_;
   int requests_to_fail_;
 
   DISALLOW_COPY_AND_ASSIGN(FailFirstNRequestsInterceptor);
 };
 
 // An interceptor that serves LinkDoctor responses.  It also allows waiting
 // until a certain number of requests have been sent.
 // TODO(mmenke):  Wait until responses have been received instead.
 class LinkDoctorInterceptor : public net::URLRequestInterceptor {
  public:
   LinkDoctorInterceptor() : num_requests_(0),
                             requests_to_wait_for_(-1),
                             weak_factory_(this) {
   }
 
   ~LinkDoctorInterceptor() override {}
 
   // net::URLRequestInterceptor implementation
   net::URLRequestJob* MaybeInterceptRequest(
       net::URLRequest* request,
-      net::NetworkDelegate* network_delegate) const override {
+      net::NetworkDelegate* network_delegate) override {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 
     BrowserThread::PostTask(
         BrowserThread::UI, FROM_HERE,
         base::Bind(&LinkDoctorInterceptor::RequestCreated,
                    weak_factory_.GetWeakPtr()));
 
     base::FilePath root_http;
     PathService::Get(chrome::DIR_TEST_DATA, &root_http);
     return new net::URLRequestMockHTTPJob(
         request,
         network_delegate,
         root_http.AppendASCII("mock-link-doctor.json"),
         BrowserThread::GetBlockingPool()->GetTaskRunnerWithShutdownBehavior(
             base::SequencedWorkerPool::SKIP_ON_SHUTDOWN));
   }
 
+  net::URLRequestJob* MaybeInterceptRedirect(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate,
+      const GURL& location) override {
+    return NULL;
+  }
+
+  net::URLRequestJob* MaybeInterceptResponse(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate) override {
+    return NULL;
+  }
+
   void WaitForRequests(int requests_to_wait_for) {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
     DCHECK_EQ(-1, requests_to_wait_for_);
     DCHECK(!run_loop_);
 
     if (requests_to_wait_for >= num_requests_)
       return;
 
     requests_to_wait_for_ = requests_to_wait_for;
     run_loop_.reset(new base::RunLoop());
@@ skipping 698 matching lines @@
 
 // Interceptor that fails all requests with net::ERR_ADDRESS_UNREACHABLE.
 class AddressUnreachableInterceptor : public net::URLRequestInterceptor {
  public:
   AddressUnreachableInterceptor() {}
   ~AddressUnreachableInterceptor() override {}
 
   // net::URLRequestInterceptor:
   net::URLRequestJob* MaybeInterceptRequest(
       net::URLRequest* request,
-      net::NetworkDelegate* network_delegate) const override {
+      net::NetworkDelegate* network_delegate) override {
     return new URLRequestFailedJob(request,
                                    network_delegate,
                                    net::ERR_ADDRESS_UNREACHABLE);
   }
 
+  net::URLRequestJob* MaybeInterceptResponse(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate) override {
+    return NULL;
+  }
+
+  net::URLRequestJob* MaybeInterceptRedirect(
+      net::URLRequest* request,
+      net::NetworkDelegate* network_delegate,
+      const GURL& location) override {
+    return NULL;
+  }
+
  private:
   DISALLOW_COPY_AND_ASSIGN(AddressUnreachableInterceptor);
 };
 
 // A test fixture that returns ERR_ADDRESS_UNREACHABLE for all navigation
 // correction requests.  ERR_NAME_NOT_RESOLVED is more typical, but need to use
 // a different error for the correction service and the original page to
 // validate the right page is being displayed.
 class ErrorPageNavigationCorrectionsFailTest : public ErrorPageTest {
  public:
@@ skipping 138 matching lines @@
       browser(),
       URLRequestFailedJob::GetMockHttpUrlForHostname(net::ERR_UNSAFE_PORT,
                                                      kHostname),
       1);
 
   ToggleHelpBox(browser());
   EXPECT_TRUE(IsDisplayingText(browser(), kHostnameJSUnicode));
 }
 
 }  // namespace