mtp_device_delegate_impl_win_unittest.cc revision d0247b1b59f9c528cb6df88b4f2b9afaf80d181e 
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 <vector>
6
7#include "base/command_line.h"
8#include "base/files/file_path.h"
9#include "base/message_loop/message_loop.h"
10#include "base/run_loop.h"
11#include "base/stl_util.h"
12#include "chrome/browser/browser_process.h"
13#include "chrome/browser/extensions/extension_service.h"
14#include "chrome/browser/extensions/extension_system.h"
15#include "chrome/browser/extensions/test_extension_system.h"
16#include "chrome/browser/media_galleries/media_file_system_registry.h"
17#include "chrome/browser/media_galleries/media_galleries_test_util.h"
18#include "chrome/browser/storage_monitor/storage_info.h"
19#include "chrome/browser/storage_monitor/storage_monitor.h"
20#include "chrome/browser/storage_monitor/test_portable_device_watcher_win.h"
21#include "chrome/browser/storage_monitor/test_storage_monitor.h"
22#include "chrome/browser/storage_monitor/test_storage_monitor_win.h"
23#include "chrome/browser/storage_monitor/test_volume_mount_watcher_win.h"
24#include "chrome/test/base/chrome_render_view_host_test_harness.h"
25#include "chrome/test/base/testing_browser_process.h"
26#include "chrome/test/base/testing_profile.h"
27#include "content/public/browser/render_view_host.h"
28#include "content/public/browser/web_contents.h"
29#include "testing/gtest/include/gtest/gtest.h"
30
31namespace {
32
33typedef std::map<MediaGalleryPrefId, MediaFileSystemInfo> FSInfoMap;
34
35void GetGalleryInfoCallback(
36    FSInfoMap* results,
37    const std::vector<MediaFileSystemInfo>& file_systems) {
38  for (size_t i = 0; i < file_systems.size(); ++i) {
39    ASSERT_FALSE(ContainsKey(*results, file_systems[i].pref_id));
40    (*results)[file_systems[i].pref_id] = file_systems[i];
41  }
42}
43
44}  // namespace
45
46class MTPDeviceDelegateImplWinTest : public ChromeRenderViewHostTestHarness {
47 protected:
48  void SetUp() OVERRIDE;
49  void TearDown() OVERRIDE;
50
51  void ProcessAttach(const std::string& id,
52                     const string16& name,
53                     const base::FilePath::StringType& location);
54  std::string AttachDevice(StorageInfo::Type type,
55                           const std::string& unique_id,
56                           const base::FilePath& location);
57  void CheckGalleryInfo(const MediaFileSystemInfo& info,
58                        const string16& name,
59                        const base::FilePath& path,
60                        bool removable,
61                        bool media_device);
62
63  // Pointer to the storage monitor. Owned by TestingBrowserProcess.
64  TestStorageMonitorWin* monitor_;
65  scoped_refptr<extensions::Extension> extension_;
66
67  EnsureMediaDirectoriesExists media_directories_;
68};
69
70void MTPDeviceDelegateImplWinTest::SetUp() {
71  ChromeRenderViewHostTestHarness::SetUp();
72
73  TestStorageMonitor::RemoveSingleton();
74  TestPortableDeviceWatcherWin* portable_device_watcher =
75      new TestPortableDeviceWatcherWin;
76  TestVolumeMountWatcherWin* mount_watcher = new TestVolumeMountWatcherWin;
77  portable_device_watcher->set_use_dummy_mtp_storage_info(true);
78  scoped_ptr<TestStorageMonitorWin> monitor(
79      new TestStorageMonitorWin(mount_watcher, portable_device_watcher));
80  TestingBrowserProcess* browser_process = TestingBrowserProcess::GetGlobal();
81  DCHECK(browser_process);
82  monitor_ = monitor.get();
83  browser_process->SetStorageMonitor(monitor.Pass());
84
85  base::RunLoop runloop;
86  monitor_->EnsureInitialized(runloop.QuitClosure());
87  runloop.Run();
88
89  extensions::TestExtensionSystem* extension_system(
90      static_cast<extensions::TestExtensionSystem*>(
91          extensions::ExtensionSystem::Get(profile())));
92  extension_system->CreateExtensionService(
93      CommandLine::ForCurrentProcess(), base::FilePath(), false);
94
95  std::vector<std::string> all_permissions;
96  all_permissions.push_back("allAutoDetected");
97  all_permissions.push_back("read");
98  extension_ = AddMediaGalleriesApp("all", all_permissions, profile());
99}
100
101void MTPDeviceDelegateImplWinTest::TearDown() {
102  // Windows storage monitor must be destroyed on the same thread
103  // as construction.
104  TestStorageMonitor::RemoveSingleton();
105
106  ChromeRenderViewHostTestHarness::TearDown();
107}
108
109void MTPDeviceDelegateImplWinTest::ProcessAttach(
110    const std::string& id,
111    const string16& label,
112    const base::FilePath::StringType& location) {
113  StorageInfo info(id, string16(), location, label, string16(), string16(), 0);
114  monitor_->receiver()->ProcessAttach(info);
115}
116
117std::string MTPDeviceDelegateImplWinTest::AttachDevice(
118    StorageInfo::Type type,
119    const std::string& unique_id,
120    const base::FilePath& location) {
121  std::string device_id = StorageInfo::MakeDeviceId(type, unique_id);
122  DCHECK(StorageInfo::IsRemovableDevice(device_id));
123  string16 label = location.LossyDisplayName();
124  ProcessAttach(device_id, label, location.value());
125  base::RunLoop().RunUntilIdle();
126  return device_id;
127}
128
129void MTPDeviceDelegateImplWinTest::CheckGalleryInfo(
130    const MediaFileSystemInfo& info,
131    const string16& name,
132    const base::FilePath& path,
133    bool removable,
134    bool media_device) {
135  EXPECT_EQ(name, info.name);
136  EXPECT_EQ(path, info.path);
137  EXPECT_EQ(removable, info.removable);
138  EXPECT_EQ(media_device, info.media_device);
139  EXPECT_NE(0UL, info.pref_id);
140
141  if (removable)
142    EXPECT_NE(0UL, info.transient_device_id.size());
143  else
144    EXPECT_EQ(0UL, info.transient_device_id.size());
145}
146
147TEST_F(MTPDeviceDelegateImplWinTest, GalleryNameMTP) {
148  base::FilePath location(
149      PortableDeviceWatcherWin::GetStoragePathFromStorageId(
150          TestPortableDeviceWatcherWin::kStorageUniqueIdA));
151  AttachDevice(StorageInfo::MTP_OR_PTP, "mtp_fake_id", location);
152
153  content::RenderViewHost* rvh = web_contents()->GetRenderViewHost();
154  FSInfoMap results;
155  MediaFileSystemRegistry* registry =
156      g_browser_process->media_file_system_registry();
157  registry->GetMediaFileSystemsForExtension(
158      rvh, extension_.get(),
159      base::Bind(&GetGalleryInfoCallback, base::Unretained(&results)));
160  base::RunLoop().RunUntilIdle();
161
162  ASSERT_EQ(media_directories_.num_galleries() + 1, results.size());
163  bool checked = false;
164  for (FSInfoMap::iterator i = results.begin(); i != results.end(); ++i) {
165    MediaFileSystemInfo info = i->second;
166    if (info.path == location) {
167      CheckGalleryInfo(info, location.LossyDisplayName(), location, true, true);
168      checked = true;
169      break;
170    }
171  }
172  EXPECT_TRUE(checked);
173}
174