chrome/browser/chromeos/first_run/chromeos_first_run_browsertest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "ash/first_run/first_run_helper.h"
   6 #include "ash/shell.h"
   7 #include "ash/system/tray/system_tray.h"
   8 #include "chrome/browser/chromeos/first_run/first_run.h"
   9 #include "chrome/browser/chromeos/first_run/first_run_controller.h"
  10 #include "chrome/browser/chromeos/first_run/step_names.h"
  11 #include "chrome/browser/chromeos/login/test/js_checker.h"
  12 #include "chrome/test/base/in_process_browser_test.h"
  13 #include "content/public/test/test_utils.h"
  14
  15 namespace chromeos {
  16
  17 class FirstRunUIBrowserTest : public InProcessBrowserTest,
  18                               public FirstRunActor::Delegate {
  19  public:
  20   FirstRunUIBrowserTest()
  21       : initialized_(false),
  22         finalized_(false) {
  23   }
  24
  25   // FirstRunActor::Delegate overrides.
  26   void OnActorInitialized() override {
  27     initialized_ = true;
  28     if (!on_initialized_callback_.is_null())
  29       on_initialized_callback_.Run();
  30     controller()->OnActorInitialized();
  31   }
  32
  33   void OnNextButtonClicked(const std::string& step_name) override {
  34     controller()->OnNextButtonClicked(step_name);
  35   }
  36
  37   void OnStepShown(const std::string& step_name) override {
  38     current_step_name_ = step_name;
  39     if (!on_step_shown_callback_.is_null())
  40       on_step_shown_callback_.Run();
  41     controller()->OnStepShown(step_name);
  42   }
  43
  44   void OnStepHidden(const std::string& step_name) override {
  45     controller()->OnStepHidden(step_name);
  46   }
  47
  48   void OnHelpButtonClicked() override { controller()->OnHelpButtonClicked(); }
  49
  50   void OnActorFinalized() override {
  51     finalized_ = true;
  52     if (!on_finalized_callback_.is_null())
  53       on_finalized_callback_.Run();
  54     controller()->OnActorFinalized();
  55   }
  56
  57   void OnActorDestroyed() override { controller()->OnActorDestroyed(); }
  58
  59   void LaunchTutorial() {
  60     chromeos::first_run::LaunchTutorial();
  61     EXPECT_TRUE(controller() != NULL);
  62     // Replacing delegate to observe all messages coming from WebUI to
  63     // controller.
  64     controller()->actor_->set_delegate(this);
  65     initialized_ = controller()->actor_->IsInitialized();
  66   }
  67
  68   void WaitForInitialization() {
  69     if (initialized_)
  70       return;
  71     WaitUntilCalled(&on_initialized_callback_);
  72     EXPECT_TRUE(initialized_);
  73     js().set_web_contents(controller()->web_contents_for_tests_);
  74   }
  75
  76   void WaitForStep(const std::string& step_name) {
  77     if (current_step_name_ == step_name)
  78       return;
  79     WaitUntilCalled(&on_step_shown_callback_);
  80     EXPECT_EQ(current_step_name_, step_name);
  81   }
  82
  83   void AdvanceStep() {
  84     js().Evaluate("cr.FirstRun.currentStep_.nextButton_.click()");
  85   }
  86
  87   void WaitForFinalization() {
  88     if (!finalized_) {
  89       WaitUntilCalled(&on_finalized_callback_);
  90       EXPECT_TRUE(finalized_);
  91     }
  92   }
  93
  94   void WaitUntilCalled(base::Closure* callback) {
  95     scoped_refptr<content::MessageLoopRunner> runner =
  96         new content::MessageLoopRunner;
  97     *callback = runner->QuitClosure();
  98     runner->Run();
  99     callback->Reset();
 100   }
 101
 102   test::JSChecker& js() { return js_; }
 103
 104   ash::FirstRunHelper* shell_helper() {
 105     return controller()->shell_helper_.get();
 106   }
 107
 108   FirstRunController* controller() {
 109     return FirstRunController::GetInstanceForTest();
 110   }
 111
 112  private:
 113   std::string current_step_name_;
 114   bool initialized_;
 115   bool finalized_;
 116   base::Closure on_initialized_callback_;
 117   base::Closure on_step_shown_callback_;
 118   base::Closure on_finalized_callback_;
 119   test::JSChecker js_;
 120 };
 121
 122 IN_PROC_BROWSER_TEST_F(FirstRunUIBrowserTest, FirstRunFlow) {
 123   LaunchTutorial();
 124   WaitForInitialization();
 125   WaitForStep(first_run::kAppListStep);
 126   EXPECT_FALSE(shell_helper()->IsTrayBubbleOpened());
 127   AdvanceStep();
 128   WaitForStep(first_run::kTrayStep);
 129   EXPECT_TRUE(shell_helper()->IsTrayBubbleOpened());
 130   AdvanceStep();
 131   WaitForStep(first_run::kHelpStep);
 132   EXPECT_TRUE(shell_helper()->IsTrayBubbleOpened());
 133   AdvanceStep();
 134   WaitForFinalization();
 135   content::RunAllPendingInMessageLoop();
 136   EXPECT_EQ(controller(), (void*)NULL);
 137   // shell_helper() is destructed already, thats why we call Shell directly.
 138   EXPECT_FALSE(ash::Shell::GetInstance()->GetPrimarySystemTray()->
 139       HasSystemBubble());
 140 }
 141
 142 }  // namespace chromeos
 143