Split out policy code from net/tools/testserver.

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/bind.h"
 #include "base/bind_helpers.h"
 #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/browser/policy/test/local_policy_test_server.h"
 #include "chrome/browser/policy/test_request_interceptor.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "net/base/upload_bytes_element_reader.h"
 #include "net/base/upload_data_stream.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_test_job.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::DoAll;
 using testing::Invoke;
 using testing::InvokeWithoutArgs;
 using testing::_;
 
@@ skipping 61 matching lines @@
   MOCK_METHOD2(OnJobDone, void(DeviceManagementStatus,
                                const em::DeviceManagementResponse&));
 
   std::string InitCannedResponse() {
     interceptor_.reset(new TestRequestInterceptor("localhost"));
     return "http://localhost";
   }
 
   std::string InitTestServer() {
     StartTestServer();
-    return test_server_->GetURL("device_management").spec();
+    return test_server_->GetServiceURL().spec();
   }
 
  protected:
   void ExpectRequest() {
     if (interceptor_)
       interceptor_->PushJobCallback(base::Bind(&ResponseJob));
   }
 
   void PerformRegistration() {
     ExpectRequest();
@@ skipping 21 matching lines @@
   }
 
   virtual void CleanUpOnMainThread() OVERRIDE {
     service_.reset();
     test_server_.reset();
     interceptor_.reset();
   }
 
   void StartTestServer() {
     test_server_.reset(
-        new net::TestServer(
-            net::TestServer::TYPE_HTTP,
-            net::TestServer::kLocalhost,
-            base::FilePath(FILE_PATH_LITERAL("chrome/test/data/policy"))));
+        new LocalPolicyTestServer("device_management_service_browsertest"));
     ASSERT_TRUE(test_server_->Start());
   }
 
   void RecordToken(DeviceManagementStatus status,
                    const em::DeviceManagementResponse& response) {
     token_ = response.register_response().device_management_token();
   }
 
   std::string token_;
   scoped_ptr<DeviceManagementService> service_;
-  scoped_ptr<net::TestServer> test_server_;
+  scoped_ptr<LocalPolicyTestServer> test_server_;
   scoped_ptr<TestRequestInterceptor> interceptor_;
 };
 
 IN_PROC_BROWSER_TEST_P(DeviceManagementServiceIntegrationTest, Registration) {
   PerformRegistration();
   EXPECT_FALSE(token_.empty());
 }
 
 IN_PROC_BROWSER_TEST_P(DeviceManagementServiceIntegrationTest, PolicyFetch) {
   PerformRegistration();
@@ skipping 44 matching lines @@
   MessageLoop::current()->Run();
 }
 
 INSTANTIATE_TEST_CASE_P(
     DeviceManagementServiceIntegrationTestInstance,
     DeviceManagementServiceIntegrationTest,
     testing::Values(&DeviceManagementServiceIntegrationTest::InitCannedResponse,
                     &DeviceManagementServiceIntegrationTest::InitTestServer));
 
 }  // namespace policy