src/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(CommandLine* command_line) {
  31   command_line->AppendSwitch(::switches::kNoStartupWindow);
  32   command_line->AppendSwitch(switches::kLoginManager);
  33 }
  34
  35 void WizardInProcessBrowserTest::SetUpOnMainThread() {
  36   SetUpWizard();
  37   if (!screen_name_.empty()) {
  38     ShowLoginWizard(screen_name_);
  39     host_ = LoginDisplayHostImpl::default_host();
  40   }
  41 }
  42
  43 void WizardInProcessBrowserTest::TearDownOnMainThread() {
  44   // LoginDisplayHost owns controllers and all windows.
  45   base::MessageLoopForUI::current()->DeleteSoon(FROM_HERE, host_);
  46
  47   base::MessageLoopForUI::current()->RunUntilIdle();
  48 }
  49
  50 }  // namespace chromeos