Index: components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h |
diff --git a/components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h b/components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h |
index 004a3a5f2be0b8d021b397197829febaf91f2e53..589856913fa09f421bd481105fa1f8adb8ab7042 100644 |
--- a/components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h |
+++ b/components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h |
@@ -43,6 +43,7 @@ namespace data_reduction_proxy { |
class DataReductionProxyConfigurator; |
class DataReductionProxyEventCreator; |
+class DataReductionProxyExperimentsStats; |
class DataReductionProxyMutableConfigValues; |
class DataReductionProxyRequestOptions; |
class DataReductionProxySettings; |
@@ -150,6 +151,7 @@ class MockDataReductionProxyService : public DataReductionProxyService { |
MockDataReductionProxyService( |
scoped_ptr<DataReductionProxyCompressionStats> compression_stats, |
DataReductionProxySettings* settings, |
+ PrefService* prefs, |
net::URLRequestContextGetter* request_context, |
scoped_refptr<base::SingleThreadTaskRunner> io_task_runner); |
~MockDataReductionProxyService() override; |
@@ -170,9 +172,13 @@ class TestDataReductionProxyIOData : public DataReductionProxyIOData { |
scoped_ptr<DataReductionProxyRequestOptions> request_options, |
scoped_ptr<DataReductionProxyConfigurator> configurator, |
scoped_ptr<DataReductionProxyConfigServiceClient> config_client, |
+ scoped_ptr<DataReductionProxyExperimentsStats> experiments_stats, |
bool enabled); |
~TestDataReductionProxyIOData() override; |
+ void SetDataReductionProxyService(base::WeakPtr<DataReductionProxyService> |
+ data_reduction_proxy_service) override; |
+ |
DataReductionProxyConfigurator* configurator() const { |
return configurator_.get(); |
} |
@@ -189,6 +195,10 @@ class TestDataReductionProxyIOData : public DataReductionProxyIOData { |
base::WeakPtr<DataReductionProxyIOData> GetWeakPtr() { |
return weak_factory_.GetWeakPtr(); |
} |
+ |
+ private: |
+ // Allowed SetDataReductionProxyService to be re-entrant. |
+ bool service_set_; |
}; |
// Builds a test version of the Data Reduction Proxy stack for use in tests. |