HTTP server for testing Google Drive.

chrome/browser/chromeos/gdata/gdata_wapi_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/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/json/json_reader.h"
 #include "chrome/browser/chromeos/gdata/gdata_wapi_service.h"
+#include "chrome/browser/chromeos/gdata/test_servers/http_test_server.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
-#include "net/test/test_server.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace gdata {
 
 namespace {
 
 class GDataTest : public InProcessBrowserTest {
  public:
   GDataTest()
-      : InProcessBrowserTest(),
-        gdata_test_server_(net::TestServer::TYPE_GDATA,
-                           net::TestServer::kLocalhost,
-                           FilePath(FILE_PATH_LITERAL("chrome/test/data"))) {
+      : InProcessBrowserTest() {
   }
 
   virtual void SetUpOnMainThread() OVERRIDE {
-    ASSERT_TRUE(gdata_test_server_.Start());
     service_.reset(new gdata::GDataWapiService);
     service_->Initialize(browser()->profile());
     service_->auth_service_for_testing()->set_access_token_for_testing(
         net::TestServer::kGDataAuthToken);
+    gdata_test_server_.reset(new gdata::test_servers::HttpTestServer(true));
+    SetUpTestServer();
+  }
+
+  virtual void SetUpTestServer() {
+    test_server_file_url_ = gdata_test_server_->RegisterFileResponse(
+        "some/path/testfile.txt",
+        FilePath(FILE_PATH_LITERAL(
+            "chrome/test/data/chromeos/gdata/testfile.txt")),
+        "text/plain",
+        gdata::test_servers::SUCCESS);
+
+    test_server_root_feed_url_ = gdata_test_server_->RegisterFileResponse(
+        FilePath(FILE_PATH_LITERAL(
+            "chrome/test/data/chromeos/gdata/root_feed.json")),
+        "text/json");
   }
 
   virtual void CleanUpOnMainThread() {
     service_.reset();
+    gdata_test_server_.reset();
   }
 
  protected:
   FilePath GetTestCachedFilePath(const FilePath& file_name) {
     return browser()->profile()->GetPath().Append(file_name);
   }
 
-  net::TestServer gdata_test_server_;
+  scoped_ptr<gdata::test_servers::HttpTestServer> gdata_test_server_;
   scoped_ptr<gdata::GDataWapiService> service_;
+  GURL test_server_file_url_;
+  GURL test_server_root_feed_url_;
 };
 
 // The test callback for GDataWapiService::DownloadFile().
 void TestDownloadCallback(gdata::GDataErrorCode* result,
                           std::string* contents,
                           gdata::GDataErrorCode error,
                           const GURL& content_url,
                           const FilePath& temp_file) {
   *result = error;
   file_util::ReadFileToString(temp_file, contents);
@@ skipping 12 matching lines @@
 }
 
 }  // namespace
 
 IN_PROC_BROWSER_TEST_F(GDataTest, Download) {
   gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
   std::string contents;
   service_->DownloadFile(
       FilePath("/dummy/gdata/testfile.txt"),
       GetTestCachedFilePath(FilePath("cached_testfile.txt")),
-      gdata_test_server_.GetURL("files/chromeos/gdata/testfile.txt"),
+      test_server_file_url_,
       base::Bind(&TestDownloadCallback, &result, &contents),
       gdata::GetContentCallback());
   content::RunMessageLoop();
 
   EXPECT_EQ(gdata::HTTP_SUCCESS, result);
-  FilePath expected_filepath = gdata_test_server_.document_root().Append(
-      FilePath(FILE_PATH_LITERAL("chromeos/gdata/testfile.txt")));
+  FilePath expected_filepath = FilePath(
+      FILE_PATH_LITERAL("chrome/test/data/chromeos/gdata/testfile.txt"));
   std::string expected_contents;
   file_util::ReadFileToString(expected_filepath, &expected_contents);
   EXPECT_EQ(expected_contents, contents);
 }
 
 IN_PROC_BROWSER_TEST_F(GDataTest, NonExistingDownload) {
   gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
   std::string dummy_contents;
   service_->DownloadFile(
       FilePath("/dummy/gdata/no-such-file.txt"),
       GetTestCachedFilePath(FilePath("cache_no-such-file.txt")),
-      gdata_test_server_.GetURL("files/chromeos/gdata/no-such-file.txt"),
+      gdata_test_server_->GetErrorURL(gdata::test_servers::NOT_FOUND),
       base::Bind(&TestDownloadCallback, &result, &dummy_contents),
       gdata::GetContentCallback());
   content::RunMessageLoop();
 
   EXPECT_EQ(gdata::HTTP_NOT_FOUND, result);
   // Do not verify the not found message.
 }
 
 IN_PROC_BROWSER_TEST_F(GDataTest, GetDocuments) {
   gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
   base::Value* result_data = NULL;
   service_->GetDocuments(
-      gdata_test_server_.GetURL("files/chromeos/gdata/root_feed.json"),
+      test_server_root_feed_url_,
       0,  // start_changestamp
       std::string(),  // search string
       std::string(),  // directory resource ID
       base::Bind(&TestGetDocumentsCallback, &result, &result_data));
   content::RunMessageLoop();
 
   EXPECT_EQ(gdata::HTTP_SUCCESS, result);
   ASSERT_TRUE(result_data);
-  FilePath expected_filepath = gdata_test_server_.document_root().Append(
-      FilePath(FILE_PATH_LITERAL("chromeos/gdata/root_feed.json")));
+  FilePath expected_filepath =  FilePath(
+      FILE_PATH_LITERAL("chrome/test/data/chromeos/gdata/root_feed.json"));
   std::string expected_contents;
   file_util::ReadFileToString(expected_filepath, &expected_contents);
   scoped_ptr<base::Value> expected_data(
       base::JSONReader::Read(expected_contents));
   EXPECT_TRUE(base::Value::Equals(expected_data.get(), result_data));
   delete result_data;
 }
 
 IN_PROC_BROWSER_TEST_F(GDataTest, GetDocumentsFailure) {
   // testfile.txt exists but the response is not JSON, so it should
   // emit a parse error instead.
   gdata::GDataErrorCode result = gdata::GDATA_OTHER_ERROR;
   base::Value* result_data = NULL;
   service_->GetDocuments(
-      gdata_test_server_.GetURL("files/chromeos/gdata/testfile.txt"),
+      test_server_file_url_,
       0,  // start_changestamp
       std::string(),  // search string
       std::string(),  // directory resource ID
       base::Bind(&TestGetDocumentsCallback, &result, &result_data));
   content::RunMessageLoop();
 
   EXPECT_EQ(gdata::GDATA_PARSE_ERROR, result);
   EXPECT_FALSE(result_data);
 }
 
 }  // namespace gdata