components/storage_monitor/test_storage_monitor.h

   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 #ifndef COMPONENTS_STORAGE_MONITOR_TEST_STORAGE_MONITOR_H_
   6 #define COMPONENTS_STORAGE_MONITOR_TEST_STORAGE_MONITOR_H_
   7
   8 #include <string>
   9
  10 #include "components/storage_monitor/storage_monitor.h"
  11
  12 namespace storage_monitor {
  13
  14 class TestStorageMonitor : public StorageMonitor {
  15  public:
  16   TestStorageMonitor();
  17   ~TestStorageMonitor() override;
  18
  19   void Init() override;
  20
  21   void MarkInitialized();
  22
  23   // Create and initialize a new TestStorageMonitor and install it
  24   // as the StorageMonitor singleton. If there is a StorageMonitor instance
  25   // already in place, NULL is returned, otherwise the TestStorageMonitor
  26   // instance is returned. Use |Destroy| to delete the singleton.
  27   static TestStorageMonitor* CreateAndInstall();
  28
  29   // Create and initialize a new TestStorageMonitor and install it
  30   // as the StorageMonitor singleton. TestStorageMonitor is returned for
  31   // convenience. Use |Destroy| to delete the singleton.
  32   static TestStorageMonitor* CreateForBrowserTests();
  33
  34   // Synchronously initialize the current storage monitor.
  35   static void SyncInitialize();
  36
  37   bool GetStorageInfoForPath(const base::FilePath& path,
  38                              StorageInfo* device_info) const override;
  39
  40 #if defined(OS_WIN)
  41   virtual bool GetMTPStorageInfoFromDeviceId(
  42       const std::string& storage_device_id,
  43       base::string16* device_location,
  44       base::string16* storage_object_id) const override;
  45 #endif
  46
  47 #if defined(OS_LINUX)
  48   virtual device::MediaTransferProtocolManager*
  49       media_transfer_protocol_manager() override;
  50 #endif
  51
  52   Receiver* receiver() const override;
  53
  54   void EjectDevice(
  55       const std::string& device_id,
  56       base::Callback<void(StorageMonitor::EjectStatus)> callback) override;
  57
  58   const std::string& ejected_device() const { return ejected_device_; }
  59
  60   bool init_called() const { return init_called_; }
  61
  62  private:
  63   // Whether TestStorageMonitor::Init() has been called for not.
  64   bool init_called_;
  65
  66   // The last device to be ejected.
  67   std::string ejected_device_;
  68
  69 #if defined(OS_LINUX)
  70   scoped_ptr<device::MediaTransferProtocolManager>
  71       media_transfer_protocol_manager_;
  72 #endif
  73 };
  74
  75 }  // namespace storage_monitor
  76
  77 #endif  // COMPONENTS_STORAGE_MONITOR_TEST_STORAGE_MONITOR_H_