Change Interceptors into URLRequestJobFactory::ProtocolHandlers.

chrome/browser/policy/device_management_service_browsertest.cc

 // Copyright (c) 2012 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 "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/stl_util.h"
 #include "chrome/browser/policy/cloud_policy_constants.h"
 #include "chrome/browser/policy/device_management_service.h"
 #include "chrome/test/base/in_process_browser_test.h"
+#include "content/public/browser/browser_thread.h"
 #include "net/base/upload_data.h"
 #include "net/test/test_server.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_request.h"
+#include "net/url_request/url_request_filter.h"
+#include "net/url_request/url_request_job_factory.h"
 #include "net/url_request/url_request_test_job.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+using content::BrowserThread;
 using testing::DoAll;
 using testing::Invoke;
 using testing::InvokeWithoutArgs;
 using testing::_;
 
 namespace em = enterprise_management;
 
 namespace policy {
 
 // Dummy service URL for testing with request interception enabled.
 const char kServiceUrl[] = "http://example.com/device_management";
 
 // Interceptor implementation that returns test data back to the service.
-class CannedResponseInterceptor : public net::URLRequest::Interceptor {
+class CannedResponseInterceptor
+    : public net::URLRequestJobFactory::ProtocolHandler {
  public:
   explicit CannedResponseInterceptor(const GURL& service_url)
       : service_url_(service_url) {
-    net::URLRequest::Deprecated::RegisterRequestInterceptor(this);
+    BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+                            base::Bind(&CannedResponseInterceptor::Register,
+                                       base::Unretained(this)));
   }
 
   virtual ~CannedResponseInterceptor() {
-    net::URLRequest::Deprecated::UnregisterRequestInterceptor(this);
+    BrowserThread::PostTask(BrowserThread::IO,
+                            FROM_HERE,
+                            base::Bind(Unregister));
   }
 
-  // net::URLRequest::Interceptor overrides.
-  virtual net::URLRequestJob* MaybeIntercept(
+  // net::URLRequestJobFactory::ProtocolHandler overrides.
+  virtual net::URLRequestJob* MaybeCreateJob(
       net::URLRequest* request,
-      net::NetworkDelegate* network_delegate) OVERRIDE {
+      net::NetworkDelegate* network_delegate) const OVERRIDE {
     em::DeviceManagementRequest dm_request;
     net::UploadData* upload = request->get_upload_mutable();
     if (request->url().GetOrigin() == service_url_.GetOrigin() &&
         request->url().path() == service_url_.path() &&
         upload != NULL &&
         upload->elements().size() == 1) {
       std::string response_data;
       ConstructResponse(upload->elements()[0]->bytes(),
                         upload->elements()[0]->bytes_length(),
                         &response_data);
       return new net::URLRequestTestJob(request,
                                         network_delegate,
                                         net::URLRequestTestJob::test_headers(),
                                         response_data,
                                         true);
     }
 
     return NULL;
   }
 
  private:
+  void Register() {
+    net::URLRequestFilter::GetInstance()->AddHostnameProtocolHandler(
+        "http", "example.com", this);
+  }
+  static void Unregister() {
+    net::URLRequestFilter::GetInstance()->RemoveHostnameHandler(
+        "http", "example.com");
+  }
   void ConstructResponse(const char* request_data,
                          uint64 request_data_length,
-                         std::string* response_data) {
+                         std::string* response_data) const {
     em::DeviceManagementRequest request;
     ASSERT_TRUE(request.ParseFromArray(request_data, request_data_length));
     em::DeviceManagementResponse response;
     if (request.has_register_request()) {
       response.mutable_register_response()->set_device_management_token(
           "fake_token");
     } else if (request.has_unregister_request()) {
       response.mutable_unregister_response();
     } else if (request.has_policy_request()) {
       response.mutable_policy_response()->add_response();
@@ skipping 129 matching lines @@
   MessageLoop::current()->Run();
 }
 
 INSTANTIATE_TEST_CASE_P(
     DeviceManagementServiceIntegrationTestInstance,
     DeviceManagementServiceIntegrationTest,
     testing::Values(&DeviceManagementServiceIntegrationTest::InitCannedResponse,
                     &DeviceManagementServiceIntegrationTest::InitTestServer));
 
 }  // namespace policy