src/chromeos/disks/mock_disk_mount_manager.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chromeos/disks/mock_disk_mount_manager.h"
   6
   7 #include <utility>
   8
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/stl_util.h"
  11 #include "base/strings/string_util.h"
  12
  13 using testing::_;
  14 using testing::AnyNumber;
  15 using testing::Invoke;
  16 using testing::ReturnRef;
  17
  18 namespace chromeos {
  19 namespace disks {
  20
  21 namespace {
  22
  23 const char* kTestSystemPath = "/this/system/path";
  24 const char* kTestSystemPathPrefix = "/this/system";
  25 const char* kTestDevicePath = "/this/device/path";
  26 const char* kTestMountPath = "/media/foofoo";
  27 const char* kTestFilePath = "/this/file/path";
  28 const char* kTestDeviceLabel = "A label";
  29 const char* kTestDriveLabel = "Another label";
  30 const char* kTestVendorId = "0123";
  31 const char* kTestVendorName = "A vendor";
  32 const char* kTestProductId = "abcd";
  33 const char* kTestProductName = "A product";
  34 const char* kTestUuid = "FFFF-FFFF";
  35
  36 }  // namespace
  37
  38 void MockDiskMountManager::AddObserverInternal(
  39     DiskMountManager::Observer* observer) {
  40   observers_.AddObserver(observer);
  41 }
  42
  43 void MockDiskMountManager::RemoveObserverInternal(
  44     DiskMountManager::Observer* observer) {
  45   observers_.RemoveObserver(observer);
  46 }
  47
  48 MockDiskMountManager::MockDiskMountManager() {
  49   ON_CALL(*this, AddObserver(_))
  50       .WillByDefault(Invoke(this, &MockDiskMountManager::AddObserverInternal));
  51   ON_CALL(*this, RemoveObserver(_))
  52       .WillByDefault(Invoke(this,
  53                             &MockDiskMountManager::RemoveObserverInternal));
  54   ON_CALL(*this, disks())
  55       .WillByDefault(Invoke(this, &MockDiskMountManager::disksInternal));
  56   ON_CALL(*this, mount_points())
  57       .WillByDefault(Invoke(this, &MockDiskMountManager::mountPointsInternal));
  58   ON_CALL(*this, FindDiskBySourcePath(_))
  59       .WillByDefault(Invoke(
  60           this, &MockDiskMountManager::FindDiskBySourcePathInternal));
  61 }
  62
  63 MockDiskMountManager::~MockDiskMountManager() {
  64   STLDeleteContainerPairSecondPointers(disks_.begin(), disks_.end());
  65   disks_.clear();
  66 }
  67
  68 void MockDiskMountManager::NotifyDeviceInsertEvents() {
  69   scoped_ptr<DiskMountManager::Disk> disk1(new DiskMountManager::Disk(
  70       std::string(kTestDevicePath),
  71       std::string(),
  72       std::string(kTestSystemPath),
  73       std::string(kTestFilePath),
  74       std::string(),
  75       std::string(kTestDriveLabel),
  76       std::string(kTestVendorId),
  77       std::string(kTestVendorName),
  78       std::string(kTestProductId),
  79       std::string(kTestProductName),
  80       std::string(kTestUuid),
  81       std::string(kTestSystemPathPrefix),
  82       DEVICE_TYPE_USB,
  83       4294967295U,
  84       false,  // is_parent
  85       false,  // is_read_only
  86       true,  // has_media
  87       false,  // on_boot_device
  88       false));  // is_hidden
  89
  90   disks_.clear();
  91   disks_.insert(std::pair<std::string, DiskMountManager::Disk*>(
  92       std::string(kTestDevicePath), disk1.get()));
  93
  94   // Device Added
  95   NotifyDeviceChanged(DEVICE_ADDED, kTestSystemPath);
  96
  97   // Disk Added
  98   NotifyDiskChanged(DISK_ADDED, disk1.get());
  99
 100   // Disk Changed
 101   scoped_ptr<DiskMountManager::Disk> disk2(new DiskMountManager::Disk(
 102       std::string(kTestDevicePath),
 103       std::string(kTestMountPath),
 104       std::string(kTestSystemPath),
 105       std::string(kTestFilePath),
 106       std::string(kTestDeviceLabel),
 107       std::string(kTestDriveLabel),
 108       std::string(kTestVendorId),
 109       std::string(kTestVendorName),
 110       std::string(kTestProductId),
 111       std::string(kTestProductName),
 112       std::string(kTestUuid),
 113       std::string(kTestSystemPathPrefix),
 114       DEVICE_TYPE_MOBILE,
 115       1073741824,
 116       false,  // is_parent
 117       false,  // is_read_only
 118       true,  // has_media
 119       false,  // on_boot_device
 120       false));  // is_hidden
 121   disks_.clear();
 122   disks_.insert(std::pair<std::string, DiskMountManager::Disk*>(
 123       std::string(kTestDevicePath), disk2.get()));
 124   NotifyDiskChanged(DISK_CHANGED, disk2.get());
 125 }
 126
 127 void MockDiskMountManager::NotifyDeviceRemoveEvents() {
 128   scoped_ptr<DiskMountManager::Disk> disk(new DiskMountManager::Disk(
 129       std::string(kTestDevicePath),
 130       std::string(kTestMountPath),
 131       std::string(kTestSystemPath),
 132       std::string(kTestFilePath),
 133       std::string(kTestDeviceLabel),
 134       std::string(kTestDriveLabel),
 135       std::string(kTestVendorId),
 136       std::string(kTestVendorName),
 137       std::string(kTestProductId),
 138       std::string(kTestProductName),
 139       std::string(kTestUuid),
 140       std::string(kTestSystemPathPrefix),
 141       DEVICE_TYPE_SD,
 142       1073741824,
 143       false,  // is_parent
 144       false,  // is_read_only
 145       true,  // has_media
 146       false,  // on_boot_device
 147       false));  // is_hidden
 148   disks_.clear();
 149   disks_.insert(std::pair<std::string, DiskMountManager::Disk*>(
 150       std::string(kTestDevicePath), disk.get()));
 151   NotifyDiskChanged(DISK_REMOVED, disk.get());
 152 }
 153
 154 void MockDiskMountManager::SetupDefaultReplies() {
 155   EXPECT_CALL(*this, AddObserver(_))
 156       .Times(AnyNumber());
 157   EXPECT_CALL(*this, RemoveObserver(_))
 158       .Times(AnyNumber());
 159   EXPECT_CALL(*this, disks())
 160       .WillRepeatedly(ReturnRef(disks_));
 161   EXPECT_CALL(*this, mount_points())
 162       .WillRepeatedly(ReturnRef(mount_points_));
 163   EXPECT_CALL(*this, FindDiskBySourcePath(_))
 164       .Times(AnyNumber());
 165   EXPECT_CALL(*this, RequestMountInfoRefresh())
 166       .Times(AnyNumber());
 167   EXPECT_CALL(*this, MountPath(_, _, _, _))
 168       .Times(AnyNumber());
 169   EXPECT_CALL(*this, UnmountPath(_, _, _))
 170       .Times(AnyNumber());
 171   EXPECT_CALL(*this, FormatMountedDevice(_))
 172       .Times(AnyNumber());
 173   EXPECT_CALL(*this, UnmountDeviceRecursively(_, _))
 174       .Times(AnyNumber());
 175 }
 176
 177 void MockDiskMountManager::CreateDiskEntryForMountDevice(
 178     const DiskMountManager::MountPointInfo& mount_info,
 179     const std::string& device_id,
 180     const std::string& device_label,
 181     const std::string& vendor_name,
 182     const std::string& product_name,
 183     DeviceType device_type,
 184     uint64 total_size_in_bytes,
 185     bool is_parent,
 186     bool has_media,
 187     bool on_boot_device) {
 188   Disk* disk = new DiskMountManager::Disk(mount_info.source_path,
 189                                           mount_info.mount_path,
 190                                           std::string(),  // system_path
 191                                           mount_info.source_path,
 192                                           device_label,
 193                                           std::string(),  // drive_label
 194                                           std::string(),  // vendor_id
 195                                           vendor_name,
 196                                           std::string(),  // product_id
 197                                           product_name,
 198                                           device_id,      // fs_uuid
 199                                           std::string(),  // system_path_prefix
 200                                           device_type,
 201                                           total_size_in_bytes,
 202                                           is_parent,
 203                                           false,  // is_read_only
 204                                           has_media,
 205                                           on_boot_device,
 206                                           false);  // is_hidden
 207   DiskMountManager::DiskMap::iterator it = disks_.find(mount_info.source_path);
 208   if (it == disks_.end()) {
 209     disks_.insert(std::make_pair(std::string(mount_info.source_path), disk));
 210   } else {
 211     delete it->second;
 212     it->second = disk;
 213   }
 214 }
 215
 216 void MockDiskMountManager::RemoveDiskEntryForMountDevice(
 217     const DiskMountManager::MountPointInfo& mount_info) {
 218   DiskMountManager::DiskMap::iterator it = disks_.find(mount_info.source_path);
 219   if (it != disks_.end()) {
 220     delete it->second;
 221     disks_.erase(it);
 222   }
 223 }
 224
 225 const DiskMountManager::MountPointMap&
 226 MockDiskMountManager::mountPointsInternal() const {
 227   return mount_points_;
 228 }
 229
 230 const DiskMountManager::Disk*
 231 MockDiskMountManager::FindDiskBySourcePathInternal(
 232     const std::string& source_path) const {
 233   DiskMap::const_iterator disk_it = disks_.find(source_path);
 234   return disk_it == disks_.end() ? NULL : disk_it->second;
 235 }
 236
 237 void MockDiskMountManager::NotifyDiskChanged(
 238     DiskEvent event,
 239     const DiskMountManager::Disk* disk) {
 240   FOR_EACH_OBSERVER(Observer, observers_, OnDiskEvent(event, disk));
 241 }
 242
 243 void MockDiskMountManager::NotifyDeviceChanged(DeviceEvent event,
 244                                                const std::string& path) {
 245   FOR_EACH_OBSERVER(Observer, observers_, OnDeviceEvent(event, path));
 246 }
 247
 248 }  // namespace disks
 249 }  // namespace chromeos