Consolidate storage_monitor/MediaDeviceNotificationsUtils into MediaStorageUtil.

chrome/browser/storage_monitor/media_storage_util_unittest.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 <string>
 
+#include "base/file_util.h"
+#include "base/files/scoped_temp_dir.h"
 #include "base/message_loop.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/storage_monitor/media_storage_util.h"
+#include "chrome/browser/storage_monitor/removable_device_constants.h"
 #include "chrome/browser/storage_monitor/storage_monitor.h"
 #include "chrome/browser/storage_monitor/test_storage_monitor.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chrome {
 
 namespace {
 
 // Sample mtp device id and unique id.
 const char kMtpDeviceId[] = "mtp:VendorModelSerial:ABC:1233:1237912873";
 const char kUniqueId[] = "VendorModelSerial:ABC:1233:1237912873";
 const char kImageCaptureDeviceId[] = "ic:xyz";
 
 }  // namespace
 
+using content::BrowserThread;
+
 class MediaStorageUtilTest : public testing::Test {
  public:
+  MediaStorageUtilTest()
+        : ui_thread_(BrowserThread::UI, &message_loop_),
+          file_thread_(BrowserThread::FILE) { }

[Greg Billock, 2013/03/06 21:48:27]

No space between braces

+  virtual ~MediaStorageUtilTest() { }
+
+  // Verify mounted device type.
+  void checkDeviceType(const base::FilePath::StringType& mount_point,

[Greg Billock, 2013/03/06 21:48:27]

Capitalize

+                       bool expected_val) {
+    if (expected_val)
+      EXPECT_TRUE(MediaStorageUtil::HasDcim(mount_point));
+    else
+      EXPECT_FALSE(MediaStorageUtil::HasDcim(mount_point));
+  }
+
   void ProcessAttach(const std::string& id,
                      const string16& name,
                      const base::FilePath::StringType& location) {
     monitor_.receiver()->ProcessAttach(StorageInfo(id, name, location));
   }
 
+ protected:
+  // Create mount point for the test device.
+  base::FilePath CreateMountPoint(bool create_dcim_dir) {
+    base::FilePath path(scoped_temp_dir_.path());
+    if (create_dcim_dir)
+      path = path.Append(kDCIMDirectoryName);
+    if (!file_util::CreateDirectory(path))
+      return base::FilePath();
+    return scoped_temp_dir_.path();
+  }
+
+  virtual void SetUp() OVERRIDE {
+    ASSERT_TRUE(scoped_temp_dir_.CreateUniqueTempDir());
+    file_thread_.Start();
+  }
+
+  virtual void TearDown() {
+    WaitForFileThread();
+  }
+
+  static void PostQuitToUIThread() {
+    BrowserThread::PostTask(BrowserThread::UI,
+                            FROM_HERE,
+                            MessageLoop::QuitClosure());
+  }
+
+  static void WaitForFileThread() {
+    BrowserThread::PostTask(BrowserThread::FILE,
+                            FROM_HERE,
+                            base::Bind(&PostQuitToUIThread));
+    MessageLoop::current()->Run();
+  }
+
+  MessageLoop message_loop_;
+
  private:
   chrome::test::TestStorageMonitor monitor_;
+  content::TestBrowserThread ui_thread_;
+  content::TestBrowserThread file_thread_;
+  base::ScopedTempDir scoped_temp_dir_;
 };
 
+// Test to verify that IsMediaDevice() function returns true for the given
+// media device mount point.
+TEST_F(MediaStorageUtilTest, MediaDeviceAttached) {
+  // Create a dummy mount point with DCIM Directory.
+  base::FilePath mount_point(CreateMountPoint(true));
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      base::Bind(&MediaStorageUtilTest::checkDeviceType,
+                 base::Unretained(this), mount_point.value(), true));
+  message_loop_.RunUntilIdle();
+}
+
+// Test to verify that IsMediaDevice() function returns false for a given
+// non-media device mount point.
+TEST_F(MediaStorageUtilTest, NonMediaDeviceAttached) {
+  // Create a dummy mount point without DCIM Directory.
+  base::FilePath mount_point(CreateMountPoint(false));
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      base::Bind(&MediaStorageUtilTest::checkDeviceType,
+                 base::Unretained(this), mount_point.value(), false));
+  message_loop_.RunUntilIdle();
+}
+
 // Test to verify |MediaStorageUtil::MakeDeviceId| functionality using a sample
 // mtp device unique id.
 TEST_F(MediaStorageUtilTest, MakeMtpDeviceId) {
   std::string device_id =
       MediaStorageUtil::MakeDeviceId(MediaStorageUtil::MTP_OR_PTP, kUniqueId);
   ASSERT_EQ(kMtpDeviceId, device_id);
 }
 
 // Test to verify |MediaStorageUtil::CrackDeviceId| functionality using a sample
 // mtp device id.
@@ skipping 17 matching lines @@
 TEST_F(MediaStorageUtilTest, CanCreateFileSystemForImageCapture) {
   EXPECT_TRUE(MediaStorageUtil::CanCreateFileSystem(kImageCaptureDeviceId,
                                                     base::FilePath()));
   EXPECT_FALSE(MediaStorageUtil::CanCreateFileSystem(
       "dcim:xyz", base::FilePath(FILE_PATH_LITERAL("relative"))));
   EXPECT_FALSE(MediaStorageUtil::CanCreateFileSystem(
       "dcim:xyz", base::FilePath(FILE_PATH_LITERAL("../refparent"))));
 }
 
 TEST_F(MediaStorageUtilTest, DetectDeviceFiltered) {
   MessageLoop loop;

[Greg Billock, 2013/03/06 21:50:53]

Probably need to use the test loop here.

   content::TestBrowserThread file_thread(content::BrowserThread::FILE, &loop);
 
   MediaStorageUtil::DeviceIdSet devices;
   devices.insert(kImageCaptureDeviceId);
 
   base::WaitableEvent event(true, false);
   MediaStorageUtil::FilterAttachedDevices(&devices,
       base::Bind(&base::WaitableEvent::Signal, base::Unretained(&event)));
   loop.RunUntilIdle();
   event.Wait();
   EXPECT_FALSE(devices.find(kImageCaptureDeviceId) != devices.end());
 
   ProcessAttach(kImageCaptureDeviceId, ASCIIToUTF16("name"),
                 FILE_PATH_LITERAL("/location"));
   devices.insert(kImageCaptureDeviceId);
   event.Reset();
   MediaStorageUtil::FilterAttachedDevices(&devices,
       base::Bind(&base::WaitableEvent::Signal, base::Unretained(&event)));
   loop.RunUntilIdle();
   event.Wait();
 
   EXPECT_TRUE(devices.find(kImageCaptureDeviceId) != devices.end());
 }
 
 }  // namespace chrome