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 9cfe056d8293e20949fc97d25bc2e3b0a936bb92..7c5e09c960ea8e336deb9dbf86d721846e576afa 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 |
@@ -96,20 +96,21 @@ class MockDataReductionProxyRequestOptions |
// finely controlling the backoff timer. |
class TestDataReductionProxyConfigServiceClient |
: public DataReductionProxyConfigServiceClient { |
public: |
TestDataReductionProxyConfigServiceClient( |
std::unique_ptr<DataReductionProxyParams> params, |
DataReductionProxyRequestOptions* request_options, |
DataReductionProxyMutableConfigValues* config_values, |
DataReductionProxyConfig* config, |
DataReductionProxyEventCreator* event_creator, |
+ DataReductionProxyIOData* io_data, |
net::NetLog* net_log, |
ConfigStorer config_storer); |
~TestDataReductionProxyConfigServiceClient() override; |
using DataReductionProxyConfigServiceClient::OnIPAddressChanged; |
void SetNow(const base::Time& time); |
void SetCustomReleaseTime(const base::TimeTicks& release_time); |
@@ -185,48 +186,61 @@ class MockDataReductionProxyService : public DataReductionProxyService { |
// the constructor in favor of explicitly passing in its owned classes. This |
// permits the use of test/mock versions of those classes. |
class TestDataReductionProxyIOData : public DataReductionProxyIOData { |
public: |
TestDataReductionProxyIOData( |
const scoped_refptr<base::SingleThreadTaskRunner>& task_runner, |
std::unique_ptr<DataReductionProxyConfig> config, |
std::unique_ptr<DataReductionProxyEventCreator> event_creator, |
std::unique_ptr<DataReductionProxyRequestOptions> request_options, |
std::unique_ptr<DataReductionProxyConfigurator> configurator, |
- std::unique_ptr<DataReductionProxyConfigServiceClient> config_client, |
net::NetLog* net_log, |
bool enabled); |
~TestDataReductionProxyIOData() override; |
void SetDataReductionProxyService(base::WeakPtr<DataReductionProxyService> |
data_reduction_proxy_service) override; |
DataReductionProxyConfigurator* configurator() const { |
return configurator_.get(); |
} |
+ void set_config_client( |
+ std::unique_ptr<DataReductionProxyConfigServiceClient> config_client) { |
+ config_client_ = std::move(config_client); |
+ } |
DataReductionProxyConfigServiceClient* config_client() const { |
return config_client_.get(); |
} |
void SetSimpleURLRequestContextGetter( |
const scoped_refptr<net::URLRequestContextGetter> context_getter) { |
basic_url_request_context_getter_ = context_getter; |
} |
base::WeakPtr<DataReductionProxyIOData> GetWeakPtr() { |
return weak_factory_.GetWeakPtr(); |
} |
+ // Records the reporting fraction that was set by parsing a config. |
+ void SetPingbackReportingFraction(float pingback_reporting_fraction) override; |
+ |
+ float pingback_reporting_fraction() const { |
+ return pingback_reporting_fraction_; |
+ } |
+ |
private: |
// Allowed SetDataReductionProxyService to be re-entrant. |
bool service_set_; |
+ |
+ // Reporting fraction last set via SetPingbackReportingFraction. |
+ float pingback_reporting_fraction_; |
}; |
// Test version of |DataStore|. Uses an in memory hash map to store data. |
class TestDataStore : public data_reduction_proxy::DataStore { |
public: |
TestDataStore(); |
~TestDataStore() override; |
void InitializeOnDBThread() override {} |