| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 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. | |
| 4 | |
| 5 #include "chrome/browser/chromeos/extensions/file_manager/drive_test_util.h" | |
| 6 | |
| 7 #include "base/files/file_path.h" | |
| 8 #include "base/run_loop.h" | |
| 9 #include "chrome/browser/chromeos/drive/drive_integration_service.h" | |
| 10 #include "chrome/browser/profiles/profile.h" | |
| 11 #include "content/public/browser/browser_context.h" | |
| 12 #include "webkit/browser/fileapi/external_mount_points.h" | |
| 13 | |
| 14 namespace file_manager { | |
| 15 namespace test_util { | |
| 16 | |
| 17 namespace { | |
| 18 | |
| 19 // Helper class used to wait for |OnFileSystemMounted| event from a drive file | |
| 20 // system. | |
| 21 class DriveMountPointWaiter : public drive::DriveIntegrationServiceObserver { | |
| 22 public: | |
| 23 explicit DriveMountPointWaiter( | |
| 24 drive::DriveIntegrationService* integration_service) | |
| 25 : integration_service_(integration_service) { | |
| 26 integration_service_->AddObserver(this); | |
| 27 } | |
| 28 | |
| 29 virtual ~DriveMountPointWaiter() { | |
| 30 integration_service_->RemoveObserver(this); | |
| 31 } | |
| 32 | |
| 33 // DriveIntegrationServiceObserver override. | |
| 34 virtual void OnFileSystemMounted() OVERRIDE { | |
| 35 // Note that it is OK for |run_loop_.Quit| to be called before | |
| 36 // |run_loop_.Run|. In this case |Run| will return immediately. | |
| 37 run_loop_.Quit(); | |
| 38 } | |
| 39 | |
| 40 // Runs loop until the file system is mounted. | |
| 41 void Wait() { | |
| 42 run_loop_.Run(); | |
| 43 } | |
| 44 | |
| 45 private: | |
| 46 drive::DriveIntegrationService* integration_service_; | |
| 47 base::RunLoop run_loop_; | |
| 48 }; | |
| 49 | |
| 50 } // namespace | |
| 51 | |
| 52 void WaitUntilDriveMountPointIsAdded(Profile* profile) { | |
| 53 DCHECK(profile); | |
| 54 | |
| 55 // Drive mount point is added by the browser when the drive system service | |
| 56 // is first initialized. It is done asynchronously after some other parts of | |
| 57 // the service are initialized (e.g. resource metadata and cache), thus racy | |
| 58 // with the test start. To handle this raciness, the test verifies that | |
| 59 // drive mount point is added before continuing. If this is not the case, | |
| 60 // drive file system is observed for FileSystemMounted event (by | |
| 61 // |mount_point_waiter|) and test continues once the event is encountered. | |
| 62 drive::DriveIntegrationService* integration_service = | |
| 63 drive::DriveIntegrationServiceFactory::FindForProfileRegardlessOfStates( | |
| 64 profile); | |
| 65 DCHECK(integration_service); | |
| 66 | |
| 67 const std::string drive_mount_point_name = | |
| 68 drive::util::GetDriveMountPointPath().BaseName().AsUTF8Unsafe(); | |
| 69 base::FilePath ignored; | |
| 70 // GetRegisteredPath succeeds iff the mount point exists. | |
| 71 if (content::BrowserContext::GetMountPoints(profile)-> | |
| 72 GetRegisteredPath(drive_mount_point_name, &ignored)) | |
| 73 return; | |
| 74 | |
| 75 DriveMountPointWaiter mount_point_waiter(integration_service); | |
| 76 LOG(INFO) << "Waiting for drive mount point to get mounted."; | |
| 77 mount_point_waiter.Wait(); | |
| 78 LOG(INFO) << "Drive mount point found."; | |
| 79 } | |
| 80 | |
| 81 } // namespace test_util | |
| 82 } // namespace file_manager | |
| OLD | NEW |