Add mechanism to auto mount file systems in response to a URL request.

content/browser/fileapi/file_system_dir_url_request_job_unittest.cc

 // Copyright 2013 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 "webkit/browser/fileapi/file_system_dir_url_request_job.h"
 
 #include <string>
 
+#include "base/file_util.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/format_macros.h"
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/platform_file.h"
 #include "base/run_loop.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/public/test/test_file_system_context.h"
 #include "net/base/net_errors.h"
 #include "net/base/net_util.h"
 #include "net/base/request_priority.h"
 #include "net/http/http_request_headers.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/icu/source/i18n/unicode/regex.h"
+#include "webkit/browser/fileapi/external_mount_points.h"
 #include "webkit/browser/fileapi/file_system_context.h"
 #include "webkit/browser/fileapi/file_system_file_util.h"
 #include "webkit/browser/fileapi/file_system_operation_context.h"
 #include "webkit/browser/fileapi/file_system_url.h"
 #include "webkit/browser/quota/mock_special_storage_policy.h"
 
 using fileapi::FileSystemContext;
 using fileapi::FileSystemOperationContext;
 using fileapi::FileSystemURL;
 
 namespace content {
 namespace {
 
 // We always use the TEMPORARY FileSystem in this test.
 static const char kFileSystemURLPrefix[] =
     "filesystem:http://remote/temporary/";
 
+static const char kValidExternalMountPoint[] = "mnt_name";

[kinuko, 2014/03/13 04:19:20]

nit: no need of static inside anon namespace? (And I realized we have static too
on the line above)

[vandebo (ex-Chrome), 2014/03/13 20:19:32]

Done.

+
+// An auto mounter that will try to mount anything for |storage_domain| =
+// "automount", but will only succeed for the mount point "mnt_name".
+bool TestAutoMountForURLRequest(
+    const base::FilePath& path,
+    const net::URLRequest* /*url_request*/,
+    const fileapi::FileSystemURL& filesystem_url,
+    const std::string& storage_domain,
+    const base::Callback<void(base::File::Error result)>& callback) {
+  if (storage_domain != "automount")
+    return false;
+
+  std::vector<base::FilePath::StringType> components;
+  filesystem_url.path().GetComponents(&components);
+#if defined(OS_POSIX)
+  std::string mount_point = components[0];
+#elif defined(OS_WIN)
+  std::string mount_point = base::WideToUTF8(components[0]);

[kinuko, 2014/03/13 04:19:20]

FilePath(components[0]).AsUTF8Unsafe() would work for either case?

[vandebo (ex-Chrome), 2014/03/13 20:19:32]

Yea, but then you make an unnecessary object...  I guess for tests it doesn't
matter. Done.

+#endif  // OS_WIN
+
+  if (mount_point == kValidExternalMountPoint) {
+    fileapi::ExternalMountPoints::GetSystemInstance()->RegisterFileSystem(
+        kValidExternalMountPoint, fileapi::kFileSystemTypeNativeLocal,
+        fileapi::FileSystemMountOption(), path);
+    callback.Run(base::File::FILE_OK);
+  } else {
+    callback.Run(base::File::FILE_ERROR_NOT_FOUND);
+  }
+  return true;
+}
+
 }  // namespace
 
 class FileSystemDirURLRequestJobTest : public testing::Test {
  protected:
   FileSystemDirURLRequestJobTest()
     : weak_factory_(this) {
   }
 
   virtual void SetUp() OVERRIDE {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
@@ skipping 28 matching lines @@
     ASSERT_EQ(base::File::FILE_OK, result);
   }
 
   void TestRequestHelper(const GURL& url, bool run_to_completion,
                          FileSystemContext* file_system_context) {
     delegate_.reset(new net::TestDelegate());
     delegate_->set_quit_on_redirect(true);
     request_ = empty_context_.CreateRequest(
         url, net::DEFAULT_PRIORITY, delegate_.get());
     job_ = new fileapi::FileSystemDirURLRequestJob(
-        request_.get(), NULL, file_system_context);
+        request_.get(), NULL, url.GetOrigin().host(), file_system_context);
 
     request_->Start();
     ASSERT_TRUE(request_->is_pending());  // verify that we're starting async
     if (run_to_completion)
       base::MessageLoop::current()->Run();
   }
 
   void TestRequest(const GURL& url) {
     TestRequestHelper(url, true, file_system_context_.get());
   }
@@ skipping 154 matching lines @@
   EXPECT_EQ("<script>start(\"foo\\\\bar\");</script>", line);
 #elif defined(OS_POSIX)
   EXPECT_EQ("<script>start(\"/foo/bar\");</script>", line);
 #endif
 
   EXPECT_TRUE(!!std::getline(in, line));
   VerifyListingEntry(line, "hoge", "hoge", false, 10);
 
   EXPECT_TRUE(!!std::getline(in, line));
   VerifyListingEntry(line, "baz", "baz", true, 0);
+  EXPECT_FALSE(!!std::getline(in, line));
 }
 
 TEST_F(FileSystemDirURLRequestJobTest, InvalidURL) {
   TestRequest(GURL("filesystem:/foo/bar/baz"));
   ASSERT_FALSE(request_->is_pending());
   EXPECT_TRUE(delegate_->request_failed());
   ASSERT_FALSE(request_->status().is_success());
   EXPECT_EQ(net::ERR_INVALID_URL, request_->status().error());
 }
 
@@ skipping 36 matching lines @@
   EXPECT_TRUE(!!std::getline(in, line));
   EXPECT_FALSE(!!std::getline(in, line));
 
   TestRequestWithContext(CreateFileSystemURL("foo"),
                          file_system_context.get());
   ASSERT_FALSE(request_->is_pending());
   ASSERT_FALSE(request_->status().is_success());
   EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request_->status().error());
 }
 
+TEST_F(FileSystemDirURLRequestJobTest, AutoMountDirectoryListing) {
+  base::FilePath mnt_point = temp_dir_.path().AppendASCII("auto_mount_dir");
+  ASSERT_TRUE(base::CreateDirectory(mnt_point));
+  ASSERT_TRUE(base::CreateDirectory(mnt_point.AppendASCII("foo")));
+  ASSERT_EQ(10,
+            base::WriteFile(mnt_point.AppendASCII("bar"), "1234567890", 10));
+
+  std::vector<fileapi::URLRequestAutoMountHandler> handlers;
+  handlers.push_back(base::Bind(&TestAutoMountForURLRequest, mnt_point));
+  scoped_refptr<FileSystemContext> file_system_context =
+      CreateFileSystemContextWithAutoMountersForTesting(NULL, handlers,
+                                                        temp_dir_.path());
+
+  TestRequestWithContext(GURL("filesystem:http://automount/external/mnt_name"),
+                         file_system_context);
+
+  ASSERT_FALSE(request_->is_pending());
+  EXPECT_EQ(1, delegate_->response_started_count());
+  EXPECT_FALSE(delegate_->received_data_before_response());
+  EXPECT_GT(delegate_->bytes_received(), 0);
+
+  std::istringstream in(delegate_->data_received());
+  std::string line;
+  EXPECT_TRUE(!!std::getline(in, line));  // |line| contains the temp dir path.
+
+  EXPECT_TRUE(!!std::getline(in, line));
+  VerifyListingEntry(line, "bar", "bar", false, 10);
+
+  EXPECT_TRUE(!!std::getline(in, line));
+  VerifyListingEntry(line, "foo", "foo", true, 4096);
+  EXPECT_FALSE(!!std::getline(in, line));
+
+  ASSERT_TRUE(
+      fileapi::ExternalMountPoints::GetSystemInstance()->RevokeFileSystem(
+          kValidExternalMountPoint));
+}
+
+TEST_F(FileSystemDirURLRequestJobTest, AutoMountInvalidRoot) {
+  base::FilePath mnt_point = temp_dir_.path().AppendASCII("auto_mount_dir");
+  ASSERT_TRUE(base::CreateDirectory(mnt_point));
+
+  std::vector<fileapi::URLRequestAutoMountHandler> handlers;
+  handlers.push_back(base::Bind(&TestAutoMountForURLRequest, mnt_point));
+  scoped_refptr<FileSystemContext> file_system_context =
+      CreateFileSystemContextWithAutoMountersForTesting(NULL, handlers,
+                                                        temp_dir_.path());
+
+  TestRequestWithContext(GURL("filesystem:http://automount/external/invalid"),
+                         file_system_context);
+  ASSERT_FALSE(request_->is_pending());
+  ASSERT_FALSE(request_->status().is_success());
+  EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request_->status().error());
+
+  ASSERT_FALSE(
+      fileapi::ExternalMountPoints::GetSystemInstance()->RevokeFileSystem(
+          "invalid"));
+}
+
+TEST_F(FileSystemDirURLRequestJobTest, AutoMountNoHandler) {
+  base::FilePath mnt_point = temp_dir_.path().AppendASCII("auto_mount_dir");
+  ASSERT_TRUE(base::CreateDirectory(mnt_point));
+
+  std::vector<fileapi::URLRequestAutoMountHandler> handlers;
+  handlers.push_back(base::Bind(&TestAutoMountForURLRequest, mnt_point));
+  scoped_refptr<FileSystemContext> file_system_context =
+      CreateFileSystemContextWithAutoMountersForTesting(NULL, handlers,
+                                                        temp_dir_.path());
+
+  TestRequestWithContext(GURL("filesystem:http://noauto/external/mnt_name"),
+                         file_system_context);
+  ASSERT_FALSE(request_->is_pending());
+  ASSERT_FALSE(request_->status().is_success());
+  EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request_->status().error());
+
+  ASSERT_FALSE(
+      fileapi::ExternalMountPoints::GetSystemInstance()->RevokeFileSystem(
+          kValidExternalMountPoint));
+}
+
 }  // namespace (anonymous)
 }  // namespace content