Merge CrosDisksClientStubImpl into FakeCrosDisksClient

chromeos/dbus/fake_cros_disks_client.cc

 // Copyright (c) 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 "chromeos/dbus/fake_cros_disks_client.h"
 
 #include "base/bind.h"
+#include "base/files/file_util.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
+#include "base/task_runner_util.h"
+#include "base/threading/worker_pool.h"
 
 namespace chromeos {
 
+namespace {
+
+// Performs fake mounting by creating a directory with a dummy file.
+MountError PerformFakeMount(const std::string& source_path,
+                            const base::FilePath& mounted_path) {
+  // Just create an empty directory and shows it as the mounted directory.
+  if (!base::CreateDirectory(mounted_path)) {
+    DLOG(ERROR) << "Failed to create directory at " << mounted_path.value();
+    return MOUNT_ERROR_DIRECTORY_CREATION_FAILED;
+  }
+
+  // Put a dummy file.
+  const base::FilePath dummy_file_path =
+      mounted_path.Append("SUCCESSFULLY_PERFORMED_FAKE_MOUNT.txt");
+  const std::string dummy_file_content = "This is a dummy file.";
+  const int write_result = base::WriteFile(
+      dummy_file_path, dummy_file_content.data(), dummy_file_content.size());
+  if (write_result != static_cast<int>(dummy_file_content.size())) {
+    DLOG(ERROR) << "Failed to put a dummy file at "
+                << dummy_file_path.value();
+    return MOUNT_ERROR_MOUNT_PROGRAM_FAILED;
+  }
+
+  return MOUNT_ERROR_NONE;
+}
+
+// Continuation of Mount().
+void DidMount(const CrosDisksClient::MountCompletedHandler&
+              mount_completed_handler,
+              const std::string& source_path,
+              MountType type,
+              const base::Closure& callback,
+              const base::FilePath& mounted_path,
+              MountError mount_error) {
+  // Tell the caller of Mount() that the mount request was accepted.
+  base::MessageLoopProxy::current()->PostTask(FROM_HERE, callback);
+
+  // Tell the caller of Mount() that the mount completed.
+  if (!mount_completed_handler.is_null()) {
+    base::MessageLoopProxy::current()->PostTask(
+        FROM_HERE,
+        base::Bind(mount_completed_handler,
+                   MountEntry(mount_error, source_path, type,
+                              mounted_path.AsUTF8Unsafe())));
+  }
+}
+
+}  // namespace
+
 FakeCrosDisksClient::FakeCrosDisksClient()
     : unmount_call_count_(0),
       unmount_success_(true),
       format_call_count_(0),
       format_success_(true) {
 }
 
 FakeCrosDisksClient::~FakeCrosDisksClient() {
 }
 
 void FakeCrosDisksClient::Init(dbus::Bus* bus) {
 }
 
 void FakeCrosDisksClient::Mount(const std::string& source_path,
                                 const std::string& source_format,
                                 const std::string& mount_label,
                                 const base::Closure& callback,
                                 const base::Closure& error_callback) {
+  // This fake implementation only accepts archive mount requests.
+  const MountType type = MOUNT_TYPE_ARCHIVE;
+
+  const base::FilePath mounted_path = GetArchiveMountPoint().Append(
+      base::FilePath::FromUTF8Unsafe(mount_label));
+  mounted_paths_.insert(mounted_path);
+
+  base::PostTaskAndReplyWithResult(
+      base::WorkerPool::GetTaskRunner(true /* task_is_slow */).get(),
+      FROM_HERE,
+      base::Bind(&PerformFakeMount, source_path, mounted_path),
+      base::Bind(&DidMount,
+                 mount_completed_handler_,
+                 source_path,
+                 type,
+                 callback,
+                 mounted_path));
 }
 
 void FakeCrosDisksClient::Unmount(const std::string& device_path,
                                   UnmountOptions options,
                                   const base::Closure& callback,
                                   const base::Closure& error_callback) {
   DCHECK(!callback.is_null());
   DCHECK(!error_callback.is_null());
 
+  // Remove the dummy mounted directory if it exists.
+  if (mounted_paths_.count(base::FilePath::FromUTF8Unsafe(device_path)) > 0) {
+    mounted_paths_.erase(base::FilePath::FromUTF8Unsafe(device_path));
+    base::WorkerPool::PostTaskAndReply(
+        FROM_HERE,
+        base::Bind(base::IgnoreResult(&base::DeleteFile),
+                   base::FilePath::FromUTF8Unsafe(device_path),
+                   true /* recursive */),
+        callback,
+        true /* task_is_slow */);
+  }
+
   unmount_call_count_++;
   last_unmount_device_path_ = device_path;
   last_unmount_options_ = options;
   base::MessageLoopProxy::current()->PostTask(
       FROM_HERE, unmount_success_ ? callback : error_callback);
   if (!unmount_listener_.is_null())
     unmount_listener_.Run();
 }
 
 void FakeCrosDisksClient::EnumerateAutoMountableDevices(
@@ skipping 67 matching lines @@
 bool FakeCrosDisksClient::SendFormatCompletedEvent(
     FormatError error_code,
     const std::string& device_path) {
   if (format_completed_handler_.is_null())
     return false;
   format_completed_handler_.Run(error_code, device_path);
   return true;
 }
 
 }  // namespace chromeos