Home | History | Annotate | Download | only in file_manager 
      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/file_manager/drive_test_util.h"
      6 
      7 #include "base/run_loop.h"
      8 #include "chrome/browser/chromeos/drive/drive_integration_service.h"
      9 
     10 namespace file_manager {
     11 namespace test_util {
     12 
     13 namespace {
     14 
     15 // Helper class used to wait for |OnFileSystemMounted| event from a drive file
     16 // system.
     17 class DriveMountPointWaiter : public drive::DriveIntegrationServiceObserver {
     18  public:
     19   explicit DriveMountPointWaiter(
     20       drive::DriveIntegrationService* integration_service)
     21       : integration_service_(integration_service) {
     22     integration_service_->AddObserver(this);
     23   }
     24 
     25   virtual ~DriveMountPointWaiter() {
     26     integration_service_->RemoveObserver(this);
     27   }
     28 
     29   // DriveIntegrationServiceObserver override.
     30   virtual void OnFileSystemMounted() OVERRIDE {
     31     // Note that it is OK for |run_loop_.Quit| to be called before
     32     // |run_loop_.Run|. In this case |Run| will return immediately.
     33     run_loop_.Quit();
     34   }
     35 
     36   // Runs loop until the file system is mounted.
     37   void Wait() {
     38     run_loop_.Run();
     39   }
     40 
     41  private:
     42   drive::DriveIntegrationService* integration_service_;
     43   base::RunLoop run_loop_;
     44 };
     45 
     46 }  // namespace
     47 
     48 void WaitUntilDriveMountPointIsAdded(Profile* profile) {
     49   DCHECK(profile);
     50 
     51   // Drive mount point is added by the browser when the drive system service
     52   // is first initialized. It is done asynchronously after some other parts of
     53   // the service are initialized (e.g. resource metadata and cache), thus racy
     54   // with the test start. To handle this raciness, the test verifies that
     55   // drive mount point is added before continuing. If this is not the case,
     56   // drive file system is observed for FileSystemMounted event (by
     57   // |mount_point_waiter|) and test continues once the event is encountered.
     58   drive::DriveIntegrationService* integration_service =
     59       drive::DriveIntegrationServiceFactory::FindForProfile(profile);
     60   DCHECK(integration_service);
     61   DCHECK(integration_service->is_enabled());
     62 
     63   if (integration_service->IsMounted())
     64     return;
     65 
     66   DriveMountPointWaiter mount_point_waiter(integration_service);
     67   VLOG(1) << "Waiting for drive mount point to get mounted.";
     68   mount_point_waiter.Wait();
     69   VLOG(1) << "Drive mount point found.";
     70 }
     71 
     72 }  // namespace test_util
     73 }  // namespace file_manager
     74