Data Reduction Proxy Interstitials

components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h

Index: components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h
diff --git a/components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h b/components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h
index 713cf3f8cb56e46703bf3452405b8143ad93540b..be897a64466fd3f045dae21b6da6d1c28f40d086 100644
--- a/components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h
+++ b/components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h
@@ -29,6 +29,22 @@ class TestDataReductionProxyParams : public DataReductionProxyParams {
                                unsigned int has_definitions);
   bool init_result() const;
 
+  // Overrides from net::DataReductionProxyParams.
+  bool IsDataReductionProxy(
+      const net::HostPortPair& host_port_pair,
+      DataReductionProxyTypeInfo* proxy_info) const override;
+  bool IsBypassedByDataReductionProxyLocalRules(
+      const net::URLRequest& request,
+      const net::ProxyConfig& data_reduction_proxy_config) const override;
+  bool AreDataReductionProxiesBypassed(
+      const net::URLRequest& request,
+      base::TimeDelta* min_retry_delay) const override;
+
+  // Once called, the mocked method will repeatedly return |return_value|.
+  void MockIsDataReductionProxy(bool return_value);
+  void MockIsBypassedByDataReductionProxyLocalRules(bool return_value);
+  void MockAreDataReductionProxiesBypassed(bool return_value);
+
   // Test values to replace the values specified in preprocessor defines.
   static std::string DefaultDevOrigin();
   static std::string DefaultDevFallbackOrigin();
@@ -72,6 +88,13 @@ class TestDataReductionProxyParams : public DataReductionProxyParams {
 
   unsigned int has_definitions_;
   bool init_result_;
+
+  bool mock_is_data_reduction_proxy_;
+  bool mock_is_bypassed_by_data_reduction_proxy_local_rules_;
+  bool mock_are_data_reduction_proxies_bypassed_;
+  bool is_data_reduction_proxy_return_value_;
+  bool is_bypassed_by_data_reduction_proxy_local_rules_return_value_;
+  bool are_data_reduction_proxies_bypassed_return_value_;
 };
 }  // namespace data_reduction_proxy
 #endif  // COMPONENTS_DATA_REDUCTION_PROXY_CORE_BROWSER_DATA_REDUCTION_PROXY_PARAMS_TEST_UTILS_H_