MediaFileSystemRegistry unit tests.

chrome/browser/media_gallery/media_file_system_registry.h

 // 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.
 
 // MediaFileSystemRegistry registers pictures directories and media devices as
 // File API filesystems and keeps track of the path to filesystem ID mappings.
 
 #ifndef CHROME_BROWSER_MEDIA_GALLERY_MEDIA_FILE_SYSTEM_REGISTRY_H_
 #define CHROME_BROWSER_MEDIA_GALLERY_MEDIA_FILE_SYSTEM_REGISTRY_H_
 
 #include <map>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/lazy_instance.h"
 #include "base/file_path.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/prefs/public/pref_change_registrar.h"
 #include "base/prefs/public/pref_observer.h"
 #include "base/system_monitor/system_monitor.h"
 #include "webkit/fileapi/media/mtp_device_file_system_config.h"
 
+#if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
+#include "chrome/browser/media_gallery/mtp_device_delegate_impl.h"
+#endif
+
 class Profile;
 
 namespace content {
 class RenderViewHost;
 }
 
 namespace extensions {
 class Extension;
 }
 
 namespace fileapi {
 class IsolatedContext;
 }
 
 namespace chrome {
 
 class ExtensionGalleriesHost;
 class MediaGalleriesPreferences;
-class ScopedMTPDeviceMapEntry;
 
 struct MediaFileSystemInfo {
   MediaFileSystemInfo(const std::string& fs_name,
                       const FilePath& fs_path,
                       const std::string& filesystem_id);
   MediaFileSystemInfo();
 
   std::string name;  // JSON string, must not contain slashes.
   FilePath path;
   std::string fsid;
 };
 
+#if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
+// Class to manage MTPDeviceDelegateImpl object for the attached MTP device.
+// Refcounted to reuse the same MTP device delegate entry across extensions.
+// This class supports WeakPtr (extends SupportsWeakPtr) to expose itself as
+// a weak pointer to MediaFileSystemRegistry.
+class ScopedMTPDeviceMapEntry
+    : public base::RefCounted<ScopedMTPDeviceMapEntry>,
+      public base::SupportsWeakPtr<ScopedMTPDeviceMapEntry> {
+ public:
+  // |no_references_callback| is called when the last ScopedMTPDeviceMapEntry
+  // reference goes away.
+  ScopedMTPDeviceMapEntry(const FilePath::StringType& device_location,
+                          const base::Closure& no_references_callback);
+
+ private:
+  // Friend declaration for ref counted implementation.
+  friend class base::RefCounted<ScopedMTPDeviceMapEntry>;
+
+  // Private because this class is ref-counted.
+  ~ScopedMTPDeviceMapEntry();
+
+  // Store the MTP or PTP device location.
+  const FilePath::StringType device_location_;
+
+  // Store a raw pointer of MTPDeviceDelegateImpl object.
+  // MTPDeviceDelegateImpl is ref-counted and owned by MTPDeviceMapService.
+  // This class tells MTPDeviceMapService to dispose of it when the last
+  // reference to |this| goes away.
+  MTPDeviceDelegateImpl* delegate_;
+
+  // A callback to call when the last reference of this object goes away.
+  base::Closure no_references_callback_;
+
+  DISALLOW_COPY_AND_ASSIGN(ScopedMTPDeviceMapEntry);
+};
+#endif
+
 class MediaFileSystemContext {
  public:
   virtual ~MediaFileSystemContext() {}
 
   // Register a media file system (filtered to media files) for |path| and
   // return the new file system id.
   virtual std::string RegisterFileSystemForMassStorage(
       const std::string& device_id, const FilePath& path) = 0;
 
 #if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
@@ skipping 33 matching lines @@
   // Called on the UI thread.
   MediaGalleriesPreferences* GetPreferences(Profile* profile);
 
   // base::SystemMonitor::DevicesChangedObserver implementation.
   virtual void OnRemovableStorageAttached(
       const std::string& id,
       const string16& name,
       const FilePath::StringType& location) OVERRIDE;
   virtual void OnRemovableStorageDetached(const std::string& id) OVERRIDE;
 
+  size_t GetExtensionHostCountForTests() const;
+
  private:
+  friend class TestMediaFileSystemContext;
   friend struct base::DefaultLazyInstanceTraits<MediaFileSystemRegistry>;
   class MediaFileSystemContextImpl;
 
   // Map an extension to the ExtensionGalleriesHost.
   typedef std::map<std::string /*extension_id*/,
                    scoped_refptr<ExtensionGalleriesHost> > ExtensionHostMap;
   // Map a profile and extension to the ExtensionGalleriesHost.
   typedef std::map<Profile*, ExtensionHostMap> ExtensionGalleriesHostMap;
   // Map a profile to a PrefChangeRegistrar.
   typedef std::map<Profile*, PrefChangeRegistrar*> PrefChangeRegistrarMap;
@@ skipping 40 matching lines @@
 #endif
 
   scoped_ptr<MediaFileSystemContext> file_system_context_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaFileSystemRegistry);
 };
 
 }  // namespace chrome
 
 #endif  // CHROME_BROWSER_MEDIA_GALLERY_MEDIA_FILE_SYSTEM_REGISTRY_H_