OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/chromeos/drive/file_system/search_operation.h" | 5 #include "chrome/browser/chromeos/drive/file_system/search_operation.h" |
6 | 6 |
7 #include "chrome/browser/chromeos/drive/file_system/operation_test_base.h" | 7 #include "chrome/browser/chromeos/drive/file_system/operation_test_base.h" |
8 #include "chrome/browser/drive/fake_drive_service.h" | 8 #include "chrome/browser/drive/fake_drive_service.h" |
9 #include "google_apis/drive/gdata_wapi_parser.h" | 9 #include "google_apis/drive/gdata_wapi_parser.h" |
10 #include "google_apis/drive/test_util.h" | 10 #include "google_apis/drive/test_util.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
12 | 12 |
13 namespace drive { | 13 namespace drive { |
14 namespace file_system { | 14 namespace file_system { |
15 | 15 |
16 namespace { | |
17 | |
18 struct SearchResultPair { | |
19 const char* path; | |
20 const bool is_directory; | |
21 }; | |
22 | |
23 } // namespace | |
24 | |
25 typedef OperationTestBase SearchOperationTest; | 16 typedef OperationTestBase SearchOperationTest; |
26 | 17 |
27 TEST_F(SearchOperationTest, ContentSearch) { | 18 TEST_F(SearchOperationTest, ContentSearch) { |
28 SearchOperation operation(blocking_task_runner(), scheduler(), metadata()); | 19 SearchOperation operation(blocking_task_runner(), scheduler(), metadata()); |
29 | 20 |
30 const SearchResultPair kExpectedResults[] = { | 21 std::set<std::string> expected_results; |
31 { "drive/root/Directory 1/Sub Directory Folder/Sub Sub Directory Folder", | 22 expected_results.insert( |
32 true }, | 23 "drive/root/Directory 1/Sub Directory Folder/Sub Sub Directory Folder"); |
33 { "drive/root/Directory 1/Sub Directory Folder", true }, | 24 expected_results.insert("drive/root/Directory 1/Sub Directory Folder"); |
34 { "drive/root/Directory 1/SubDirectory File 1.txt", false }, | 25 expected_results.insert("drive/root/Directory 1/SubDirectory File 1.txt"); |
35 { "drive/root/Directory 1", true }, | 26 expected_results.insert("drive/root/Directory 1"); |
36 { "drive/root/Directory 2 excludeDir-test", true }, | 27 expected_results.insert("drive/root/Directory 2 excludeDir-test"); |
37 }; | |
38 | 28 |
39 FileError error = FILE_ERROR_FAILED; | 29 FileError error = FILE_ERROR_FAILED; |
40 GURL next_link; | 30 GURL next_link; |
41 scoped_ptr<std::vector<SearchResultInfo> > results; | 31 scoped_ptr<std::vector<SearchResultInfo> > results; |
42 | 32 |
43 operation.Search("Directory", GURL(), | 33 operation.Search("Directory", GURL(), |
44 google_apis::test_util::CreateCopyResultCallback( | 34 google_apis::test_util::CreateCopyResultCallback( |
45 &error, &next_link, &results)); | 35 &error, &next_link, &results)); |
46 test_util::RunBlockingPoolTask(); | 36 test_util::RunBlockingPoolTask(); |
47 | 37 |
48 EXPECT_EQ(FILE_ERROR_OK, error); | 38 EXPECT_EQ(FILE_ERROR_OK, error); |
49 EXPECT_TRUE(next_link.is_empty()); | 39 EXPECT_TRUE(next_link.is_empty()); |
50 EXPECT_EQ(ARRAYSIZE_UNSAFE(kExpectedResults), results->size()); | 40 EXPECT_EQ(expected_results.size(), results->size()); |
51 for (size_t i = 0; i < results->size(); i++) { | 41 for (size_t i = 0; i < results->size(); i++) { |
52 EXPECT_EQ(kExpectedResults[i].path, results->at(i).path.AsUTF8Unsafe()); | 42 EXPECT_TRUE(expected_results.count(results->at(i).path.AsUTF8Unsafe())) |
53 EXPECT_EQ(kExpectedResults[i].is_directory, results->at(i).is_directory); | 43 << results->at(i).path.AsUTF8Unsafe(); |
54 } | 44 } |
55 } | 45 } |
56 | 46 |
57 TEST_F(SearchOperationTest, ContentSearchWithNewEntry) { | 47 TEST_F(SearchOperationTest, ContentSearchWithNewEntry) { |
58 SearchOperation operation(blocking_task_runner(), scheduler(), metadata()); | 48 SearchOperation operation(blocking_task_runner(), scheduler(), metadata()); |
59 | 49 |
60 // Create a new directory in the drive service. | 50 // Create a new directory in the drive service. |
61 google_apis::GDataErrorCode gdata_error = google_apis::GDATA_OTHER_ERROR; | 51 google_apis::GDataErrorCode gdata_error = google_apis::GDATA_OTHER_ERROR; |
62 scoped_ptr<google_apis::ResourceEntry> resource_entry; | 52 scoped_ptr<google_apis::ResourceEntry> resource_entry; |
63 fake_service()->AddNewDirectory( | 53 fake_service()->AddNewDirectory( |
64 fake_service()->GetRootResourceId(), | 54 fake_service()->GetRootResourceId(), |
65 "New Directory 1!", | 55 "New Directory 1!", |
66 google_apis::test_util::CreateCopyResultCallback( | 56 google_apis::test_util::CreateCopyResultCallback( |
67 &gdata_error, &resource_entry)); | 57 &gdata_error, &resource_entry)); |
68 test_util::RunBlockingPoolTask(); | 58 test_util::RunBlockingPoolTask(); |
69 ASSERT_EQ(google_apis::HTTP_CREATED, gdata_error); | 59 ASSERT_EQ(google_apis::HTTP_CREATED, gdata_error); |
70 | 60 |
71 // As the result of the first Search(), only entries in the current file | 61 // As the result of the first Search(), only entries in the current file |
72 // system snapshot are expected to be returned in the "right" path. New | 62 // system snapshot are expected to be returned in the "right" path. New |
73 // entries like "New Directory 1!" is temporarily added to "drive/other". | 63 // entries like "New Directory 1!" is temporarily added to "drive/other". |
74 const SearchResultPair kExpectedResultsBeforeLoad[] = { | 64 std::set<std::string> expected_results; |
75 { "drive/root/Directory 1", true }, | 65 expected_results.insert("drive/root/Directory 1"); |
76 { "drive/other/New Directory 1!", true }, | 66 expected_results.insert("drive/other/New Directory 1!"); |
77 }; | |
78 | 67 |
79 FileError error = FILE_ERROR_FAILED; | 68 FileError error = FILE_ERROR_FAILED; |
80 GURL next_link; | 69 GURL next_link; |
81 scoped_ptr<std::vector<SearchResultInfo> > results; | 70 scoped_ptr<std::vector<SearchResultInfo> > results; |
82 | 71 |
83 operation.Search("\"Directory 1\"", GURL(), | 72 operation.Search("\"Directory 1\"", GURL(), |
84 google_apis::test_util::CreateCopyResultCallback( | 73 google_apis::test_util::CreateCopyResultCallback( |
85 &error, &next_link, &results)); | 74 &error, &next_link, &results)); |
86 test_util::RunBlockingPoolTask(); | 75 test_util::RunBlockingPoolTask(); |
87 | 76 |
88 EXPECT_EQ(FILE_ERROR_OK, error); | 77 EXPECT_EQ(FILE_ERROR_OK, error); |
89 EXPECT_TRUE(next_link.is_empty()); | 78 EXPECT_TRUE(next_link.is_empty()); |
90 ASSERT_EQ(ARRAYSIZE_UNSAFE(kExpectedResultsBeforeLoad), results->size()); | 79 ASSERT_EQ(expected_results.size(), results->size()); |
91 for (size_t i = 0; i < results->size(); i++) { | 80 for (size_t i = 0; i < results->size(); i++) { |
92 EXPECT_EQ(kExpectedResultsBeforeLoad[i].path, | 81 EXPECT_TRUE(expected_results.count(results->at(i).path.AsUTF8Unsafe())) |
93 results->at(i).path.AsUTF8Unsafe()); | 82 << results->at(i).path.AsUTF8Unsafe(); |
94 EXPECT_EQ(kExpectedResultsBeforeLoad[i].is_directory, | |
95 results->at(i).is_directory); | |
96 } | 83 } |
97 | 84 |
98 // Load the change from FakeDriveService. | 85 // Load the change from FakeDriveService. |
99 ASSERT_EQ(FILE_ERROR_OK, CheckForUpdates()); | 86 ASSERT_EQ(FILE_ERROR_OK, CheckForUpdates()); |
100 | 87 |
101 // Now the new entry must be reported to be in the right directory. | 88 // Now the new entry must be reported to be in the right directory. |
102 const SearchResultPair kExpectedResultsAfterLoad[] = { | 89 expected_results.clear(); |
103 { "drive/root/Directory 1", true }, | 90 expected_results.insert("drive/root/Directory 1"); |
104 { "drive/root/New Directory 1!", true }, | 91 expected_results.insert("drive/root/New Directory 1!"); |
105 }; | |
106 error = FILE_ERROR_FAILED; | 92 error = FILE_ERROR_FAILED; |
107 operation.Search("\"Directory 1\"", GURL(), | 93 operation.Search("\"Directory 1\"", GURL(), |
108 google_apis::test_util::CreateCopyResultCallback( | 94 google_apis::test_util::CreateCopyResultCallback( |
109 &error, &next_link, &results)); | 95 &error, &next_link, &results)); |
110 test_util::RunBlockingPoolTask(); | 96 test_util::RunBlockingPoolTask(); |
111 | 97 |
112 EXPECT_EQ(FILE_ERROR_OK, error); | 98 EXPECT_EQ(FILE_ERROR_OK, error); |
113 EXPECT_TRUE(next_link.is_empty()); | 99 EXPECT_TRUE(next_link.is_empty()); |
114 ASSERT_EQ(ARRAYSIZE_UNSAFE(kExpectedResultsAfterLoad), results->size()); | 100 ASSERT_EQ(expected_results.size(), results->size()); |
115 for (size_t i = 0; i < results->size(); i++) { | 101 for (size_t i = 0; i < results->size(); i++) { |
116 EXPECT_EQ(kExpectedResultsAfterLoad[i].path, | 102 EXPECT_TRUE(expected_results.count(results->at(i).path.AsUTF8Unsafe())) |
117 results->at(i).path.AsUTF8Unsafe()); | 103 << results->at(i).path.AsUTF8Unsafe(); |
118 EXPECT_EQ(kExpectedResultsAfterLoad[i].is_directory, | |
119 results->at(i).is_directory); | |
120 } | 104 } |
121 } | 105 } |
122 | 106 |
123 TEST_F(SearchOperationTest, ContentSearchEmptyResult) { | 107 TEST_F(SearchOperationTest, ContentSearchEmptyResult) { |
124 SearchOperation operation(blocking_task_runner(), scheduler(), metadata()); | 108 SearchOperation operation(blocking_task_runner(), scheduler(), metadata()); |
125 | 109 |
126 FileError error = FILE_ERROR_FAILED; | 110 FileError error = FILE_ERROR_FAILED; |
127 GURL next_link; | 111 GURL next_link; |
128 scoped_ptr<std::vector<SearchResultInfo> > results; | 112 scoped_ptr<std::vector<SearchResultInfo> > results; |
129 | 113 |
130 operation.Search("\"no-match query\"", GURL(), | 114 operation.Search("\"no-match query\"", GURL(), |
131 google_apis::test_util::CreateCopyResultCallback( | 115 google_apis::test_util::CreateCopyResultCallback( |
132 &error, &next_link, &results)); | 116 &error, &next_link, &results)); |
133 test_util::RunBlockingPoolTask(); | 117 test_util::RunBlockingPoolTask(); |
134 | 118 |
135 EXPECT_EQ(FILE_ERROR_OK, error); | 119 EXPECT_EQ(FILE_ERROR_OK, error); |
136 EXPECT_TRUE(next_link.is_empty()); | 120 EXPECT_TRUE(next_link.is_empty()); |
137 EXPECT_EQ(0U, results->size()); | 121 EXPECT_EQ(0U, results->size()); |
138 } | 122 } |
139 | 123 |
140 } // namespace file_system | 124 } // namespace file_system |
141 } // namespace drive | 125 } // namespace drive |
OLD | NEW |