chrome/browser/chromeos/login/test/wizard_in_process_browser_test.cc

   1 // Copyright 2014 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/login/test/wizard_in_process_browser_test.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "chrome/browser/chrome_notification_types.h"
  10 #include "chrome/browser/chromeos/login/login_wizard.h"
  11 #include "chrome/browser/chromeos/login/ui/login_display_host_impl.h"
  12 #include "chrome/browser/chromeos/login/wizard_controller.h"
  13 #include "chrome/browser/ui/browser.h"
  14 #include "chrome/common/chrome_switches.h"
  15 #include "chromeos/chromeos_switches.h"
  16 #include "content/public/browser/notification_service.h"
  17
  18 namespace chromeos {
  19
  20 WizardInProcessBrowserTest::WizardInProcessBrowserTest(const char* screen_name)
  21     : screen_name_(screen_name),
  22       host_(NULL) {
  23 }
  24
  25 void WizardInProcessBrowserTest::SetUp() {
  26   WizardController::SetZeroDelays();
  27   InProcessBrowserTest::SetUp();
  28 }
  29
  30 void WizardInProcessBrowserTest::SetUpCommandLine(
  31     base::CommandLine* command_line) {
  32   command_line->AppendSwitch(::switches::kNoStartupWindow);
  33   command_line->AppendSwitch(switches::kLoginManager);
  34 }
  35
  36 void WizardInProcessBrowserTest::SetUpOnMainThread() {
  37   SetUpWizard();
  38   if (!screen_name_.empty()) {
  39     ShowLoginWizard(screen_name_);
  40     host_ = LoginDisplayHostImpl::default_host();
  41   }
  42 }
  43
  44 void WizardInProcessBrowserTest::TearDownOnMainThread() {
  45   // LoginDisplayHost owns controllers and all windows.
  46   base::MessageLoopForUI::current()->DeleteSoon(FROM_HERE, host_);
  47
  48   base::MessageLoopForUI::current()->RunUntilIdle();
  49 }
  50
  51 }  // namespace chromeos