Move handling of invalid referrer to the network delegate

net/url_request/url_request_test_util.h

 // 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.
 
 #ifndef NET_URL_REQUEST_URL_REQUEST_TEST_UTIL_H_
 #define NET_URL_REQUEST_URL_REQUEST_TEST_UTIL_H_
 
 #include <stdlib.h>
 
 #include <map>
@@ skipping 251 matching lines @@
   int blocked_get_cookies_count() const { return blocked_get_cookies_count_; }
   int blocked_set_cookie_count() const { return blocked_set_cookie_count_; }
   int set_cookie_count() const { return set_cookie_count_; }
 
   void set_can_access_files(bool val) { can_access_files_ = val; }
   bool can_access_files() const { return can_access_files_; }
 
   void set_can_throttle_requests(bool val) { can_throttle_requests_ = val; }
   bool can_throttle_requests() const { return can_throttle_requests_; }
 
+  void set_can_correct_invalid_referrer_header(bool val) {
+    can_correct_invalid_referrer_header_ = val;
+  }
+
   int observed_before_proxy_headers_sent_callbacks() const {
     return observed_before_proxy_headers_sent_callbacks_;
   }
 
   // Last observed proxy in proxy header sent callback.
   HostPortPair last_observed_proxy() {
     return last_observed_proxy_;
   }
 
  protected:
@@ skipping 35 matching lines @@
   virtual bool OnCanSetCookie(const URLRequest& request,
                               const std::string& cookie_line,
                               CookieOptions* options) OVERRIDE;
   virtual bool OnCanAccessFile(const URLRequest& request,
                                const base::FilePath& path) const OVERRIDE;
   virtual bool OnCanThrottleRequest(
       const URLRequest& request) const OVERRIDE;
   virtual int OnBeforeSocketStreamConnect(
       SocketStream* stream,
       const CompletionCallback& callback) OVERRIDE;
+  virtual bool OnCanCorrectInvalidReferrerHeader(
+      const URLRequest& request,
+      const GURL& target_url,
+      const GURL& referrer_url) const OVERRIDE;
 
   void InitRequestStatesIfNew(int request_id);
 
   GURL redirect_on_headers_received_url_;
   // URL marked as safe for redirection at the onHeadersReceived stage.
   GURL allowed_unsafe_redirect_url_;
 
   int last_error_;
   int error_count_;
   int created_requests_;
@@ skipping 19 matching lines @@
   std::map<int, std::string> event_order_;
 
   LoadTimingInfo load_timing_info_before_redirect_;
   bool has_load_timing_info_before_redirect_;
 
   LoadTimingInfo load_timing_info_before_auth_;
   bool has_load_timing_info_before_auth_;
 
   bool can_access_files_;  // true by default
   bool can_throttle_requests_;  // true by default
+  bool can_correct_invalid_referrer_header_;  // true by default
 };
 
 // Overrides the host used by the LocalHttpTestServer in
 // url_request_unittest.cc . This is used by the chrome_frame_net_tests due to
 // a mysterious bug when tests execute over the loopback adapter. See
 // http://crbug.com/114369 .
 class ScopedCustomUrlRequestTestHttpHost {
  public:
   // Sets the host name to be used. The previous hostname will be stored and
   // restored upon destruction. Note that if the lifetimes of two or more
@@ skipping 26 matching lines @@
       NetworkDelegate* network_delegate) const OVERRIDE;
   void set_main_intercept_job(URLRequestJob* job);
 
  private:
   mutable URLRequestJob* main_intercept_job_;
 };
 
 }  // namespace net
 
 #endif  // NET_URL_REQUEST_URL_REQUEST_TEST_UTIL_H_