Home | History | Annotate | Download | only in login 
      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 "base/command_line.h"
      6 #include "chrome/browser/chrome_notification_types.h"
      7 #include "chrome/browser/chromeos/login/existing_user_controller.h"
      8 #include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h"
      9 #include "chrome/browser/chromeos/login/ui/login_display_host_impl.h"
     10 #include "chrome/browser/chromeos/login/ui/webui_login_display.h"
     11 #include "chrome/browser/chromeos/login/wizard_controller.h"
     12 #include "chrome/browser/lifetime/application_lifetime.h"
     13 #include "chrome/browser/ui/webui/chromeos/login/signin_screen_handler.h"
     14 #include "chrome/common/chrome_switches.h"
     15 #include "chrome/test/base/in_process_browser_test.h"
     16 #include "chromeos/chromeos_switches.h"
     17 #include "content/public/test/test_utils.h"
     18 #include "google_apis/gaia/fake_gaia.h"
     19 #include "google_apis/gaia/gaia_switches.h"
     20 #include "net/test/embedded_test_server/embedded_test_server.h"
     21 #include "net/test/embedded_test_server/http_response.h"
     22 #include "testing/gtest/include/gtest/gtest.h"
     23 #include "ui/base/test/ui_controls.h"
     24 #include "ui/views/widget/widget.h"
     25 
     26 using namespace net::test_server;
     27 
     28 namespace chromeos {
     29 
     30 class OobeTest : public InProcessBrowserTest {
     31  public:
     32   OobeTest() {}
     33   virtual ~OobeTest() {}
     34 
     35   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     36     command_line->AppendSwitch(chromeos::switches::kLoginManager);
     37     command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests);
     38     command_line->AppendSwitchASCII(chromeos::switches::kLoginProfile, "user");
     39     command_line->AppendSwitchASCII(
     40         ::switches::kAuthExtensionPath, "gaia_auth");
     41     fake_gaia_.Initialize();
     42   }
     43 
     44   virtual void SetUpOnMainThread() OVERRIDE {
     45     CHECK(embedded_test_server()->InitializeAndWaitUntilReady());
     46     embedded_test_server()->RegisterRequestHandler(
     47         base::Bind(&FakeGaia::HandleRequest, base::Unretained(&fake_gaia_)));
     48     LOG(INFO) << "Set up http server at " << embedded_test_server()->base_url();
     49 
     50     CommandLine::ForCurrentProcess()->AppendSwitchASCII(
     51         ::switches::kGaiaUrl, embedded_test_server()->base_url().spec());
     52   }
     53 
     54   virtual void TearDownOnMainThread() OVERRIDE {
     55     // If the login display is still showing, exit gracefully.
     56     if (LoginDisplayHostImpl::default_host()) {
     57       base::MessageLoop::current()->PostTask(FROM_HERE,
     58                                              base::Bind(&chrome::AttemptExit));
     59       content::RunMessageLoop();
     60     }
     61   }
     62 
     63   chromeos::WebUILoginDisplay* GetLoginDisplay() {
     64     chromeos::ExistingUserController* controller =
     65         chromeos::ExistingUserController::current_controller();
     66     CHECK(controller);
     67     return static_cast<chromeos::WebUILoginDisplay*>(
     68         controller->login_display());
     69   }
     70 
     71   views::Widget* GetLoginWindowWidget() {
     72     return static_cast<chromeos::LoginDisplayHostImpl*>(
     73         chromeos::LoginDisplayHostImpl::default_host())
     74         ->login_window_for_test();
     75   }
     76 
     77  private:
     78   FakeGaia fake_gaia_;
     79   DISALLOW_COPY_AND_ASSIGN(OobeTest);
     80 };
     81 
     82 IN_PROC_BROWSER_TEST_F(OobeTest, NewUser) {
     83   chromeos::WizardController::SkipPostLoginScreensForTesting();
     84   chromeos::WizardController* wizard_controller =
     85       chromeos::WizardController::default_controller();
     86   CHECK(wizard_controller);
     87   wizard_controller->SkipToLoginForTesting(LoginScreenContext());
     88 
     89   content::WindowedNotificationObserver(
     90     chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE,
     91     content::NotificationService::AllSources()).Wait();
     92 
     93   // TODO(glotov): mock GAIA server (test_server()) should support
     94   // username/password configuration.
     95   GetLoginDisplay()->ShowSigninScreenForCreds("username", "password");
     96 
     97   content::WindowedNotificationObserver(
     98     chrome::NOTIFICATION_SESSION_STARTED,
     99     content::NotificationService::AllSources()).Wait();
    100 }
    101 
    102 IN_PROC_BROWSER_TEST_F(OobeTest, Accelerator) {
    103   chromeos::WizardController::SkipPostLoginScreensForTesting();
    104   chromeos::WizardController* wizard_controller =
    105       chromeos::WizardController::default_controller();
    106   CHECK(wizard_controller);
    107   wizard_controller->SkipToLoginForTesting(LoginScreenContext());
    108 
    109   content::WindowedNotificationObserver(
    110     chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE,
    111     content::NotificationService::AllSources()).Wait();
    112 
    113   gfx::NativeWindow login_window = GetLoginWindowWidget()->GetNativeWindow();
    114 
    115   ui_controls::SendKeyPress(login_window,
    116                             ui::VKEY_E,
    117                             true,    // control
    118                             false,   // shift
    119                             true,    // alt
    120                             false);  // command
    121   OobeScreenWaiter(OobeDisplay::SCREEN_OOBE_ENROLLMENT).Wait();
    122 }
    123 
    124 }  // namespace chromeos
    125