Created a helper function "FindDiskBySourcePath" in DiskMountManager.

chrome/browser/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 "chrome/browser/chromeos/disks/mock_disk_mount_manager.h"
 
 #include <utility>
 
 #include "base/message_loop.h"
 #include "base/stl_util.h"
@@ skipping 35 matching lines @@
 MockDiskMountManager::MockDiskMountManager() {
   ON_CALL(*this, AddObserver(_))
       .WillByDefault(Invoke(this, &MockDiskMountManager::AddObserverInternal));
   ON_CALL(*this, RemoveObserver(_))
       .WillByDefault(Invoke(this,
                             &MockDiskMountManager::RemoveObserverInternal));
   ON_CALL(*this, disks())
       .WillByDefault(Invoke(this, &MockDiskMountManager::disksInternal));
   ON_CALL(*this, mount_points())
       .WillByDefault(Invoke(this, &MockDiskMountManager::mountPointsInternal));
+  ON_CALL(*this, FindDiskBySourcePath(_))
+      .WillByDefault(Invoke(
+          this, &MockDiskMountManager::FindDiskBySourcePathInternal));
 }
 
 MockDiskMountManager::~MockDiskMountManager() {
   STLDeleteContainerPairSecondPointers(disks_.begin(), disks_.end());
   disks_.clear();
 }
 
 void MockDiskMountManager::NotifyDeviceInsertEvents() {
   scoped_ptr<DiskMountManager::Disk> disk1(new DiskMountManager::Disk(
       std::string(kTestDevicePath),
@@ skipping 74 matching lines @@
 
 void MockDiskMountManager::SetupDefaultReplies() {
   EXPECT_CALL(*this, AddObserver(_))
       .Times(AnyNumber());
   EXPECT_CALL(*this, RemoveObserver(_))
       .Times(AnyNumber());
   EXPECT_CALL(*this, disks())
       .WillRepeatedly(ReturnRef(disks_));
   EXPECT_CALL(*this, mount_points())
       .WillRepeatedly(ReturnRef(mount_points_));
+  EXPECT_CALL(*this, FindDiskBySourcePath(_))
+      .Times(AnyNumber());
   EXPECT_CALL(*this, RequestMountInfoRefresh())
       .Times(AnyNumber());
   EXPECT_CALL(*this, MountPath(_, _, _, _))
       .Times(AnyNumber());
   EXPECT_CALL(*this, UnmountPath(_))
       .Times(AnyNumber());
   EXPECT_CALL(*this, FormatUnmountedDevice(_))
       .Times(AnyNumber());
   EXPECT_CALL(*this, FormatMountedDevice(_))
       .Times(AnyNumber());
@@ skipping 35 matching lines @@
     delete it->second;
     disks_.erase(it);
   }
 }
 
 const DiskMountManager::MountPointMap&
 MockDiskMountManager::mountPointsInternal() const {
   return mount_points_;
 }
 
+const DiskMountManager::Disk*
+MockDiskMountManager::FindDiskBySourcePathInternal(
+    const std::string& source_path) const {
+  DiskMap::const_iterator disk_it = disks_.find(source_path);
+  return disk_it == disks_.end() ? NULL : disk_it->second;
+}
+
 void MockDiskMountManager::NotifyDiskChanged(
     DiskMountManagerEventType event,
     const DiskMountManager::Disk* disk) {
   // Make sure we run on UI thread.
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   FOR_EACH_OBSERVER(Observer, observers_, DiskChanged(event, disk));
 }
 
 void MockDiskMountManager::NotifyDeviceChanged(DiskMountManagerEventType event,
                                                const std::string& path) {
   // Make sure we run on UI thread.
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   FOR_EACH_OBSERVER(Observer, observers_, DeviceChanged(event, path));
 }
 
 }  // namespace disks
 }  // namespace chromeos