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.
5 #include "chrome/browser/chromeos/drive/file_system/search_operation.h"
7 #include "base/callback_helpers.h"
8 #include "chrome/browser/chromeos/drive/change_list_loader.h"
9 #include "chrome/browser/chromeos/drive/file_system/operation_test_base.h"
10 #include "components/drive/service/fake_drive_service.h"
11 #include "content/public/test/test_utils.h"
12 #include "google_apis/drive/drive_api_parser.h"
13 #include "google_apis/drive/test_util.h"
14 #include "testing/gtest/include/gtest/gtest.h"
17 namespace file_system
{
19 typedef OperationTestBase SearchOperationTest
;
21 TEST_F(SearchOperationTest
, ContentSearch
) {
22 SearchOperation
operation(blocking_task_runner(), scheduler(), metadata(),
25 std::set
<std::string
> expected_results
;
26 expected_results
.insert(
27 "drive/root/Directory 1/Sub Directory Folder/Sub Sub Directory Folder");
28 expected_results
.insert("drive/root/Directory 1/Sub Directory Folder");
29 expected_results
.insert("drive/root/Directory 1/SubDirectory File 1.txt");
30 expected_results
.insert("drive/root/Directory 1");
31 expected_results
.insert("drive/root/Directory 2 excludeDir-test");
33 FileError error
= FILE_ERROR_FAILED
;
35 scoped_ptr
<std::vector
<SearchResultInfo
> > results
;
37 operation
.Search("Directory", GURL(),
38 google_apis::test_util::CreateCopyResultCallback(
39 &error
, &next_link
, &results
));
40 content::RunAllBlockingPoolTasksUntilIdle();
42 EXPECT_EQ(FILE_ERROR_OK
, error
);
43 EXPECT_TRUE(next_link
.is_empty());
44 EXPECT_EQ(expected_results
.size(), results
->size());
45 for (size_t i
= 0; i
< results
->size(); i
++) {
46 EXPECT_TRUE(expected_results
.count(results
->at(i
).path
.AsUTF8Unsafe()))
47 << results
->at(i
).path
.AsUTF8Unsafe();
51 TEST_F(SearchOperationTest
, ContentSearchWithNewEntry
) {
52 SearchOperation
operation(blocking_task_runner(), scheduler(), metadata(),
55 // Create a new directory in the drive service.
56 google_apis::DriveApiErrorCode status
= google_apis::DRIVE_OTHER_ERROR
;
57 scoped_ptr
<google_apis::FileResource
> server_entry
;
58 fake_service()->AddNewDirectory(
59 fake_service()->GetRootResourceId(), "New Directory 1!",
60 AddNewDirectoryOptions(),
61 google_apis::test_util::CreateCopyResultCallback(&status
, &server_entry
));
62 content::RunAllBlockingPoolTasksUntilIdle();
63 ASSERT_EQ(google_apis::HTTP_CREATED
, status
);
65 // As the result of the first Search(), only entries in the current file
66 // system snapshot are expected to be returned in the "right" path. New
67 // entries like "New Directory 1!" is temporarily added to "drive/other".
68 std::set
<std::string
> expected_results
;
69 expected_results
.insert("drive/root/Directory 1");
70 expected_results
.insert("drive/other/New Directory 1!");
72 FileError error
= FILE_ERROR_FAILED
;
74 scoped_ptr
<std::vector
<SearchResultInfo
> > results
;
76 operation
.Search("\"Directory 1\"", GURL(),
77 google_apis::test_util::CreateCopyResultCallback(
78 &error
, &next_link
, &results
));
79 content::RunAllBlockingPoolTasksUntilIdle();
81 EXPECT_EQ(FILE_ERROR_OK
, error
);
82 EXPECT_TRUE(next_link
.is_empty());
83 ASSERT_EQ(expected_results
.size(), results
->size());
84 for (size_t i
= 0; i
< results
->size(); i
++) {
85 EXPECT_TRUE(expected_results
.count(results
->at(i
).path
.AsUTF8Unsafe()))
86 << results
->at(i
).path
.AsUTF8Unsafe();
89 // Load the change from FakeDriveService.
90 ASSERT_EQ(FILE_ERROR_OK
, CheckForUpdates());
92 // Now the new entry must be reported to be in the right directory.
93 expected_results
.clear();
94 expected_results
.insert("drive/root/Directory 1");
95 expected_results
.insert("drive/root/New Directory 1!");
96 error
= FILE_ERROR_FAILED
;
97 operation
.Search("\"Directory 1\"", GURL(),
98 google_apis::test_util::CreateCopyResultCallback(
99 &error
, &next_link
, &results
));
100 content::RunAllBlockingPoolTasksUntilIdle();
102 EXPECT_EQ(FILE_ERROR_OK
, error
);
103 EXPECT_TRUE(next_link
.is_empty());
104 ASSERT_EQ(expected_results
.size(), results
->size());
105 for (size_t i
= 0; i
< results
->size(); i
++) {
106 EXPECT_TRUE(expected_results
.count(results
->at(i
).path
.AsUTF8Unsafe()))
107 << results
->at(i
).path
.AsUTF8Unsafe();
111 TEST_F(SearchOperationTest
, ContentSearchEmptyResult
) {
112 SearchOperation
operation(blocking_task_runner(), scheduler(), metadata(),
113 loader_controller());
115 FileError error
= FILE_ERROR_FAILED
;
117 scoped_ptr
<std::vector
<SearchResultInfo
> > results
;
119 operation
.Search("\"no-match query\"", GURL(),
120 google_apis::test_util::CreateCopyResultCallback(
121 &error
, &next_link
, &results
));
122 content::RunAllBlockingPoolTasksUntilIdle();
124 EXPECT_EQ(FILE_ERROR_OK
, error
);
125 EXPECT_TRUE(next_link
.is_empty());
126 EXPECT_EQ(0U, results
->size());
129 TEST_F(SearchOperationTest
, Lock
) {
130 SearchOperation
operation(blocking_task_runner(), scheduler(), metadata(),
131 loader_controller());
134 scoped_ptr
<base::ScopedClosureRunner
> lock
= loader_controller()->GetLock();
136 // Search does not return the result as long as lock is alive.
137 FileError error
= FILE_ERROR_FAILED
;
139 scoped_ptr
<std::vector
<SearchResultInfo
> > results
;
141 operation
.Search("\"Directory 1\"", GURL(),
142 google_apis::test_util::CreateCopyResultCallback(
143 &error
, &next_link
, &results
));
144 content::RunAllBlockingPoolTasksUntilIdle();
145 EXPECT_EQ(FILE_ERROR_FAILED
, error
);
146 EXPECT_FALSE(results
);
148 // Unlock, this should resume the pending search.
150 content::RunAllBlockingPoolTasksUntilIdle();
151 EXPECT_EQ(FILE_ERROR_OK
, error
);
152 ASSERT_TRUE(results
);
153 EXPECT_EQ(1u, results
->size());
156 } // namespace file_system