| Index: components/data_reduction_proxy/core/browser/data_reduction_proxy_usage_stats_unittest.cc
|
| diff --git a/components/data_reduction_proxy/core/browser/data_reduction_proxy_usage_stats_unittest.cc b/components/data_reduction_proxy/core/browser/data_reduction_proxy_usage_stats_unittest.cc
|
| index 338bdec6fa6f42a4ea078b0e655b14ada82045ce..8278b5107b4fb16e480ec002c01be48f734a8d45 100644
|
| --- a/components/data_reduction_proxy/core/browser/data_reduction_proxy_usage_stats_unittest.cc
|
| +++ b/components/data_reduction_proxy/core/browser/data_reduction_proxy_usage_stats_unittest.cc
|
| @@ -50,27 +50,6 @@ namespace data_reduction_proxy {
|
|
|
| namespace {
|
|
|
| -class DataReductionProxyParamsMock :
|
| - public TestDataReductionProxyParams {
|
| - public:
|
| - DataReductionProxyParamsMock() :
|
| - TestDataReductionProxyParams(
|
| - 0, TestDataReductionProxyParams::HAS_EVERYTHING) {}
|
| - virtual ~DataReductionProxyParamsMock() {}
|
| -
|
| - MOCK_CONST_METHOD2(
|
| - IsDataReductionProxy,
|
| - bool(const net::HostPortPair& host_port_pair,
|
| - DataReductionProxyTypeInfo* proxy_info));
|
| - MOCK_CONST_METHOD2(
|
| - WasDataReductionProxyUsed,
|
| - bool(const net::URLRequest*,
|
| - DataReductionProxyTypeInfo* proxy_info));
|
| -
|
| - private:
|
| - DISALLOW_COPY_AND_ASSIGN(DataReductionProxyParamsMock);
|
| -};
|
| -
|
| const std::string kBody = "hello";
|
| const std::string kNextBody = "hello again";
|
| const std::string kErrorBody = "bad";
|
| @@ -99,13 +78,10 @@ class DataReductionProxyUsageStatsTest : public testing::Test {
|
| TestDataReductionProxyParams::HAS_EVERYTHING &
|
| ~TestDataReductionProxyParams::HAS_DEV_ORIGIN &
|
| ~TestDataReductionProxyParams::HAS_DEV_FALLBACK_ORIGIN)
|
| + .WithMockConfig()
|
| .Build();
|
| mock_url_request_ = context_.CreateRequest(GURL(), net::IDLE, &delegate_,
|
| NULL);
|
| - scoped_ptr<DataReductionProxyParamsMock> mock_params =
|
| - make_scoped_ptr(new DataReductionProxyParamsMock());
|
| - mock_params_ = mock_params.get();
|
| - test_context_->config()->ResetParamsForTest(mock_params.Pass());
|
| }
|
|
|
| scoped_ptr<net::URLRequest> CreateURLRequestWithResponseHeaders(
|
| @@ -147,8 +123,8 @@ class DataReductionProxyUsageStatsTest : public testing::Test {
|
| return mock_url_request_.get();
|
| }
|
|
|
| - DataReductionProxyParamsMock* params() const {
|
| - return mock_params_;
|
| + MockDataReductionProxyConfig* config() const {
|
| + return test_context_->mock_config();
|
| }
|
|
|
| void RunUntilIdle() {
|
| @@ -158,7 +134,6 @@ class DataReductionProxyUsageStatsTest : public testing::Test {
|
| private:
|
| net::TestURLRequestContext context_;
|
| net::TestDelegate delegate_;
|
| - DataReductionProxyParamsMock* mock_params_;
|
| scoped_ptr<net::URLRequest> mock_url_request_;
|
| // |test_job_interceptor_| is owned by |test_job_factory_|.
|
| net::TestJobInterceptor* test_job_interceptor_;
|
| @@ -200,10 +175,10 @@ TEST_F(DataReductionProxyUsageStatsTest, IsDataReductionProxyUnreachable) {
|
| for (size_t i = 0; i < arraysize(test_cases); ++i) {
|
| TestCase test_case = test_cases[i];
|
|
|
| - EXPECT_CALL(*params(), IsDataReductionProxy(testing::_, testing::_))
|
| + EXPECT_CALL(*config(), IsDataReductionProxy(testing::_, testing::_))
|
| .WillRepeatedly(testing::Return(
|
| test_case.fallback_proxy_server_is_data_reduction_proxy));
|
| - EXPECT_CALL(*params(),
|
| + EXPECT_CALL(*config(),
|
| WasDataReductionProxyUsed(url_request(), testing::_))
|
| .WillRepeatedly(testing::Return(test_case.was_proxy_used));
|
|
|
| @@ -222,9 +197,9 @@ TEST_F(DataReductionProxyUsageStatsTest, ProxyUnreachableThenReachable) {
|
| net::ProxyServer fallback_proxy_server =
|
| net::ProxyServer::FromURI("foo.com", net::ProxyServer::SCHEME_HTTP);
|
| scoped_ptr<DataReductionProxyUsageStats> usage_stats = BuildUsageStats();
|
| - EXPECT_CALL(*params(), IsDataReductionProxy(testing::_, testing::_))
|
| + EXPECT_CALL(*config(), IsDataReductionProxy(testing::_, testing::_))
|
| .WillOnce(testing::Return(true));
|
| - EXPECT_CALL(*params(),
|
| + EXPECT_CALL(*config(),
|
| WasDataReductionProxyUsed(url_request(), testing::_))
|
| .WillOnce(testing::Return(true));
|
|
|
| @@ -244,10 +219,10 @@ TEST_F(DataReductionProxyUsageStatsTest, ProxyReachableThenUnreachable) {
|
| net::ProxyServer fallback_proxy_server =
|
| net::ProxyServer::FromURI("foo.com", net::ProxyServer::SCHEME_HTTP);
|
| scoped_ptr<DataReductionProxyUsageStats> usage_stats = BuildUsageStats();
|
| - EXPECT_CALL(*params(),
|
| + EXPECT_CALL(*config(),
|
| WasDataReductionProxyUsed(url_request(), testing::_))
|
| .WillOnce(testing::Return(true));
|
| - EXPECT_CALL(*params(), IsDataReductionProxy(testing::_, testing::_))
|
| + EXPECT_CALL(*config(), IsDataReductionProxy(testing::_, testing::_))
|
| .WillRepeatedly(testing::Return(true));
|
|
|
| // Proxy succeeds.
|
| @@ -456,7 +431,7 @@ TEST_F(DataReductionProxyUsageStatsTest, RecordMissingViaHeaderBytes) {
|
| raw_headers));
|
| fake_request->set_received_response_content_length(kResponseContentLength);
|
|
|
| - EXPECT_CALL(*params(),
|
| + EXPECT_CALL(*config(),
|
| WasDataReductionProxyUsed(fake_request.get(), testing::_))
|
| .WillRepeatedly(Return(test_cases[i].was_proxy_used));
|
|
|
| @@ -537,7 +512,7 @@ TEST_F(DataReductionProxyUsageStatsTest, RequestCompletionErrorCodes) {
|
|
|
| DataReductionProxyTypeInfo proxy_info;
|
| proxy_info.is_fallback = test_cases[i].is_fallback;
|
| - EXPECT_CALL(*params(), WasDataReductionProxyUsed(fake_request.get(),
|
| + EXPECT_CALL(*config(), WasDataReductionProxyUsed(fake_request.get(),
|
| testing::NotNull()))
|
| .WillRepeatedly(testing::DoAll(testing::SetArgPointee<1>(proxy_info),
|
| Return(test_cases[i].was_proxy_used)));
|
| @@ -615,7 +590,7 @@ class DataReductionProxyUsageStatsEndToEndTest : public testing::Test {
|
| context_.set_job_factory(job_factory_.get());
|
|
|
| test_context_->configurator()->Enable(false, true,
|
| - params()->origin().ToURI(),
|
| + config()->Origin().ToURI(),
|
| std::string(), std::string());
|
| test_context_->RunUntilIdle();
|
| }
|
| @@ -680,8 +655,8 @@ class DataReductionProxyUsageStatsEndToEndTest : public testing::Test {
|
| return test_context_->settings();
|
| }
|
|
|
| - const DataReductionProxyParams* params() const {
|
| - return test_context_->config()->test_params();
|
| + DataReductionProxyConfig* config() const {
|
| + return test_context_->config();
|
| }
|
|
|
| void ClearBadProxies() {
|
| @@ -944,7 +919,7 @@ TEST_F(DataReductionProxyUsageStatsEndToEndTest, BypassedBytesNetErrorOther) {
|
| // Make the data reduction proxy host fail to resolve.
|
| scoped_ptr<net::MockHostResolver> host_resolver(new net::MockHostResolver());
|
| host_resolver->rules()->AddSimulatedFailure(
|
| - params()->origin().host_port_pair().host());
|
| + config()->Origin().host_port_pair().host());
|
| set_host_resolver(host_resolver.get());
|
| InitializeContext();
|
|
|
|
|