Allow URLRequests from one context to have different NetworkDelegates.

components/data_reduction_proxy/browser/data_reduction_proxy_usage_stats_unittest.cc

 // 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.
 
 #include "components/data_reduction_proxy/browser/data_reduction_proxy_usage_stats.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "net/base/request_priority.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_status.h"
@@ skipping 26 matching lines @@
 };
 
 }  // namespace
 
 namespace data_reduction_proxy {
 
 class DataReductionProxyUsageStatsTest : public testing::Test {
  public:
   DataReductionProxyUsageStatsTest()
       : loop_proxy_(MessageLoopProxy::current().get()),
-        context_(true),
-        mock_url_request_(GURL(), net::IDLE, &delegate_, &context_) {
+        context_(true) {
     context_.Init();
+    mock_url_request_ = context_.CreateRequest(GURL(), net::IDLE, &delegate_,
+                                               NULL);
   }
   // Required for MessageLoopProxy::current().
   base::MessageLoopForUI loop_;
   MessageLoopProxy* loop_proxy_;
 
  protected:
   TestURLRequestContext context_;
   TestDelegate delegate_;
   DataReductionProxyParamsMock mock_params_;
-  URLRequest mock_url_request_;
+  scoped_ptr<URLRequest> mock_url_request_;
 };
 
 TEST_F(DataReductionProxyUsageStatsTest, isDataReductionProxyUnreachable) {
   struct TestCase {
     bool is_proxy_eligible;
     bool was_proxy_used;
     bool is_unreachable;
   };
   const TestCase test_cases[] = {
     {
       false,
       false,
       false
     },
     {
       true,
       true,
       false
     },
     {
       true,
       false,
       true
     }
   };
   for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_cases); ++i) {
     TestCase test_case = test_cases[i];
 
-    EXPECT_CALL(mock_params_, IsDataReductionProxyEligible(&mock_url_request_))
-        .WillRepeatedly(Return(test_case.is_proxy_eligible));
     EXPECT_CALL(mock_params_,
-                WasDataReductionProxyUsed(&mock_url_request_, NULL))
+                IsDataReductionProxyEligible(mock_url_request_.get()))
+                    .WillRepeatedly(Return(test_case.is_proxy_eligible));
+    EXPECT_CALL(mock_params_,
+                WasDataReductionProxyUsed(mock_url_request_.get(), NULL))
         .WillRepeatedly(Return(test_case.was_proxy_used));
 
     scoped_ptr<DataReductionProxyUsageStats> usage_stats(
         new DataReductionProxyUsageStats(
             &mock_params_, loop_proxy_, loop_proxy_));
 
-    usage_stats->OnUrlRequestCompleted(&mock_url_request_, false);
+    usage_stats->OnUrlRequestCompleted(mock_url_request_.get(), false);
     MessageLoop::current()->RunUntilIdle();
 
     EXPECT_EQ(test_case.is_unreachable,
               usage_stats->isDataReductionProxyUnreachable());
   }
 }
 
 }  // namespace data_reduction_proxy