mtp_device_delegate_impl_win_unittest.cc revision 4e180b6a0b4720a9b8e9e959a882386f690f08ff 
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  browser_process->media_file_system_registry()->GetPreferences(profile())->
87      EnsureInitialized(runloop.QuitClosure());
88  runloop.Run();
89
90  extensions::TestExtensionSystem* extension_system(
91      static_cast<extensions::TestExtensionSystem*>(
92          extensions::ExtensionSystem::Get(profile())));
93  extension_system->CreateExtensionService(
94      CommandLine::ForCurrentProcess(), base::FilePath(), false);
95
96  std::vector<std::string> all_permissions;
97  all_permissions.push_back("allAutoDetected");
98  all_permissions.push_back("read");
99  extension_ = AddMediaGalleriesApp("all", all_permissions, profile());
100}
101
102void MTPDeviceDelegateImplWinTest::TearDown() {
103  // Windows storage monitor must be destroyed on the same thread
104  // as construction.
105  TestStorageMonitor::RemoveSingleton();
106
107  ChromeRenderViewHostTestHarness::TearDown();
108}
109
110void MTPDeviceDelegateImplWinTest::ProcessAttach(
111    const std::string& id,
112    const string16& label,
113    const base::FilePath::StringType& location) {
114  StorageInfo info(id, string16(), location, label, string16(), string16(), 0);
115  monitor_->receiver()->ProcessAttach(info);
116}
117
118std::string MTPDeviceDelegateImplWinTest::AttachDevice(
119    StorageInfo::Type type,
120    const std::string& unique_id,
121    const base::FilePath& location) {
122  std::string device_id = StorageInfo::MakeDeviceId(type, unique_id);
123  DCHECK(StorageInfo::IsRemovableDevice(device_id));
124  string16 label = location.LossyDisplayName();
125  ProcessAttach(device_id, label, location.value());
126  base::RunLoop().RunUntilIdle();
127  return device_id;
128}
129
130void MTPDeviceDelegateImplWinTest::CheckGalleryInfo(
131    const MediaFileSystemInfo& info,
132    const string16& name,
133    const base::FilePath& path,
134    bool removable,
135    bool media_device) {
136  EXPECT_EQ(name, info.name);
137  EXPECT_EQ(path, info.path);
138  EXPECT_EQ(removable, info.removable);
139  EXPECT_EQ(media_device, info.media_device);
140  EXPECT_NE(0UL, info.pref_id);
141
142  if (removable)
143    EXPECT_NE(0UL, info.transient_device_id.size());
144  else
145    EXPECT_EQ(0UL, info.transient_device_id.size());
146}
147
148TEST_F(MTPDeviceDelegateImplWinTest, GalleryNameMTP) {
149  base::FilePath location(
150      PortableDeviceWatcherWin::GetStoragePathFromStorageId(
151          TestPortableDeviceWatcherWin::kStorageUniqueIdA));
152  AttachDevice(StorageInfo::MTP_OR_PTP, "mtp_fake_id", location);
153
154  content::RenderViewHost* rvh = web_contents()->GetRenderViewHost();
155  FSInfoMap results;
156  MediaFileSystemRegistry* registry =
157      g_browser_process->media_file_system_registry();
158  registry->GetMediaFileSystemsForExtension(
159      rvh, extension_.get(),
160      base::Bind(&GetGalleryInfoCallback, base::Unretained(&results)));
161  base::RunLoop().RunUntilIdle();
162
163  ASSERT_EQ(media_directories_.num_galleries() + 1, results.size());
164  bool checked = false;
165  for (FSInfoMap::iterator i = results.begin(); i != results.end(); ++i) {
166    MediaFileSystemInfo info = i->second;
167    if (info.path == location) {
168      CheckGalleryInfo(info, location.LossyDisplayName(), location, true, true);
169      checked = true;
170      break;
171    }
172  }
173  EXPECT_TRUE(checked);
174}
175