Created a helper function "FindDiskBySourcePath" in DiskMountManager.

chrome/browser/chromeos/extensions/file_browser_private_apitest.cc

Index: chrome/browser/chromeos/extensions/file_browser_private_apitest.cc
diff --git a/chrome/browser/chromeos/extensions/file_browser_private_apitest.cc b/chrome/browser/chromeos/extensions/file_browser_private_apitest.cc
index 303dbcd4b34a4624c0adf28a378bddded4aaaf33..3df89e1d37fa95ee6e6a9f331477e84bfbc658ed 100644
--- a/chrome/browser/chromeos/extensions/file_browser_private_apitest.cc
+++ b/chrome/browser/chromeos/extensions/file_browser_private_apitest.cc
@@ -15,6 +15,7 @@
 #include "webkit/fileapi/file_system_mount_point_provider.h"
 
 using ::testing::_;
+using ::testing::AnyNumber;
 using ::testing::ReturnRef;
 using ::testing::StrEq;
 using content::BrowserContext;
@@ -152,6 +153,11 @@ class ExtensionFileBrowserPrivateApiTest : public ExtensionApiTest {
     chromeos::disks::DiskMountManager::InitializeForTesting(
         disk_mount_manager_mock_);
     disk_mount_manager_mock_->SetupDefaultReplies();
+
+    // OVERRIDE FindDiskBySourcePath mock function.
+    ON_CALL(*disk_mount_manager_mock_, FindDiskBySourcePath(_)).
+        WillByDefault(Invoke(
+            this, &ExtensionFileBrowserPrivateApiTest::FindVolumeBySourcePath));
   }
 
   // ExtensionApiTest override
@@ -210,6 +216,13 @@ class ExtensionFileBrowserPrivateApiTest : public ExtensionApiTest {
     }
   }
 
+  const DiskMountManager::Disk* FindVolumeBySourcePath(
+      const std::string& source_path) {
+    DiskMountManager::DiskMap::const_iterator volume_it =
+        volumes_.find(source_path);
+    return volume_it == volumes_.end() ? NULL : volume_it->second;

[Ben Chan, 2012/08/06 21:51:27]

(volume_it == volumes_.end()) ? 

[kmadhusu, 2012/08/06 21:55:48]

sure. Done.

+  }
+
  protected:
   chromeos::disks::MockDiskMountManager* disk_mount_manager_mock_;
   DiskMountManager::DiskMap volumes_;