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
26using namespace net::test_server;
27
28namespace chromeos {
29
30class 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
82IN_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
102IN_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