Home | History | Annotate | Download | only in file_manager 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/chromeos/file_manager/mounted_disk_monitor.h"
      6 
      7 #include "base/basictypes.h"
      8 #include "base/message_loop/message_loop.h"
      9 #include "base/run_loop.h"
     10 #include "base/stl_util.h"
     11 #include "chrome/browser/chromeos/file_manager/fake_disk_mount_manager.h"
     12 #include "chromeos/dbus/fake_power_manager_client.h"
     13 #include "testing/gtest/include/gtest/gtest.h"
     14 
     15 namespace file_manager {
     16 namespace {
     17 
     18 // Creates a fake disk with |device_path| and |fs_uuid|.
     19 scoped_ptr<chromeos::disks::DiskMountManager::Disk> CreateDisk(
     20     const std::string& device_path,
     21     const std::string& fs_uuid) {
     22   return make_scoped_ptr(
     23       new chromeos::disks::DiskMountManager::Disk(
     24           device_path, "", "", "", "", "", "", "", "", "", fs_uuid, "",
     25           chromeos::DEVICE_TYPE_USB, 0, false, false, false, false, false));
     26 }
     27 
     28 }  // namespace
     29 
     30 class MountedDiskMonitorTest : public testing::Test {
     31  protected:
     32   MountedDiskMonitorTest() {
     33   }
     34 
     35   virtual ~MountedDiskMonitorTest() {
     36   }
     37 
     38   virtual void SetUp() OVERRIDE {
     39     power_manager_client_.reset(new chromeos::FakePowerManagerClient);
     40     disk_mount_manager_.reset(new FakeDiskMountManager);
     41     mounted_disk_monitor_.reset(new MountedDiskMonitor(
     42         power_manager_client_.get(),
     43         disk_mount_manager_.get()));
     44     mounted_disk_monitor_->set_resuming_time_span_for_testing(
     45         base::TimeDelta::FromSeconds(0));
     46   }
     47 
     48   base::MessageLoop message_loop_;
     49   scoped_ptr<chromeos::FakePowerManagerClient> power_manager_client_;
     50   scoped_ptr<FakeDiskMountManager> disk_mount_manager_;
     51   scoped_ptr<MountedDiskMonitor> mounted_disk_monitor_;
     52 };
     53 
     54 // Makes sure that just mounting and unmounting repeatedly doesn't affect to
     55 // "remounting" state.
     56 TEST_F(MountedDiskMonitorTest, WithoutSuspend) {
     57   scoped_ptr<chromeos::disks::DiskMountManager::Disk> disk(
     58       CreateDisk("removable_device1", "uuid1"));
     59 
     60   chromeos::disks::DiskMountManager::Disk* disk_ptr = disk.get();
     61 
     62   const chromeos::disks::DiskMountManager::MountPointInfo kMountPoint(
     63       "removable_device1", "/tmp/removable_device1",
     64       chromeos::MOUNT_TYPE_DEVICE, chromeos::disks::MOUNT_CONDITION_NONE);
     65 
     66   ASSERT_TRUE(disk_mount_manager_->AddDiskForTest(disk.release()));
     67 
     68   // First, the disk is not remounting.
     69   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk_ptr));
     70 
     71   // Simple mounting and unmounting doesn't affect remounting state.
     72   mounted_disk_monitor_->OnMountEvent(
     73       chromeos::disks::DiskMountManager::MOUNTING,
     74       chromeos::MOUNT_ERROR_NONE,
     75       kMountPoint);
     76   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk_ptr));
     77 
     78   mounted_disk_monitor_->OnMountEvent(
     79       chromeos::disks::DiskMountManager::UNMOUNTING,
     80       chromeos::MOUNT_ERROR_NONE,
     81       kMountPoint);
     82   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk_ptr));
     83 
     84   // Mounting again also should not affect remounting state.
     85   mounted_disk_monitor_->OnMountEvent(
     86       chromeos::disks::DiskMountManager::MOUNTING,
     87       chromeos::MOUNT_ERROR_NONE,
     88       kMountPoint);
     89   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk_ptr));
     90 }
     91 
     92 // Makes sure that the unmounting after system resuming triggers the
     93 // "remounting" state, then after some period, the state is reset.
     94 TEST_F(MountedDiskMonitorTest, SuspendAndResume) {
     95   scoped_ptr<chromeos::disks::DiskMountManager::Disk> disk1(
     96       CreateDisk("removable_device1", "uuid1"));
     97   scoped_ptr<chromeos::disks::DiskMountManager::Disk> disk2(
     98       CreateDisk("removable_device2", "uuid2"));
     99 
    100   chromeos::disks::DiskMountManager::Disk* disk1_ptr = disk1.get();
    101   chromeos::disks::DiskMountManager::Disk* disk2_ptr = disk2.get();
    102 
    103   const chromeos::disks::DiskMountManager::MountPointInfo kMountPoint1(
    104       "removable_device1", "/tmp/removable_device1",
    105       chromeos::MOUNT_TYPE_DEVICE, chromeos::disks::MOUNT_CONDITION_NONE);
    106   const chromeos::disks::DiskMountManager::MountPointInfo kMountPoint2(
    107       "removable_device2", "/tmp/removable_device2",
    108       chromeos::MOUNT_TYPE_DEVICE, chromeos::disks::MOUNT_CONDITION_NONE);
    109 
    110   ASSERT_TRUE(disk_mount_manager_->AddDiskForTest(disk1.release()));
    111   ASSERT_TRUE(disk_mount_manager_->AddDiskForTest(disk2.release()));
    112 
    113   // Mount |disk1|.
    114   mounted_disk_monitor_->OnMountEvent(
    115       chromeos::disks::DiskMountManager::MOUNTING,
    116       chromeos::MOUNT_ERROR_NONE,
    117       kMountPoint1);
    118   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk1_ptr));
    119 
    120   // Pseudo system suspend and resume.
    121   mounted_disk_monitor_->SuspendImminent();
    122   mounted_disk_monitor_->SuspendDone(base::TimeDelta::FromSeconds(0));
    123 
    124   // On system resume, we expect unmount and then mount immediately.
    125   // During the phase, we expect the disk is remounting.
    126   mounted_disk_monitor_->OnMountEvent(
    127       chromeos::disks::DiskMountManager::UNMOUNTING,
    128       chromeos::MOUNT_ERROR_NONE,
    129       kMountPoint1);
    130   EXPECT_TRUE(mounted_disk_monitor_->DiskIsRemounting(*disk1_ptr));
    131 
    132   mounted_disk_monitor_->OnMountEvent(
    133       chromeos::disks::DiskMountManager::MOUNTING,
    134       chromeos::MOUNT_ERROR_NONE,
    135       kMountPoint1);
    136   EXPECT_TRUE(mounted_disk_monitor_->DiskIsRemounting(*disk1_ptr));
    137 
    138   // New disk should not be "remounting."
    139   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk2_ptr));
    140   mounted_disk_monitor_->OnMountEvent(
    141       chromeos::disks::DiskMountManager::MOUNTING,
    142       chromeos::MOUNT_ERROR_NONE,
    143       kMountPoint2);
    144   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk2_ptr));
    145 
    146   // After certain period, remounting state should be cleared.
    147   base::RunLoop().RunUntilIdle();  // Emulate time passage.
    148   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk1_ptr));
    149   EXPECT_FALSE(mounted_disk_monitor_->DiskIsRemounting(*disk2_ptr));
    150 }
    151 
    152 }  // namespace file_manager
    153