[Media Galleries] Move RemovableStorageInfo notifications to chrome namespace (part 2)

chrome/browser/system_monitor/removable_device_notifications_mac_unittest.mm

 // 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/system_monitor/removable_device_notifications_mac.h"
 
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/mac/foundation_util.h"
 #include "base/message_loop.h"
 #include "base/sys_string_conversions.h"
-#include "base/system_monitor/system_monitor.h"
-#include "base/test/mock_devices_changed_observer.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/system_monitor/media_storage_util.h"
 #include "chrome/browser/system_monitor/removable_device_constants.h"
+#include "chrome/browser/system_monitor/removable_storage_notifications_test_util.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chrome {
 
 uint64 kTestSize = 1000000ULL;
 
 namespace {
 
 DiskInfoMac CreateDiskInfoMac(const std::string& unique_id,
@@ skipping 25 matching lines @@
 }  // namespace
 
 class RemovableDeviceNotificationsMacTest : public testing::Test {
  public:
   RemovableDeviceNotificationsMacTest()
       : message_loop_(MessageLoop::TYPE_IO),
         file_thread_(content::BrowserThread::FILE, &message_loop_) {
   }
 
   virtual void SetUp() OVERRIDE {
+    notifications_ = new RemovableDeviceNotificationsMac;
     base::SystemMonitor::AllocateSystemIOPorts();
-    system_monitor_.reset(new base::SystemMonitor());
 
-    mock_devices_changed_observer_.reset(new base::MockDevicesChangedObserver);
-    system_monitor_->AddDevicesChangedObserver(
-        mock_devices_changed_observer_.get());
+    mock_storage_observer_.reset(new MockRemovableStorageObserver);
+    notifications_->AddObserver(mock_storage_observer_.get());
 
-    notifications_ = new RemovableDeviceNotificationsMac;
 
     unique_id_ = "test_id";
     display_name_ = ASCIIToUTF16("977 KB Test Display Name");
     mount_point_ = FilePath("/unused_test_directory");
     device_id_ = MediaStorageUtil::MakeDeviceId(
         MediaStorageUtil::REMOVABLE_MASS_STORAGE_NO_DCIM, unique_id_);
     disk_info_ = CreateDiskInfoMac(unique_id_, "",
                                    ASCIIToUTF16("Test Display Name"),
                                    mount_point_, kTestSize);
   }
 
  protected:
   // The message loop and file thread to run tests on.
   MessageLoop message_loop_;
   content::TestBrowserThread file_thread_;
 
-  // SystemMonitor and DevicesChangedObserver to hook together to test.
-  scoped_ptr<base::SystemMonitor> system_monitor_;
-  scoped_ptr<base::MockDevicesChangedObserver> mock_devices_changed_observer_;
+  scoped_ptr<MockRemovableStorageObserver> mock_storage_observer_;
 
   // Information about the disk.
   std::string unique_id_;
   string16 display_name_;
   FilePath mount_point_;
   std::string device_id_;
   DiskInfoMac disk_info_;
 
   scoped_refptr<RemovableDeviceNotificationsMac> notifications_;
 };
 
 TEST_F(RemovableDeviceNotificationsMacTest, AddRemove) {
   {
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageAttached(device_id_,
-                                           display_name_,
-                                           mount_point_.value()));
     notifications_->UpdateDisk(
         disk_info_, RemovableDeviceNotificationsMac::UPDATE_DEVICE_ADDED);
     message_loop_.RunUntilIdle();
+    EXPECT_EQ(1, mock_storage_observer_->attach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->attached_last().device_id);
+    EXPECT_EQ(display_name_, mock_storage_observer_->attached_last().name);
+    EXPECT_EQ(mount_point_.value(),
+              mock_storage_observer_->attached_last().location);
   }
 
   {
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageDetached(device_id_));
     notifications_->UpdateDisk(
         disk_info_, RemovableDeviceNotificationsMac::UPDATE_DEVICE_REMOVED);
     message_loop_.RunUntilIdle();
+    EXPECT_EQ(1, mock_storage_observer_->attach_calls());
+    EXPECT_EQ(1, mock_storage_observer_->detach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->detached_last().device_id);
   }
 }
 
 TEST_F(RemovableDeviceNotificationsMacTest, UpdateVolumeName) {
   {
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageAttached(device_id_,
-                                           display_name_,
-                                           mount_point_.value()));
     notifications_->UpdateDisk(
         disk_info_, RemovableDeviceNotificationsMac::UPDATE_DEVICE_ADDED);
     message_loop_.RunUntilIdle();
+
+    EXPECT_EQ(1, mock_storage_observer_->attach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->attached_last().device_id);
+    EXPECT_EQ(display_name_, mock_storage_observer_->attached_last().name);
+    EXPECT_EQ(mount_point_.value(),
+              mock_storage_observer_->attached_last().location);
   }
 
   {
     string16 new_display_name(ASCIIToUTF16("977 KB Test Display Name"));
     DiskInfoMac info2 = CreateDiskInfoMac(
         unique_id_, "", ASCIIToUTF16("Test Display Name"), mount_point_,
         kTestSize);
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageDetached(device_id_));
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageAttached(device_id_,
-                                           new_display_name,
-                                           mount_point_.value()));
     notifications_->UpdateDisk(
         info2, RemovableDeviceNotificationsMac::UPDATE_DEVICE_CHANGED);
     message_loop_.RunUntilIdle();
+
+    EXPECT_EQ(1, mock_storage_observer_->detach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->detached_last().device_id);
+    EXPECT_EQ(2, mock_storage_observer_->attach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->attached_last().device_id);
+    EXPECT_EQ(new_display_name, mock_storage_observer_->attached_last().name);
+    EXPECT_EQ(mount_point_.value(),
+              mock_storage_observer_->attached_last().location);
   }
 }
 
 TEST_F(RemovableDeviceNotificationsMacTest, DCIM) {
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   file_util::CreateDirectory(temp_dir.path().Append(kDCIMDirectoryName));
 
   FilePath mount_point = temp_dir.path();
   DiskInfoMac info = CreateDiskInfoMac(
       unique_id_, "", ASCIIToUTF16("Test Display Name"), mount_point,
       kTestSize);
   std::string device_id = MediaStorageUtil::MakeDeviceId(
       MediaStorageUtil::REMOVABLE_MASS_STORAGE_WITH_DCIM, unique_id_);
 
   {
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageAttached(device_id,
-                                           display_name_,
-                                           mount_point.value()));
     notifications_->UpdateDisk(
         info, RemovableDeviceNotificationsMac::UPDATE_DEVICE_ADDED);
     message_loop_.RunUntilIdle();
+
+    EXPECT_EQ(1, mock_storage_observer_->attach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->attached_last().device_id);
+    EXPECT_EQ(display_name_, mock_storage_observer_->attached_last().name);
+    EXPECT_EQ(mount_point_.value(),
+              mock_storage_observer_->attached_last().location);
   }
 }
 
 TEST_F(RemovableDeviceNotificationsMacTest, GetDeviceInfo) {
   {
-    EXPECT_CALL(*mock_devices_changed_observer_,
-                OnRemovableStorageAttached(device_id_,
-                                           display_name_,
-                                           mount_point_.value()));
     notifications_->UpdateDisk(
         disk_info_, RemovableDeviceNotificationsMac::UPDATE_DEVICE_ADDED);
     message_loop_.RunUntilIdle();
+
+    EXPECT_EQ(1, mock_storage_observer_->attach_calls());
+    EXPECT_EQ(device_id_, mock_storage_observer_->attached_last().device_id);
+    EXPECT_EQ(display_name_, mock_storage_observer_->attached_last().name);
+    EXPECT_EQ(mount_point_.value(),
+              mock_storage_observer_->attached_last().location);
   }
 
-  base::SystemMonitor::RemovableStorageInfo info;
+  RemovableStorageNotifications::StorageInfo info;
   EXPECT_TRUE(notifications_->GetDeviceInfoForPath(
       mount_point_.AppendASCII("foo"), &info));
   EXPECT_EQ(info.device_id, device_id_);
   EXPECT_EQ(info.name, ASCIIToUTF16("Test Display Name"));
   EXPECT_EQ(info.location, mount_point_.value());
 
   EXPECT_FALSE(notifications_->GetDeviceInfoForPath(
       FilePath("/non/matching/path"), &info));
 }
 
 TEST_F(RemovableDeviceNotificationsMacTest, GetStorageSize) {
-  EXPECT_CALL(*mock_devices_changed_observer_,
-              OnRemovableStorageAttached(testing::_,
-                                         testing::_,
-                                         testing::_));
   notifications_->UpdateDisk(
       disk_info_, RemovableDeviceNotificationsMac::UPDATE_DEVICE_ADDED);
   message_loop_.RunUntilIdle();
+  EXPECT_EQ(1, mock_storage_observer_->attach_calls());
 
   EXPECT_EQ(kTestSize,
             notifications_->GetStorageSize("/unused_test_directory"));
 }
 
 // Test that mounting a DMG doesn't send a notification.
 TEST_F(RemovableDeviceNotificationsMacTest, DMG) {
-  EXPECT_CALL(*mock_devices_changed_observer_,
-              OnRemovableStorageAttached(testing::_,
-                                         testing::_,
-                                         testing::_)).Times(0);
   DiskInfoMac info = CreateDiskInfoMac(
       unique_id_, "Disk Image", display_name_, mount_point_, kTestSize);
   notifications_->UpdateDisk(
       info, RemovableDeviceNotificationsMac::UPDATE_DEVICE_ADDED);
   message_loop_.RunUntilIdle();
+  EXPECT_EQ(0, mock_storage_observer_->attach_calls());
 }
 
 }  // namespace chrome