Add browser tests for enabling Google Drive offline mode on first run.

chrome/browser/chromeos/first_run/drive_first_run_controller.h

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 #ifndef CHROME_BROWSER_CHROMEOS_FIRST_RUN_DRIVE_FIRST_RUN_CONTROLLER_H_
 #define CHROME_BROWSER_CHROMEOS_FIRST_RUN_DRIVE_FIRST_RUN_CONTROLLER_H_
 
 #include "base/basictypes.h"
+#include "base/observer_list.h"
 #include "base/timer/timer.h"
 #include "chrome/browser/profiles/profile.h"
 
 namespace chromeos {
 
 class DriveWebContentsManager;
 
 // This class is responsible for kicking off the Google Drive offline
 // initialization process. There is an initial delay to avoid contention when
 // the session starts. DriveFirstRunController will manage its own lifetime and
 // destroy itself when the initialization succeeds or fails.
 class DriveFirstRunController {
  public:
+  class Observer {
+   public:
+    // Called when enabling offline mode times out. OnCompletion will be called
+    // immediately afterwards.
+    virtual void OnTimedOut() = 0;
+
+    // Called when the first run flow finishes, informing the observer of
+    // success or failure.
+    virtual void OnCompletion(bool success) = 0;
+
+   protected:
+    virtual ~Observer() {}
+  };
+
   DriveFirstRunController();
   ~DriveFirstRunController();
 
   // Starts the process to enable offline mode for the user's Drive account.
   void EnableOfflineMode();
 
+  // Manages observers of the first run flow.
+  void AddObserver(Observer* observer);
+  void RemoveObserver(Observer* observer);
+
+  // Set delay times for testing purposes.
+  void SetDelaysForTest(int initial_delay_secs, int timeout_secs);
+
+  // Set the app id and endpoint url for testing purposes.
+  void SetAppInfoForTest(const std::string& app_id,
+                         const std::string& endpoint_url);
+
  private:
   // Used as a callback to indicate whether the offline initialization
   // succeeds or fails.
   void OnOfflineInit(bool success);
 
   // Called when timed out waiting for offline initialization to complete.
   void OnWebContentsTimedOut();
 
   // Cleans up internal state and schedules self for deletion.
   void CleanUp();
 
   Profile* profile_;
   scoped_ptr<DriveWebContentsManager> web_contents_manager_;
   base::OneShotTimer<DriveFirstRunController> web_contents_timer_;
   base::OneShotTimer<DriveFirstRunController> initial_delay_timer_;
   bool started_;
+  ObserverList<Observer> observer_list_;
+
+  int initial_delay_secs_;
+  int web_contents_timeout_secs_;
+  std::string drive_offline_endpoint_url_;
+  std::string drive_hosted_app_id_;
 
   DISALLOW_COPY_AND_ASSIGN(DriveFirstRunController);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_FIRST_RUN_DRIVE_FIRST_RUN_CONTROLLER_H_