Adds a new removable storage provider for imageWriterPrivate on Chrome OS

chromeos/disks/mock_disk_mount_manager.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 "chromeos/disks/mock_disk_mount_manager.h"
 
 #include <utility>
 
 #include "base/message_loop/message_loop.h"
 #include "base/stl_util.h"
@@ skipping 192 matching lines @@
                                           false);  // is_hidden
   DiskMountManager::DiskMap::iterator it = disks_.find(mount_info.source_path);
   if (it == disks_.end()) {
     disks_.insert(std::make_pair(std::string(mount_info.source_path), disk));
   } else {
     delete it->second;
     it->second = disk;
   }
 }
 
+void MockDiskMountManager::AddDiskEntry(Disk* disk) {
+  DiskMountManager::DiskMap::iterator it = disks_.find(disk->device_path());
+  if (it == disks_.end()) {
+    disks_.insert(std::make_pair(std::string(disk->device_path()), disk));
+  } else {
+    delete it->second;
+    it->second = disk;
+  }
+}
+
+void MockDiskMountManager::RemoveDiskEntry(Disk* disk) {
+  DiskMountManager::DiskMap::iterator it = disks_.find(disk->device_path());
+  if (it != disks_.end()) {
+    delete it->second;
+    disks_.erase(it);
+  }
+}
+
 void MockDiskMountManager::RemoveDiskEntryForMountDevice(
     const DiskMountManager::MountPointInfo& mount_info) {
   DiskMountManager::DiskMap::iterator it = disks_.find(mount_info.source_path);
   if (it != disks_.end()) {
     delete it->second;
     disks_.erase(it);
   }
 }
 
 const DiskMountManager::MountPointMap&
@@ skipping 14 matching lines @@
   FOR_EACH_OBSERVER(Observer, observers_, OnDiskEvent(event, disk));
 }
 
 void MockDiskMountManager::NotifyDeviceChanged(DeviceEvent event,
                                                const std::string& path) {
   FOR_EACH_OBSERVER(Observer, observers_, OnDeviceEvent(event, path));
 }
 
 }  // namespace disks
 }  // namespace chromeos