Add probe to warm data reduction proxy

components/data_reduction_proxy/browser/data_reduction_proxy_settings_test_utils.h

 // 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.
 
 #ifndef COMPONENTS_DATA_REDUCTION_PROXY_BROWSER_DATA_REDUCTION_PROXY_SETTINGS_TEST_UTILS_H_
 #define COMPONENTS_DATA_REDUCTION_PROXY_BROWSER_DATA_REDUCTION_PROXY_SETTINGS_TEST_UTILS_H_
 
 
 #include "base/prefs/testing_pref_service.h"
 #include "components/data_reduction_proxy/browser/data_reduction_proxy_configurator.h"
@@ skipping 43 matching lines @@
 template <class C>
 class MockDataReductionProxySettings : public C {
  public:
   MockDataReductionProxySettings<C>() : DataReductionProxySettings(
       new DataReductionProxyParams(
           DataReductionProxyParams::kAllowed |
           DataReductionProxyParams::kFallbackAllowed |
           DataReductionProxyParams::kPromoAllowed)) {}
   MockDataReductionProxySettings<C>(int flags)
       : C(new DataReductionProxyParams(flags)) {}
-  MOCK_METHOD0(GetURLFetcher, net::URLFetcher*());
+  MOCK_METHOD0(GetURLFetcherForAvailabilityCheck, net::URLFetcher*());
+  MOCK_METHOD0(GetURLFetcherForWarmup, net::URLFetcher*());
   MOCK_METHOD0(GetOriginalProfilePrefs, PrefService*());
   MOCK_METHOD0(GetLocalStatePrefs, PrefService*());
   MOCK_METHOD3(LogProxyState, void(
       bool enabled, bool restricted, bool at_startup));
   MOCK_METHOD1(RecordProbeURLFetchResult,
                void(ProbeURLFetchResult result));
   MOCK_METHOD1(RecordStartupState,
                void(ProxyStartupState state));
 
   // SetProxyConfigs should always call LogProxyState exactly once.
@@ skipping 25 matching lines @@
                                         bool fallback_allowed,
                                         bool alt_allowed,
                                         bool promo_allowed);
   virtual void ResetSettings(bool allowed,
                              bool fallback_allowed,
                              bool alt_allowed,
                              bool promo_allowed) = 0;
 
   template <class C> void SetProbeResult(
       const std::string& test_url,
+      const std::string& warmup_test_url,
       const std::string& response,
       ProbeURLFetchResult state,
       bool success,
       int expected_calls);
   virtual void SetProbeResult(const std::string& test_url,
+                              const std::string& warmup_test_url,
                               const std::string& response,
                               ProbeURLFetchResult result,
                               bool success,
                               int expected_calls) = 0;
 
   void CheckProxyConfigs(bool expected_enabled,
                          bool expected_restricted,
                          bool expected_fallback_restricted);
   void CheckProbe(bool initially_enabled,
                   const std::string& probe_url,
+                  const std::string& warmup_url,
                   const std::string& response,
                   bool request_success,
                   bool expected_enabled,
                   bool expected_restricted,
                   bool expected_fallback_restricted);
   void CheckProbeOnIPChange(const std::string& probe_url,
+                            const std::string& warmup_url,
                             const std::string& response,
                             bool request_success,
                             bool expected_enabled,
                             bool expected_fallback_restricted);
   void CheckOnPrefChange(bool enabled, bool expected_enabled, bool managed);
   void CheckInitDataReductionProxy(bool enabled_at_startup);
 
   TestingPrefServiceSimple pref_service_;
   scoped_ptr<DataReductionProxySettings> settings_;
   base::Time last_update_time_;
 };
 
 // Test implementations should be subclasses of an instantiation of this
 // class parameterized for whatever DataReductionProxySettings class
 // is being tested.
 template <class C>
 class ConcreteDataReductionProxySettingsTest
     : public DataReductionProxySettingsTestBase {
  public:
   typedef MockDataReductionProxySettings<C> MockSettings;
   virtual void ResetSettings(bool allowed,
                              bool fallback_allowed,
                              bool alt_allowed,
                              bool promo_allowed) OVERRIDE {
     return DataReductionProxySettingsTestBase::ResetSettings<C>(
         allowed, fallback_allowed, alt_allowed, promo_allowed);
   }
 
   virtual void SetProbeResult(const std::string& test_url,
+                              const std::string& warmup_test_url,
                               const std::string& response,
                               ProbeURLFetchResult result,
                               bool success,
                               int expected_calls) OVERRIDE {
     return DataReductionProxySettingsTestBase::SetProbeResult<C>(
-  test_url, response, result, success, expected_calls);
+        test_url,
+        warmup_test_url,
+        response,
+        result,
+        success,
+        expected_calls);
   }
 };
 
 }  // namespace data_reduction_proxy
 
 #endif  // COMPONENTS_DATA_REDUCTION_PROXY_BROWSER_DATA_REDUCTION_PROXY_SETTINGS_TEST_UTILS_H_