Home | History | Annotate | Download | only in storage_monitor 
      1 // Copyright 2014 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 "base/message_loop/message_loop.h"
      6 #include "base/strings/utf_string_conversions.h"
      7 #include "base/synchronization/waitable_event.h"
      8 #include "components/storage_monitor/mock_removable_storage_observer.h"
      9 #include "components/storage_monitor/storage_monitor.h"
     10 #include "components/storage_monitor/test_storage_monitor.h"
     11 #include "testing/gtest/include/gtest/gtest.h"
     12 
     13 namespace {
     14 
     15 void SetLatch(bool* called) {
     16   *called = true;
     17 }
     18 
     19 }  // namespace
     20 
     21 namespace storage_monitor {
     22 
     23 TEST(StorageMonitorTest, TestInitialize) {
     24   TestStorageMonitor::Destroy();
     25   TestStorageMonitor monitor;
     26   EXPECT_FALSE(monitor.init_called());
     27 
     28   bool initialized = false;
     29   monitor.EnsureInitialized(base::Bind(&SetLatch, &initialized));
     30   EXPECT_TRUE(monitor.init_called());
     31   EXPECT_FALSE(initialized);
     32   monitor.MarkInitialized();
     33   EXPECT_TRUE(initialized);
     34 }
     35 
     36 TEST(StorageMonitorTest, DeviceAttachDetachNotifications) {
     37   TestStorageMonitor::Destroy();
     38   base::MessageLoop message_loop;
     39   const std::string kDeviceId1 = "dcim:UUID:FFF0-0001";
     40   const std::string kDeviceId2 = "dcim:UUID:FFF0-0002";
     41   MockRemovableStorageObserver observer1;
     42   MockRemovableStorageObserver observer2;
     43   TestStorageMonitor monitor;
     44   monitor.AddObserver(&observer1);
     45   monitor.AddObserver(&observer2);
     46 
     47   StorageInfo info(kDeviceId1, FILE_PATH_LITERAL("path"), base::string16(),
     48                    base::string16(), base::string16(), 0);
     49   monitor.receiver()->ProcessAttach(info);
     50   message_loop.RunUntilIdle();
     51 
     52   EXPECT_EQ(kDeviceId1, observer1.last_attached().device_id());
     53   EXPECT_EQ(FILE_PATH_LITERAL("path"), observer1.last_attached().location());
     54   EXPECT_EQ(kDeviceId1, observer2.last_attached().device_id());
     55   EXPECT_EQ(FILE_PATH_LITERAL("path"), observer2.last_attached().location());
     56   EXPECT_EQ(1, observer1.attach_calls());
     57   EXPECT_EQ(0, observer1.detach_calls());
     58 
     59   monitor.receiver()->ProcessDetach(kDeviceId1);
     60   monitor.receiver()->ProcessDetach(kDeviceId2);
     61   message_loop.RunUntilIdle();
     62 
     63   EXPECT_EQ(kDeviceId1, observer1.last_detached().device_id());
     64   EXPECT_EQ(FILE_PATH_LITERAL("path"), observer1.last_detached().location());
     65   EXPECT_EQ(kDeviceId1, observer2.last_detached().device_id());
     66   EXPECT_EQ(FILE_PATH_LITERAL("path"), observer2.last_detached().location());
     67 
     68   EXPECT_EQ(1, observer1.attach_calls());
     69   EXPECT_EQ(1, observer2.attach_calls());
     70 
     71   // The kDeviceId2 won't be notified since it was never attached.
     72   EXPECT_EQ(1, observer1.detach_calls());
     73   EXPECT_EQ(1, observer2.detach_calls());
     74 
     75   monitor.RemoveObserver(&observer1);
     76   monitor.RemoveObserver(&observer2);
     77 }
     78 
     79 TEST(StorageMonitorTest, GetAllAvailableStoragesEmpty) {
     80   TestStorageMonitor::Destroy();
     81   base::MessageLoop message_loop;
     82   TestStorageMonitor monitor;
     83   std::vector<StorageInfo> devices = monitor.GetAllAvailableStorages();
     84   EXPECT_EQ(0U, devices.size());
     85 }
     86 
     87 TEST(StorageMonitorTest, GetAllAvailableStorageAttachDetach) {
     88   TestStorageMonitor::Destroy();
     89   base::MessageLoop message_loop;
     90   TestStorageMonitor monitor;
     91   const std::string kDeviceId1 = "dcim:UUID:FFF0-0042";
     92   const base::FilePath kDevicePath1(FILE_PATH_LITERAL("/testfoo"));
     93   StorageInfo info1(kDeviceId1, kDevicePath1.value(), base::string16(),
     94                     base::string16(), base::string16(), 0);
     95   monitor.receiver()->ProcessAttach(info1);
     96   message_loop.RunUntilIdle();
     97   std::vector<StorageInfo> devices = monitor.GetAllAvailableStorages();
     98   ASSERT_EQ(1U, devices.size());
     99   EXPECT_EQ(kDeviceId1, devices[0].device_id());
    100   EXPECT_EQ(kDevicePath1.value(), devices[0].location());
    101 
    102   const std::string kDeviceId2 = "dcim:UUID:FFF0-0044";
    103   const base::FilePath kDevicePath2(FILE_PATH_LITERAL("/testbar"));
    104   StorageInfo info2(kDeviceId2, kDevicePath2.value(), base::string16(),
    105                     base::string16(), base::string16(), 0);
    106   monitor.receiver()->ProcessAttach(info2);
    107   message_loop.RunUntilIdle();
    108   devices = monitor.GetAllAvailableStorages();
    109   ASSERT_EQ(2U, devices.size());
    110   EXPECT_EQ(kDeviceId1, devices[0].device_id());
    111   EXPECT_EQ(kDevicePath1.value(), devices[0].location());
    112   EXPECT_EQ(kDeviceId2, devices[1].device_id());
    113   EXPECT_EQ(kDevicePath2.value(), devices[1].location());
    114 
    115   monitor.receiver()->ProcessDetach(kDeviceId1);
    116   message_loop.RunUntilIdle();
    117   devices = monitor.GetAllAvailableStorages();
    118   ASSERT_EQ(1U, devices.size());
    119   EXPECT_EQ(kDeviceId2, devices[0].device_id());
    120   EXPECT_EQ(kDevicePath2.value(), devices[0].location());
    121 
    122   monitor.receiver()->ProcessDetach(kDeviceId2);
    123   message_loop.RunUntilIdle();
    124   devices = monitor.GetAllAvailableStorages();
    125   EXPECT_EQ(0U, devices.size());
    126 }
    127 
    128 }  // namespace storage_monitor
    129