Chromium Code Reviews| Index: chrome/browser/storage_monitor/media_storage_util_unittest.cc |
| diff --git a/chrome/browser/storage_monitor/media_storage_util_unittest.cc b/chrome/browser/storage_monitor/media_storage_util_unittest.cc |
| index f5d8ebaff6126ce7a24c349c124abcddcc24d36d..6b86a118fef0e0b0c989526d8cbc8bb4e23381e9 100644 |
| --- a/chrome/browser/storage_monitor/media_storage_util_unittest.cc |
| +++ b/chrome/browser/storage_monitor/media_storage_util_unittest.cc |
| @@ -4,10 +4,13 @@ |
| #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" |
| @@ -25,18 +28,96 @@ 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) { |