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.
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"
33 typedef std::map<MediaGalleryPrefId, MediaFileSystemInfo> FSInfoMap;
35 void GetGalleryInfoCallback(
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];
46 class MTPDeviceDelegateImplWinTest : public ChromeRenderViewHostTestHarness {
48 void SetUp() OVERRIDE;
49 void TearDown() OVERRIDE;
51 void ProcessAttach(const std::string& id,
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,
59 const base::FilePath& path,
63 // Pointer to the storage monitor. Owned by TestingBrowserProcess.
64 TestStorageMonitorWin* monitor_;
65 scoped_refptr<extensions::Extension> extension_;
67 EnsureMediaDirectoriesExists media_directories_;
70 void MTPDeviceDelegateImplWinTest::SetUp() {
71 ChromeRenderViewHostTestHarness::SetUp();
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());
85 base::RunLoop runloop;
86 browser_process->media_file_system_registry()->GetPreferences(profile())->
87 EnsureInitialized(runloop.QuitClosure());
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);
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());
102 void MTPDeviceDelegateImplWinTest::TearDown() {
103 // Windows storage monitor must be destroyed on the same thread
105 TestStorageMonitor::RemoveSingleton();
107 ChromeRenderViewHostTestHarness::TearDown();
110 void 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);
118 std::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();
130 void MTPDeviceDelegateImplWinTest::CheckGalleryInfo(
131 const MediaFileSystemInfo& info,
132 const string16& name,
133 const base::FilePath& path,
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);
143 EXPECT_NE(0UL, info.transient_device_id.size());
145 EXPECT_EQ(0UL, info.transient_device_id.size());
148 TEST_F(MTPDeviceDelegateImplWinTest, GalleryNameMTP) {
149 base::FilePath location(
150 PortableDeviceWatcherWin::GetStoragePathFromStorageId(
151 TestPortableDeviceWatcherWin::kStorageUniqueIdA));
152 AttachDevice(StorageInfo::MTP_OR_PTP, "mtp_fake_id", location);
154 content::RenderViewHost* rvh = web_contents()->GetRenderViewHost();
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();
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);
173 EXPECT_TRUE(checked);