Index: chromeos/disks/mock_disk_mount_manager.cc |
diff --git a/chromeos/disks/mock_disk_mount_manager.cc b/chromeos/disks/mock_disk_mount_manager.cc |
index 6e21d753f55d5b18367a40c3e2166b800a4857ee..468c770524b8c1efa685219a94fd8aedf6469e8a 100644 |
--- a/chromeos/disks/mock_disk_mount_manager.cc |
+++ b/chromeos/disks/mock_disk_mount_manager.cc |
@@ -210,6 +210,24 @@ void MockDiskMountManager::CreateDiskEntryForMountDevice( |
} |
} |
+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); |