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.
5 #include "chrome/browser/media_galleries/media_folder_finder.h"
10 #include "base/base_paths.h"
11 #include "base/bind.h"
12 #include "base/files/file_util.h"
13 #include "base/files/scoped_temp_dir.h"
14 #include "base/strings/stringprintf.h"
15 #include "base/test/scoped_path_override.h"
16 #include "base/threading/sequenced_worker_pool.h"
17 #include "chrome/browser/media_galleries/media_scan_types.h"
18 #include "chrome/common/chrome_paths.h"
19 #include "content/public/browser/browser_thread.h"
20 #include "content/public/test/test_browser_thread_bundle.h"
21 #include "content/public/test/test_utils.h"
23 class MediaFolderFinderTest : public testing::Test {
25 MediaFolderFinderTest() {
28 virtual ~MediaFolderFinderTest() {
31 virtual void SetUp() OVERRIDE {
32 ASSERT_TRUE(fake_dir_.CreateUniqueTempDir());
35 virtual void TearDown() OVERRIDE {
36 ASSERT_EQ(NULL, media_folder_finder_.get());
40 void CreateMediaFolderFinder(
41 const std::vector<base::FilePath> roots,
42 bool expected_success,
43 const MediaFolderFinder::MediaFolderFinderResults& expected_results) {
44 EXPECT_EQ(NULL, media_folder_finder_.get());
45 received_results_ = false;
46 expected_success_ = expected_success;
47 expected_results_ = expected_results;
48 media_folder_finder_.reset(
49 new MediaFolderFinder(base::Bind(&MediaFolderFinderTest::OnGotResults,
50 base::Unretained(this))));
51 media_folder_finder_->SetRootsForTesting(roots);
55 media_folder_finder_->StartScan();
58 void DeleteMediaFolderFinder() {
59 EXPECT_TRUE(media_folder_finder_.get() != NULL);
60 media_folder_finder_.reset();
63 bool received_results() const {
64 return received_results_;
67 const base::FilePath& fake_dir() const {
68 return fake_dir_.path();
71 void CreateTestDir(const base::FilePath& parent_dir) {
72 if (parent_dir == fake_dir())
75 ASSERT_TRUE(fake_dir().IsParent(parent_dir));
76 ASSERT_TRUE(base::CreateDirectory(parent_dir));
79 void CreateTestFile(const base::FilePath& parent_dir,
80 MediaGalleryScanFileType type,
83 MediaFolderFinder::MediaFolderFinderResults* results) {
84 CreateTestDir(parent_dir);
86 std::string extension;
88 MediaGalleryScanResult& result = (*results)[parent_dir];
90 case MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE:
93 result.image_count += count;
95 case MEDIA_GALLERY_SCAN_FILE_TYPE_AUDIO:
98 result.audio_count += count;
100 case MEDIA_GALLERY_SCAN_FILE_TYPE_VIDEO:
103 result.video_count += count;
105 case MEDIA_GALLERY_SCAN_FILE_TYPE_UNKNOWN:
108 if (IsEmptyScanResult(result))
109 results->erase(parent_dir);
118 for (size_t i = 0; i < count; ++i) {
119 base::FilePath test_file(parent_dir.AppendASCII("dummy." + extension));
121 base::GetUniquePathNumber(test_file, base::FilePath::StringType());
122 if (uniquifier > 0) {
123 test_file = test_file.InsertBeforeExtensionASCII(
124 base::StringPrintf(" (%d)", uniquifier));
125 filesize += uniquifier;
128 std::string dummy_data;
129 dummy_data.resize(filesize);
132 base::WriteFile(test_file, dummy_data.c_str(), filesize);
133 ASSERT_GE(bytes_written, 0);
134 ASSERT_EQ(filesize, static_cast<size_t>(bytes_written));
138 void RunLoopUntilReceivedCallback() {
139 while (!received_results())
140 content::RunAllBlockingPoolTasksUntilIdle();
146 const MediaFolderFinder::MediaFolderFinderResults& results) {
147 received_results_ = true;
148 EXPECT_EQ(expected_success_, success);
149 std::set<base::FilePath> expected_keys =
150 GetKeysFromResults(expected_results_);
151 ASSERT_EQ(expected_keys, GetKeysFromResults(results));
152 for (MediaFolderFinder::MediaFolderFinderResults::const_iterator it =
154 it != results.end(); ++it) {
155 const base::FilePath& folder = it->first;
156 const MediaGalleryScanResult& expected = it->second;
157 const MediaGalleryScanResult& actual = results.find(folder)->second;
158 EXPECT_EQ(expected.image_count, actual.image_count)
159 << " Image count for " << folder.value();
160 EXPECT_EQ(expected.audio_count, actual.audio_count)
161 << " Audio count for " << folder.value();
162 EXPECT_EQ(expected.video_count, actual.video_count)
163 << " Video count for " << folder.value();
167 std::set<base::FilePath> GetKeysFromResults(
168 const MediaFolderFinder::MediaFolderFinderResults& results) {
169 std::set<base::FilePath> keys;
170 for (MediaFolderFinder::MediaFolderFinderResults::const_iterator it =
172 it != results.end(); ++it) {
173 keys.insert(it->first);
178 content::TestBrowserThreadBundle thread_bundle_;
180 base::ScopedTempDir fake_dir_;
182 scoped_ptr<MediaFolderFinder> media_folder_finder_;
184 bool expected_success_;
185 MediaFolderFinder::MediaFolderFinderResults expected_results_;
186 bool received_results_;
188 DISALLOW_COPY_AND_ASSIGN(MediaFolderFinderTest);
191 TEST_F(MediaFolderFinderTest, NoScan) {
192 MediaFolderFinder::MediaFolderFinderResults expected_results;
193 std::vector<base::FilePath> folders;
194 folders.push_back(fake_dir());
195 CreateMediaFolderFinder(folders, false, expected_results);
196 DeleteMediaFolderFinder();
197 EXPECT_TRUE(received_results());
200 TEST_F(MediaFolderFinderTest, ScanAndCancel) {
201 MediaFolderFinder::MediaFolderFinderResults expected_results;
202 std::vector<base::FilePath> folders;
203 folders.push_back(fake_dir());
204 CreateMediaFolderFinder(folders, false, expected_results);
206 DeleteMediaFolderFinder();
207 content::RunAllBlockingPoolTasksUntilIdle();
208 EXPECT_TRUE(received_results());
211 TEST_F(MediaFolderFinderTest, ScanNothing) {
212 MediaFolderFinder::MediaFolderFinderResults expected_results;
213 std::vector<base::FilePath> folders;
214 CreateMediaFolderFinder(folders, true, expected_results);
216 RunLoopUntilReceivedCallback();
217 DeleteMediaFolderFinder();
220 TEST_F(MediaFolderFinderTest, EmptyScan) {
221 MediaFolderFinder::MediaFolderFinderResults expected_results;
222 std::vector<base::FilePath> folders;
223 folders.push_back(fake_dir());
224 CreateMediaFolderFinder(folders, true, expected_results);
226 RunLoopUntilReceivedCallback();
227 DeleteMediaFolderFinder();
230 TEST_F(MediaFolderFinderTest, ScanMediaFiles) {
231 MediaFolderFinder::MediaFolderFinderResults expected_results;
232 std::vector<base::FilePath> folders;
233 folders.push_back(fake_dir());
235 base::FilePath dir1 = fake_dir().AppendASCII("dir1");
236 base::FilePath dir2 = fake_dir().AppendASCII("dir2");
237 base::FilePath dir2_3 = dir2.AppendASCII("dir2_3");
238 base::FilePath dir2_4 = dir2.AppendASCII("dir2_4");
239 base::FilePath dir2_4_5 = dir2_4.AppendASCII("dir2_4_5");
240 base::FilePath dir2_4_empty = dir2_4.AppendASCII("dir2_4_empty");
241 base::FilePath dir_empty = fake_dir().AppendASCII("dir_empty");
243 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 2, true,
245 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, false,
247 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_UNKNOWN, 1, false,
249 CreateTestFile(dir2_3, MEDIA_GALLERY_SCAN_FILE_TYPE_AUDIO, 4, true,
251 CreateTestFile(dir2_3, MEDIA_GALLERY_SCAN_FILE_TYPE_AUDIO, 3, false,
253 CreateTestFile(dir2_4, MEDIA_GALLERY_SCAN_FILE_TYPE_UNKNOWN, 5, false,
255 CreateTestFile(dir2_4_5, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 2, true,
257 CreateTestFile(dir2_4_5, MEDIA_GALLERY_SCAN_FILE_TYPE_AUDIO, 4, true,
259 CreateTestFile(dir2_4_5, MEDIA_GALLERY_SCAN_FILE_TYPE_VIDEO, 1, true,
261 CreateTestFile(dir2_4_5, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 5, false,
263 CreateTestFile(dir2_4_5, MEDIA_GALLERY_SCAN_FILE_TYPE_VIDEO, 3, false,
265 CreateTestFile(dir2_4_5, MEDIA_GALLERY_SCAN_FILE_TYPE_UNKNOWN, 3, true,
267 CreateTestDir(dir2_4_empty);
268 CreateTestDir(dir_empty);
270 CreateMediaFolderFinder(folders, true, expected_results);
272 RunLoopUntilReceivedCallback();
273 DeleteMediaFolderFinder();
276 TEST_F(MediaFolderFinderTest, SkipHiddenFiles) {
277 MediaFolderFinder::MediaFolderFinderResults expected_results;
278 std::vector<base::FilePath> folders;
279 folders.push_back(fake_dir());
281 base::FilePath hidden_dir = fake_dir().AppendASCII(".hidden");
283 CreateTestFile(hidden_dir, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 2, true,
285 expected_results.erase(hidden_dir);
287 CreateMediaFolderFinder(folders, true, expected_results);
289 RunLoopUntilReceivedCallback();
290 DeleteMediaFolderFinder();
293 TEST_F(MediaFolderFinderTest, ScanIgnoresSmallMediaFiles) {
294 MediaFolderFinder::MediaFolderFinderResults expected_results;
295 std::vector<base::FilePath> folders;
296 folders.push_back(fake_dir());
298 base::FilePath dir1 = fake_dir().AppendASCII("dir1");
299 base::FilePath dir2 = fake_dir().AppendASCII("dir2");
300 base::FilePath dir_empty = fake_dir().AppendASCII("dir_empty");
302 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 2, true,
304 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_VIDEO, 1, false,
306 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_UNKNOWN, 1, false,
308 CreateTestFile(dir2, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, false,
310 CreateTestFile(dir2, MEDIA_GALLERY_SCAN_FILE_TYPE_AUDIO, 3, false,
312 CreateTestFile(dir2, MEDIA_GALLERY_SCAN_FILE_TYPE_VIDEO, 5, false,
314 CreateTestFile(dir2, MEDIA_GALLERY_SCAN_FILE_TYPE_UNKNOWN, 1, true,
316 CreateTestDir(dir_empty);
317 ASSERT_EQ(1U, expected_results.erase(dir2));
319 CreateMediaFolderFinder(folders, true, expected_results);
321 RunLoopUntilReceivedCallback();
322 DeleteMediaFolderFinder();
325 TEST_F(MediaFolderFinderTest, Overlap) {
326 MediaFolderFinder::MediaFolderFinderResults expected_results;
327 std::vector<base::FilePath> folders;
328 folders.push_back(fake_dir());
329 folders.push_back(fake_dir());
331 base::FilePath dir1 = fake_dir().AppendASCII("dir1");
332 folders.push_back(dir1);
333 folders.push_back(dir1);
335 CreateTestFile(dir1, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
338 CreateMediaFolderFinder(folders, true, expected_results);
340 RunLoopUntilReceivedCallback();
341 DeleteMediaFolderFinder();
344 TEST_F(MediaFolderFinderTest, Prune) {
345 MediaFolderFinder::MediaFolderFinderResults expected_results;
346 std::vector<base::FilePath> folders;
347 folders.push_back(fake_dir());
350 int pruned_dir_key = base::DIR_IE_INTERNET_CACHE;
351 #elif defined(OS_MACOSX)
352 int pruned_dir_key = chrome::DIR_USER_LIBRARY;
354 int pruned_dir_key = base::DIR_CACHE;
357 base::FilePath fake_pruned_dir = fake_dir().AppendASCII("dir1");
358 base::ScopedPathOverride scoped_fake_pruned_dir_override(pruned_dir_key,
361 CreateTestFile(fake_dir(), MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
363 CreateTestFile(fake_pruned_dir, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
366 base::FilePath test_dir = fake_pruned_dir.AppendASCII("dir2");
367 CreateTestFile(test_dir, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
370 // |fake_pruned_dir| and |test_dir| are pruned.
371 expected_results.erase(fake_pruned_dir);
372 expected_results.erase(test_dir);
374 CreateMediaFolderFinder(folders, true, expected_results);
376 RunLoopUntilReceivedCallback();
377 DeleteMediaFolderFinder();
380 TEST_F(MediaFolderFinderTest, Graylist) {
381 MediaFolderFinder::MediaFolderFinderResults expected_results;
382 std::vector<base::FilePath> folders;
383 folders.push_back(fake_dir());
385 base::FilePath fake_home_dir = fake_dir().AppendASCII("dir1");
386 base::FilePath test_dir = fake_home_dir.AppendASCII("dir2");
387 base::ScopedPathOverride scoped_fake_home_dir_override(base::DIR_HOME,
390 CreateTestFile(fake_dir(), MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
392 CreateTestFile(fake_home_dir, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
394 CreateTestFile(test_dir, MEDIA_GALLERY_SCAN_FILE_TYPE_IMAGE, 1, true,
397 // |fake_home_dir| and its ancestors do not show up in results.
398 expected_results.erase(fake_dir());
399 expected_results.erase(fake_home_dir);
401 CreateMediaFolderFinder(folders, true, expected_results);
403 RunLoopUntilReceivedCallback();
404 DeleteMediaFolderFinder();