OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 "chrome/browser/chromeos/drive/sync_client.h" | |
6 | |
7 #include "base/files/file_path.h" | |
8 #include "base/files/file_util.h" | |
9 #include "base/files/scoped_temp_dir.h" | |
10 #include "base/memory/scoped_ptr.h" | |
11 #include "base/prefs/testing_pref_service.h" | |
12 #include "base/run_loop.h" | |
13 #include "base/single_thread_task_runner.h" | |
14 #include "base/test/test_timeouts.h" | |
15 #include "base/thread_task_runner_handle.h" | |
16 #include "components/drive/change_list_loader.h" | |
17 #include "components/drive/drive.pb.h" | |
18 #include "components/drive/drive_test_util.h" | |
19 #include "components/drive/event_logger.h" | |
20 #include "components/drive/fake_free_disk_space_getter.h" | |
21 #include "components/drive/file_cache.h" | |
22 #include "components/drive/file_change.h" | |
23 #include "components/drive/file_system/move_operation.h" | |
24 #include "components/drive/file_system/operation_delegate.h" | |
25 #include "components/drive/file_system/remove_operation.h" | |
26 #include "components/drive/file_system_core_util.h" | |
27 #include "components/drive/job_scheduler.h" | |
28 #include "components/drive/resource_entry_conversion.h" | |
29 #include "components/drive/resource_metadata.h" | |
30 #include "components/drive/service/fake_drive_service.h" | |
31 #include "content/public/test/test_browser_thread_bundle.h" | |
32 #include "google_apis/drive/drive_api_parser.h" | |
33 #include "google_apis/drive/test_util.h" | |
34 #include "testing/gtest/include/gtest/gtest.h" | |
35 | |
36 namespace drive { | |
37 namespace internal { | |
38 | |
39 namespace { | |
40 | |
41 // The content of files initially stored in the cache. | |
42 const char kLocalContent[] = "Hello!"; | |
43 | |
44 // The content of files stored in the service. | |
45 const char kRemoteContent[] = "World!"; | |
46 | |
47 // SyncClientTestDriveService will return DRIVE_CANCELLED when a request is | |
48 // made with the specified resource ID. | |
49 class SyncClientTestDriveService : public ::drive::FakeDriveService { | |
50 public: | |
51 SyncClientTestDriveService() : download_file_count_(0) {} | |
52 | |
53 // FakeDriveService override: | |
54 google_apis::CancelCallback DownloadFile( | |
55 const base::FilePath& local_cache_path, | |
56 const std::string& resource_id, | |
57 const google_apis::DownloadActionCallback& download_action_callback, | |
58 const google_apis::GetContentCallback& get_content_callback, | |
59 const google_apis::ProgressCallback& progress_callback) override { | |
60 ++download_file_count_; | |
61 if (resource_id == resource_id_to_be_cancelled_) { | |
62 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
63 FROM_HERE, | |
64 base::Bind(download_action_callback, | |
65 google_apis::DRIVE_CANCELLED, | |
66 base::FilePath())); | |
67 return google_apis::CancelCallback(); | |
68 } | |
69 if (resource_id == resource_id_to_be_paused_) { | |
70 paused_action_ = base::Bind(download_action_callback, | |
71 google_apis::DRIVE_OTHER_ERROR, | |
72 base::FilePath()); | |
73 return google_apis::CancelCallback(); | |
74 } | |
75 return FakeDriveService::DownloadFile(local_cache_path, | |
76 resource_id, | |
77 download_action_callback, | |
78 get_content_callback, | |
79 progress_callback); | |
80 } | |
81 | |
82 int download_file_count() const { return download_file_count_; } | |
83 | |
84 void set_resource_id_to_be_cancelled(const std::string& resource_id) { | |
85 resource_id_to_be_cancelled_ = resource_id; | |
86 } | |
87 | |
88 void set_resource_id_to_be_paused(const std::string& resource_id) { | |
89 resource_id_to_be_paused_ = resource_id; | |
90 } | |
91 | |
92 const base::Closure& paused_action() const { return paused_action_; } | |
93 | |
94 private: | |
95 int download_file_count_; | |
96 std::string resource_id_to_be_cancelled_; | |
97 std::string resource_id_to_be_paused_; | |
98 base::Closure paused_action_; | |
99 }; | |
100 | |
101 } // namespace | |
102 | |
103 class SyncClientTest : public testing::Test { | |
104 public: | |
105 void SetUp() override { | |
106 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
107 | |
108 pref_service_.reset(new TestingPrefServiceSimple); | |
109 test_util::RegisterDrivePrefs(pref_service_->registry()); | |
110 | |
111 fake_network_change_notifier_.reset( | |
112 new test_util::FakeNetworkChangeNotifier); | |
113 | |
114 logger_.reset(new EventLogger); | |
115 | |
116 drive_service_.reset(new SyncClientTestDriveService); | |
117 | |
118 scheduler_.reset(new JobScheduler( | |
119 pref_service_.get(), | |
120 logger_.get(), | |
121 drive_service_.get(), | |
122 base::ThreadTaskRunnerHandle::Get().get())); | |
123 | |
124 metadata_storage_.reset(new ResourceMetadataStorage( | |
125 temp_dir_.path(), base::ThreadTaskRunnerHandle::Get().get())); | |
126 ASSERT_TRUE(metadata_storage_->Initialize()); | |
127 | |
128 cache_.reset(new FileCache(metadata_storage_.get(), | |
129 temp_dir_.path(), | |
130 base::ThreadTaskRunnerHandle::Get().get(), | |
131 NULL /* free_disk_space_getter */)); | |
132 ASSERT_TRUE(cache_->Initialize()); | |
133 | |
134 metadata_.reset(new internal::ResourceMetadata( | |
135 metadata_storage_.get(), cache_.get(), | |
136 base::ThreadTaskRunnerHandle::Get())); | |
137 ASSERT_EQ(FILE_ERROR_OK, metadata_->Initialize()); | |
138 | |
139 about_resource_loader_.reset(new AboutResourceLoader(scheduler_.get())); | |
140 loader_controller_.reset(new LoaderController); | |
141 change_list_loader_.reset(new ChangeListLoader( | |
142 logger_.get(), | |
143 base::ThreadTaskRunnerHandle::Get().get(), | |
144 metadata_.get(), | |
145 scheduler_.get(), | |
146 about_resource_loader_.get(), | |
147 loader_controller_.get())); | |
148 ASSERT_NO_FATAL_FAILURE(SetUpTestData()); | |
149 | |
150 sync_client_.reset(new SyncClient(base::ThreadTaskRunnerHandle::Get().get(), | |
151 &delegate_, | |
152 scheduler_.get(), | |
153 metadata_.get(), | |
154 cache_.get(), | |
155 loader_controller_.get(), | |
156 temp_dir_.path())); | |
157 | |
158 // Disable delaying so that DoSyncLoop() starts immediately. | |
159 sync_client_->set_delay_for_testing(base::TimeDelta::FromSeconds(0)); | |
160 } | |
161 | |
162 // Adds a file to the service root and |resource_ids_|. | |
163 void AddFileEntry(const std::string& title) { | |
164 google_apis::DriveApiErrorCode error = google_apis::DRIVE_FILE_ERROR; | |
165 scoped_ptr<google_apis::FileResource> entry; | |
166 drive_service_->AddNewFile( | |
167 "text/plain", | |
168 kRemoteContent, | |
169 drive_service_->GetRootResourceId(), | |
170 title, | |
171 false, // shared_with_me | |
172 google_apis::test_util::CreateCopyResultCallback(&error, &entry)); | |
173 base::RunLoop().RunUntilIdle(); | |
174 ASSERT_EQ(google_apis::HTTP_CREATED, error); | |
175 ASSERT_TRUE(entry); | |
176 resource_ids_[title] = entry->file_id(); | |
177 } | |
178 | |
179 // Sets up data for tests. | |
180 void SetUpTestData() { | |
181 // Prepare a temp file. | |
182 base::FilePath temp_file; | |
183 EXPECT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &temp_file)); | |
184 ASSERT_TRUE(google_apis::test_util::WriteStringToFile(temp_file, | |
185 kLocalContent)); | |
186 | |
187 // Add file entries to the service. | |
188 ASSERT_NO_FATAL_FAILURE(AddFileEntry("foo")); | |
189 ASSERT_NO_FATAL_FAILURE(AddFileEntry("bar")); | |
190 ASSERT_NO_FATAL_FAILURE(AddFileEntry("baz")); | |
191 ASSERT_NO_FATAL_FAILURE(AddFileEntry("fetched")); | |
192 ASSERT_NO_FATAL_FAILURE(AddFileEntry("dirty")); | |
193 ASSERT_NO_FATAL_FAILURE(AddFileEntry("removed")); | |
194 ASSERT_NO_FATAL_FAILURE(AddFileEntry("moved")); | |
195 | |
196 // Load data from the service to the metadata. | |
197 FileError error = FILE_ERROR_FAILED; | |
198 change_list_loader_->LoadIfNeeded( | |
199 google_apis::test_util::CreateCopyResultCallback(&error)); | |
200 base::RunLoop().RunUntilIdle(); | |
201 EXPECT_EQ(FILE_ERROR_OK, error); | |
202 | |
203 // Prepare 3 pinned-but-not-present files. | |
204 EXPECT_EQ(FILE_ERROR_OK, cache_->Pin(GetLocalId("foo"))); | |
205 EXPECT_EQ(FILE_ERROR_OK, cache_->Pin(GetLocalId("bar"))); | |
206 EXPECT_EQ(FILE_ERROR_OK, cache_->Pin(GetLocalId("baz"))); | |
207 | |
208 // Prepare a pinned-and-fetched file. | |
209 const std::string md5_fetched = "md5"; | |
210 EXPECT_EQ(FILE_ERROR_OK, | |
211 cache_->Store(GetLocalId("fetched"), md5_fetched, | |
212 temp_file, FileCache::FILE_OPERATION_COPY)); | |
213 EXPECT_EQ(FILE_ERROR_OK, cache_->Pin(GetLocalId("fetched"))); | |
214 | |
215 // Prepare a pinned-and-fetched-and-dirty file. | |
216 EXPECT_EQ(FILE_ERROR_OK, | |
217 cache_->Store(GetLocalId("dirty"), std::string(), | |
218 temp_file, FileCache::FILE_OPERATION_COPY)); | |
219 EXPECT_EQ(FILE_ERROR_OK, cache_->Pin(GetLocalId("dirty"))); | |
220 | |
221 // Prepare a removed file. | |
222 file_system::RemoveOperation remove_operation( | |
223 base::ThreadTaskRunnerHandle::Get().get(), &delegate_, metadata_.get(), | |
224 cache_.get()); | |
225 remove_operation.Remove( | |
226 util::GetDriveMyDriveRootPath().AppendASCII("removed"), | |
227 false, // is_recursive | |
228 google_apis::test_util::CreateCopyResultCallback(&error)); | |
229 base::RunLoop().RunUntilIdle(); | |
230 EXPECT_EQ(FILE_ERROR_OK, error); | |
231 | |
232 // Prepare a moved file. | |
233 file_system::MoveOperation move_operation( | |
234 base::ThreadTaskRunnerHandle::Get().get(), &delegate_, metadata_.get()); | |
235 move_operation.Move( | |
236 util::GetDriveMyDriveRootPath().AppendASCII("moved"), | |
237 util::GetDriveMyDriveRootPath().AppendASCII("moved_new_title"), | |
238 google_apis::test_util::CreateCopyResultCallback(&error)); | |
239 base::RunLoop().RunUntilIdle(); | |
240 EXPECT_EQ(FILE_ERROR_OK, error); | |
241 } | |
242 | |
243 protected: | |
244 std::string GetLocalId(const std::string& title) { | |
245 EXPECT_EQ(1U, resource_ids_.count(title)); | |
246 std::string local_id; | |
247 EXPECT_EQ(FILE_ERROR_OK, | |
248 metadata_->GetIdByResourceId(resource_ids_[title], &local_id)); | |
249 return local_id; | |
250 } | |
251 | |
252 content::TestBrowserThreadBundle thread_bundle_; | |
253 base::ScopedTempDir temp_dir_; | |
254 scoped_ptr<TestingPrefServiceSimple> pref_service_; | |
255 scoped_ptr<test_util::FakeNetworkChangeNotifier> | |
256 fake_network_change_notifier_; | |
257 scoped_ptr<EventLogger> logger_; | |
258 scoped_ptr<SyncClientTestDriveService> drive_service_; | |
259 file_system::OperationDelegate delegate_; | |
260 scoped_ptr<JobScheduler> scheduler_; | |
261 scoped_ptr<ResourceMetadataStorage, | |
262 test_util::DestroyHelperForTests> metadata_storage_; | |
263 scoped_ptr<FileCache, test_util::DestroyHelperForTests> cache_; | |
264 scoped_ptr<ResourceMetadata, test_util::DestroyHelperForTests> metadata_; | |
265 scoped_ptr<AboutResourceLoader> about_resource_loader_; | |
266 scoped_ptr<LoaderController> loader_controller_; | |
267 scoped_ptr<ChangeListLoader> change_list_loader_; | |
268 scoped_ptr<SyncClient> sync_client_; | |
269 | |
270 std::map<std::string, std::string> resource_ids_; // Name-to-id map. | |
271 }; | |
272 | |
273 TEST_F(SyncClientTest, StartProcessingBacklog) { | |
274 sync_client_->StartProcessingBacklog(); | |
275 base::RunLoop().RunUntilIdle(); | |
276 | |
277 ResourceEntry entry; | |
278 // Pinned files get downloaded. | |
279 EXPECT_EQ(FILE_ERROR_OK, | |
280 metadata_->GetResourceEntryById(GetLocalId("foo"), &entry)); | |
281 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | |
282 | |
283 EXPECT_EQ(FILE_ERROR_OK, | |
284 metadata_->GetResourceEntryById(GetLocalId("bar"), &entry)); | |
285 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | |
286 | |
287 EXPECT_EQ(FILE_ERROR_OK, | |
288 metadata_->GetResourceEntryById(GetLocalId("baz"), &entry)); | |
289 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | |
290 | |
291 // Dirty file gets uploaded. | |
292 EXPECT_EQ(FILE_ERROR_OK, | |
293 metadata_->GetResourceEntryById(GetLocalId("dirty"), &entry)); | |
294 EXPECT_FALSE(entry.file_specific_info().cache_state().is_dirty()); | |
295 | |
296 // Removed entry is not found. | |
297 google_apis::DriveApiErrorCode status = google_apis::DRIVE_OTHER_ERROR; | |
298 scoped_ptr<google_apis::FileResource> server_entry; | |
299 drive_service_->GetFileResource( | |
300 resource_ids_["removed"], | |
301 google_apis::test_util::CreateCopyResultCallback(&status, &server_entry)); | |
302 base::RunLoop().RunUntilIdle(); | |
303 EXPECT_EQ(google_apis::HTTP_SUCCESS, status); | |
304 ASSERT_TRUE(server_entry); | |
305 EXPECT_TRUE(server_entry->labels().is_trashed()); | |
306 | |
307 // Moved entry was moved. | |
308 status = google_apis::DRIVE_OTHER_ERROR; | |
309 drive_service_->GetFileResource( | |
310 resource_ids_["moved"], | |
311 google_apis::test_util::CreateCopyResultCallback(&status, &server_entry)); | |
312 base::RunLoop().RunUntilIdle(); | |
313 EXPECT_EQ(google_apis::HTTP_SUCCESS, status); | |
314 ASSERT_TRUE(server_entry); | |
315 EXPECT_EQ("moved_new_title", server_entry->title()); | |
316 } | |
317 | |
318 TEST_F(SyncClientTest, AddFetchTask) { | |
319 sync_client_->AddFetchTask(GetLocalId("foo")); | |
320 base::RunLoop().RunUntilIdle(); | |
321 | |
322 ResourceEntry entry; | |
323 EXPECT_EQ(FILE_ERROR_OK, | |
324 metadata_->GetResourceEntryById(GetLocalId("foo"), &entry)); | |
325 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | |
326 } | |
327 | |
328 TEST_F(SyncClientTest, AddFetchTaskAndCancelled) { | |
329 // Trigger fetching of a file which results in cancellation. | |
330 drive_service_->set_resource_id_to_be_cancelled(resource_ids_["foo"]); | |
331 sync_client_->AddFetchTask(GetLocalId("foo")); | |
332 base::RunLoop().RunUntilIdle(); | |
333 | |
334 // The file should be unpinned if the user wants the download to be cancelled. | |
335 ResourceEntry entry; | |
336 EXPECT_EQ(FILE_ERROR_OK, | |
337 metadata_->GetResourceEntryById(GetLocalId("foo"), &entry)); | |
338 EXPECT_FALSE(entry.file_specific_info().cache_state().is_pinned()); | |
339 } | |
340 | |
341 TEST_F(SyncClientTest, RemoveFetchTask) { | |
342 sync_client_->AddFetchTask(GetLocalId("foo")); | |
343 sync_client_->AddFetchTask(GetLocalId("bar")); | |
344 sync_client_->AddFetchTask(GetLocalId("baz")); | |
345 | |
346 sync_client_->RemoveFetchTask(GetLocalId("foo")); | |
347 sync_client_->RemoveFetchTask(GetLocalId("baz")); | |
348 base::RunLoop().RunUntilIdle(); | |
349 | |
350 // Only "bar" should be fetched. | |
351 ResourceEntry entry; | |
352 EXPECT_EQ(FILE_ERROR_OK, | |
353 metadata_->GetResourceEntryById(GetLocalId("foo"), &entry)); | |
354 EXPECT_FALSE(entry.file_specific_info().cache_state().is_present()); | |
355 | |
356 EXPECT_EQ(FILE_ERROR_OK, | |
357 metadata_->GetResourceEntryById(GetLocalId("bar"), &entry)); | |
358 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | |
359 | |
360 EXPECT_EQ(FILE_ERROR_OK, | |
361 metadata_->GetResourceEntryById(GetLocalId("baz"), &entry)); | |
362 EXPECT_FALSE(entry.file_specific_info().cache_state().is_present()); | |
363 | |
364 } | |
365 | |
366 TEST_F(SyncClientTest, ExistingPinnedFiles) { | |
367 // Start checking the existing pinned files. This will collect the resource | |
368 // IDs of pinned files, with stale local cache files. | |
369 sync_client_->StartCheckingExistingPinnedFiles(); | |
370 base::RunLoop().RunUntilIdle(); | |
371 | |
372 // "fetched" and "dirty" are the existing pinned files. | |
373 // The non-dirty one should be synced, but the dirty one should not. | |
374 base::FilePath cache_file; | |
375 std::string content; | |
376 EXPECT_EQ(FILE_ERROR_OK, cache_->GetFile(GetLocalId("fetched"), &cache_file)); | |
377 EXPECT_TRUE(base::ReadFileToString(cache_file, &content)); | |
378 EXPECT_EQ(kRemoteContent, content); | |
379 content.clear(); | |
380 | |
381 EXPECT_EQ(FILE_ERROR_OK, cache_->GetFile(GetLocalId("dirty"), &cache_file)); | |
382 EXPECT_TRUE(base::ReadFileToString(cache_file, &content)); | |
383 EXPECT_EQ(kLocalContent, content); | |
384 } | |
385 | |
386 TEST_F(SyncClientTest, RetryOnDisconnection) { | |
387 // Let the service go down. | |
388 drive_service_->set_offline(true); | |
389 // Change the network connection state after some delay, to test that | |
390 // FILE_ERROR_NO_CONNECTION is handled by SyncClient correctly. | |
391 // Without this delay, JobScheduler will keep the jobs unrun and SyncClient | |
392 // will receive no error. | |
393 base::ThreadTaskRunnerHandle::Get()->PostDelayedTask( | |
394 FROM_HERE, | |
395 base::Bind(&test_util::FakeNetworkChangeNotifier::SetConnectionType, | |
396 base::Unretained(fake_network_change_notifier_.get()), | |
397 net::NetworkChangeNotifier::CONNECTION_NONE), | |
398 TestTimeouts::tiny_timeout()); | |
399 | |
400 // Try fetch and upload. | |
401 sync_client_->AddFetchTask(GetLocalId("foo")); | |
402 sync_client_->AddUpdateTask(ClientContext(USER_INITIATED), | |
403 GetLocalId("dirty")); | |
404 base::RunLoop().RunUntilIdle(); | |
405 | |
406 // Not yet fetched nor uploaded. | |
407 ResourceEntry entry; | |
408 EXPECT_EQ(FILE_ERROR_OK, | |
409 metadata_->GetResourceEntryById(GetLocalId("foo"), &entry)); | |
410 EXPECT_FALSE(entry.file_specific_info().cache_state().is_present()); | |
411 EXPECT_EQ(FILE_ERROR_OK, | |
412 metadata_->GetResourceEntryById(GetLocalId("dirty"), &entry)); | |
413 EXPECT_TRUE(entry.file_specific_info().cache_state().is_dirty()); | |
414 | |
415 // Switch to online. | |
416 fake_network_change_notifier_->SetConnectionType( | |
417 net::NetworkChangeNotifier::CONNECTION_WIFI); | |
418 drive_service_->set_offline(false); | |
419 base::RunLoop().RunUntilIdle(); | |
420 | |
421 // Fetched and uploaded. | |
422 EXPECT_EQ(FILE_ERROR_OK, | |
423 metadata_->GetResourceEntryById(GetLocalId("foo"), &entry)); | |
424 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | |
425 EXPECT_EQ(FILE_ERROR_OK, | |
426 metadata_->GetResourceEntryById(GetLocalId("dirty"), &entry)); | |
427 EXPECT_FALSE(entry.file_specific_info().cache_state().is_dirty()); | |
428 } | |
429 | |
430 TEST_F(SyncClientTest, ScheduleRerun) { | |
431 // Add a fetch task for "foo", this should result in being paused. | |
432 drive_service_->set_resource_id_to_be_paused(resource_ids_["foo"]); | |
433 sync_client_->AddFetchTask(GetLocalId("foo")); | |
434 base::RunLoop().RunUntilIdle(); | |
435 | |
436 // While the first task is paused, add a task again. | |
437 // This results in scheduling rerun of the task. | |
438 sync_client_->AddFetchTask(GetLocalId("foo")); | |
439 base::RunLoop().RunUntilIdle(); | |
440 | |
441 // Resume the paused task. | |
442 drive_service_->set_resource_id_to_be_paused(std::string()); | |
443 ASSERT_FALSE(drive_service_->paused_action().is_null()); | |
444 drive_service_->paused_action().Run(); | |
445 base::RunLoop().RunUntilIdle(); | |
446 | |
447 // Task should be run twice. | |
448 EXPECT_EQ(2, drive_service_->download_file_count()); | |
449 } | |
450 | |
451 TEST_F(SyncClientTest, Dependencies) { | |
452 // Create directories locally. | |
453 const base::FilePath kPath1(FILE_PATH_LITERAL("drive/root/dir1")); | |
454 const base::FilePath kPath2 = kPath1.AppendASCII("dir2"); | |
455 | |
456 ResourceEntry parent; | |
457 EXPECT_EQ(FILE_ERROR_OK, | |
458 metadata_->GetResourceEntryByPath(kPath1.DirName(), &parent)); | |
459 | |
460 ResourceEntry entry1; | |
461 entry1.set_parent_local_id(parent.local_id()); | |
462 entry1.set_title(kPath1.BaseName().AsUTF8Unsafe()); | |
463 entry1.mutable_file_info()->set_is_directory(true); | |
464 entry1.set_metadata_edit_state(ResourceEntry::DIRTY); | |
465 std::string local_id1; | |
466 EXPECT_EQ(FILE_ERROR_OK, metadata_->AddEntry(entry1, &local_id1)); | |
467 | |
468 ResourceEntry entry2; | |
469 entry2.set_parent_local_id(local_id1); | |
470 entry2.set_title(kPath2.BaseName().AsUTF8Unsafe()); | |
471 entry2.mutable_file_info()->set_is_directory(true); | |
472 entry2.set_metadata_edit_state(ResourceEntry::DIRTY); | |
473 std::string local_id2; | |
474 EXPECT_EQ(FILE_ERROR_OK, metadata_->AddEntry(entry2, &local_id2)); | |
475 | |
476 // Start syncing the child first. | |
477 sync_client_->AddUpdateTask(ClientContext(USER_INITIATED), local_id2); | |
478 // Start syncing the parent later. | |
479 sync_client_->AddUpdateTask(ClientContext(USER_INITIATED), local_id1); | |
480 base::RunLoop().RunUntilIdle(); | |
481 | |
482 // Both entries are synced. | |
483 EXPECT_EQ(FILE_ERROR_OK, metadata_->GetResourceEntryById(local_id1, &entry1)); | |
484 EXPECT_EQ(ResourceEntry::CLEAN, entry1.metadata_edit_state()); | |
485 EXPECT_EQ(FILE_ERROR_OK, metadata_->GetResourceEntryById(local_id2, &entry2)); | |
486 EXPECT_EQ(ResourceEntry::CLEAN, entry2.metadata_edit_state()); | |
487 } | |
488 | |
489 TEST_F(SyncClientTest, WaitForUpdateTaskToComplete) { | |
490 // Create a directory locally. | |
491 const base::FilePath kPath(FILE_PATH_LITERAL("drive/root/dir1")); | |
492 | |
493 ResourceEntry parent; | |
494 EXPECT_EQ(FILE_ERROR_OK, | |
495 metadata_->GetResourceEntryByPath(kPath.DirName(), &parent)); | |
496 | |
497 ResourceEntry entry; | |
498 entry.set_parent_local_id(parent.local_id()); | |
499 entry.set_title(kPath.BaseName().AsUTF8Unsafe()); | |
500 entry.mutable_file_info()->set_is_directory(true); | |
501 entry.set_metadata_edit_state(ResourceEntry::DIRTY); | |
502 std::string local_id; | |
503 EXPECT_EQ(FILE_ERROR_OK, metadata_->AddEntry(entry, &local_id)); | |
504 | |
505 // Sync task is not yet avialable. | |
506 FileError error = FILE_ERROR_FAILED; | |
507 EXPECT_FALSE(sync_client_->WaitForUpdateTaskToComplete( | |
508 local_id, google_apis::test_util::CreateCopyResultCallback(&error))); | |
509 | |
510 // Start syncing the directory and wait for it to complete. | |
511 sync_client_->AddUpdateTask(ClientContext(USER_INITIATED), local_id); | |
512 | |
513 EXPECT_TRUE(sync_client_->WaitForUpdateTaskToComplete( | |
514 local_id, google_apis::test_util::CreateCopyResultCallback(&error))); | |
515 | |
516 base::RunLoop().RunUntilIdle(); | |
517 | |
518 // The callback is called. | |
519 EXPECT_EQ(FILE_ERROR_OK, error); | |
520 } | |
521 | |
522 } // namespace internal | |
523 } // namespace drive | |
OLD | NEW |