chrome/browser/chromeos/login/login_ui_keyboard_browsertest.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 "base/command_line.h"
   6 #include "base/prefs/pref_service.h"
   7 #include "chrome/browser/browser_process.h"
   8 #include "chrome/browser/chromeos/input_method/input_method_persistence.h"
   9 #include "chrome/browser/chromeos/language_preferences.h"
  10 #include "chrome/browser/chromeos/login/login_manager_test.h"
  11 #include "chrome/browser/chromeos/login/startup_utils.h"
  12 #include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h"
  13 #include "chrome/browser/chromeos/login/ui/login_display_host_impl.h"
  14 #include "chrome/browser/ui/webui/chromeos/login/signin_screen_handler.h"
  15 #include "chrome/common/pref_names.h"
  16 #include "chromeos/chromeos_switches.h"
  17 #include "content/public/test/test_utils.h"
  18
  19 namespace chromeos {
  20
  21 namespace {
  22
  23 const char kTestUser1[] = "test-user1@gmail.com";
  24 const char kTestUser2[] = "test-user2@gmail.com";
  25 const char kTestUser3[] = "test-user3@gmail.com";
  26
  27 void Append_en_US_InputMethods(std::vector<std::string>* out) {
  28   out->push_back("xkb:us::eng");
  29   out->push_back("xkb:us:intl:eng");
  30   out->push_back("xkb:us:altgr-intl:eng");
  31   out->push_back("xkb:us:dvorak:eng");
  32   out->push_back("xkb:us:dvp:eng");
  33   out->push_back("xkb:us:colemak:eng");
  34   chromeos::input_method::InputMethodManager::Get()->MigrateInputMethods(out);
  35 }
  36
  37 class FocusPODWaiter {
  38  public:
  39   FocusPODWaiter() : focused_(false), runner_(new content::MessageLoopRunner) {
  40     GetOobeUI()
  41         ->signin_screen_handler_for_test()
  42         ->SetFocusPODCallbackForTesting(
  43             base::Bind(&FocusPODWaiter::OnFocusPOD, base::Unretained(this)));
  44   }
  45   ~FocusPODWaiter() {
  46     GetOobeUI()
  47         ->signin_screen_handler_for_test()
  48         ->SetFocusPODCallbackForTesting(base::Closure());
  49   }
  50
  51   void OnFocusPOD() {
  52     focused_ = true;
  53     if (runner_.get())
  54       base::MessageLoopForUI::current()->PostTask(
  55           FROM_HERE,
  56           base::Bind(&FocusPODWaiter::ExitMessageLoop, base::Unretained(this)));
  57   }
  58
  59   void ExitMessageLoop() { runner_->Quit(); }
  60
  61   void Wait() {
  62     if (focused_)
  63       return;
  64     runner_->Run();
  65     GetOobeUI()
  66         ->signin_screen_handler_for_test()
  67         ->SetFocusPODCallbackForTesting(base::Closure());
  68     runner_ = NULL;
  69   }
  70
  71  private:
  72   OobeUI* GetOobeUI() {
  73     OobeUI* oobe_ui =
  74         static_cast<chromeos::LoginDisplayHostImpl*>(
  75             chromeos::LoginDisplayHostImpl::default_host())->GetOobeUI();
  76     CHECK(oobe_ui);
  77     return oobe_ui;
  78   }
  79
  80   bool focused_;
  81
  82   scoped_refptr<content::MessageLoopRunner> runner_;
  83 };
  84
  85 }  // anonymous namespace
  86
  87 class LoginUIKeyboardTest : public chromeos::LoginManagerTest {
  88  public:
  89   LoginUIKeyboardTest() : LoginManagerTest(false) {}
  90   ~LoginUIKeyboardTest() override {}
  91
  92   void SetUpOnMainThread() override {
  93     user_input_methods.push_back("xkb:fr::fra");
  94     user_input_methods.push_back("xkb:de::ger");
  95
  96     chromeos::input_method::InputMethodManager::Get()->MigrateInputMethods(
  97         &user_input_methods);
  98
  99     LoginManagerTest::SetUpOnMainThread();
 100   }
 101
 102   // Should be called from PRE_ test so that local_state is saved to disk, and
 103   // reloaded in the main test.
 104   void InitUserLRUInputMethod() {
 105     PrefService* local_state = g_browser_process->local_state();
 106
 107     input_method::SetUserLRUInputMethodPreferenceForTesting(
 108         kTestUser1, user_input_methods[0], local_state);
 109     input_method::SetUserLRUInputMethodPreferenceForTesting(
 110         kTestUser2, user_input_methods[1], local_state);
 111   }
 112
 113  protected:
 114   std::vector<std::string> user_input_methods;
 115 };
 116
 117 IN_PROC_BROWSER_TEST_F(LoginUIKeyboardTest, PRE_CheckPODScreenDefault) {
 118   RegisterUser(kTestUser1);
 119   RegisterUser(kTestUser2);
 120
 121   StartupUtils::MarkOobeCompleted();
 122 }
 123
 124 // Check default IME initialization, when there is no IME configuration in
 125 // local_state.
 126 IN_PROC_BROWSER_TEST_F(LoginUIKeyboardTest, CheckPODScreenDefault) {
 127   js_checker().ExpectEQ("$('pod-row').pods.length", 2);
 128
 129   std::vector<std::string> expected_input_methods;
 130   Append_en_US_InputMethods(&expected_input_methods);
 131
 132   EXPECT_EQ(expected_input_methods,
 133             input_method::InputMethodManager::Get()
 134                 ->GetActiveIMEState()
 135                 ->GetActiveInputMethodIds());
 136 }
 137
 138 IN_PROC_BROWSER_TEST_F(LoginUIKeyboardTest, PRE_CheckPODScreenWithUsers) {
 139   RegisterUser(kTestUser1);
 140   RegisterUser(kTestUser2);
 141
 142   InitUserLRUInputMethod();
 143
 144   StartupUtils::MarkOobeCompleted();
 145 }
 146
 147 IN_PROC_BROWSER_TEST_F(LoginUIKeyboardTest, CheckPODScreenWithUsers) {
 148   js_checker().ExpectEQ("$('pod-row').pods.length", 2);
 149
 150   EXPECT_EQ(user_input_methods[0],
 151             input_method::InputMethodManager::Get()
 152                 ->GetActiveIMEState()
 153                 ->GetCurrentInputMethod()
 154                 .id());
 155
 156   std::vector<std::string> expected_input_methods;
 157   Append_en_US_InputMethods(&expected_input_methods);
 158   // Active IM for the first user (active user POD).
 159   expected_input_methods.push_back(user_input_methods[0]);
 160
 161   EXPECT_EQ(expected_input_methods,
 162             input_method::InputMethodManager::Get()
 163                 ->GetActiveIMEState()
 164                 ->GetActiveInputMethodIds());
 165
 166   FocusPODWaiter waiter;
 167   js_checker().Evaluate("$('pod-row').focusPod($('pod-row').pods[1])");
 168   waiter.Wait();
 169
 170   EXPECT_EQ(user_input_methods[1],
 171             input_method::InputMethodManager::Get()
 172                 ->GetActiveIMEState()
 173                 ->GetCurrentInputMethod()
 174                 .id());
 175 }
 176
 177 class LoginUIKeyboardTestWithUsersAndOwner : public chromeos::LoginManagerTest {
 178  public:
 179   LoginUIKeyboardTestWithUsersAndOwner() : LoginManagerTest(false) {}
 180   ~LoginUIKeyboardTestWithUsersAndOwner() override {}
 181
 182   void SetUpCommandLine(base::CommandLine* command_line) override {
 183     LoginManagerTest::SetUpCommandLine(command_line);
 184     command_line->AppendSwitch(switches::kStubCrosSettings);
 185
 186     LoginManagerTest::SetUpCommandLine(command_line);
 187   }
 188
 189   void SetUpOnMainThread() override {
 190     user_input_methods.push_back("xkb:fr::fra");
 191     user_input_methods.push_back("xkb:de::ger");
 192     user_input_methods.push_back("xkb:pl::pol");
 193
 194     chromeos::input_method::InputMethodManager::Get()->MigrateInputMethods(
 195         &user_input_methods);
 196
 197     CrosSettings::Get()->SetString(kDeviceOwner, kTestUser3);
 198
 199     LoginManagerTest::SetUpOnMainThread();
 200   }
 201
 202   // Should be called from PRE_ test so that local_state is saved to disk, and
 203   // reloaded in the main test.
 204   void InitUserLRUInputMethod() {
 205     PrefService* local_state = g_browser_process->local_state();
 206
 207     input_method::SetUserLRUInputMethodPreferenceForTesting(
 208         kTestUser1, user_input_methods[0], local_state);
 209     input_method::SetUserLRUInputMethodPreferenceForTesting(
 210         kTestUser2, user_input_methods[1], local_state);
 211     input_method::SetUserLRUInputMethodPreferenceForTesting(
 212         kTestUser3, user_input_methods[2], local_state);
 213
 214     local_state->SetString(language_prefs::kPreferredKeyboardLayout,
 215                            user_input_methods[2]);
 216   }
 217
 218   void CheckGaiaKeyboard();
 219
 220  protected:
 221   std::vector<std::string> user_input_methods;
 222 };
 223
 224 void LoginUIKeyboardTestWithUsersAndOwner::CheckGaiaKeyboard() {
 225   std::vector<std::string> expected_input_methods;
 226   // kPreferredKeyboardLayout is now set to last focused POD.
 227   expected_input_methods.push_back(user_input_methods[0]);
 228   // Locale default input methods (the first one also is hardware IM).
 229   Append_en_US_InputMethods(&expected_input_methods);
 230
 231   EXPECT_EQ(expected_input_methods,
 232             input_method::InputMethodManager::Get()
 233                 ->GetActiveIMEState()
 234                 ->GetActiveInputMethodIds());
 235 }
 236
 237 IN_PROC_BROWSER_TEST_F(LoginUIKeyboardTestWithUsersAndOwner,
 238                        PRE_CheckPODScreenKeyboard) {
 239   RegisterUser(kTestUser1);
 240   RegisterUser(kTestUser2);
 241   RegisterUser(kTestUser3);
 242
 243   InitUserLRUInputMethod();
 244
 245   StartupUtils::MarkOobeCompleted();
 246 }
 247
 248 IN_PROC_BROWSER_TEST_F(LoginUIKeyboardTestWithUsersAndOwner,
 249                        CheckPODScreenKeyboard) {
 250   js_checker().ExpectEQ("$('pod-row').pods.length", 3);
 251
 252   std::vector<std::string> expected_input_methods;
 253   // Owner input method.
 254   expected_input_methods.push_back(user_input_methods[2]);
 255   // Locale default input methods (the first one also is hardware IM).
 256   Append_en_US_InputMethods(&expected_input_methods);
 257   // Active IM for the first user (active user POD).
 258   expected_input_methods.push_back(user_input_methods[0]);
 259
 260   EXPECT_EQ(expected_input_methods,
 261             input_method::InputMethodManager::Get()
 262                 ->GetActiveIMEState()
 263                 ->GetActiveInputMethodIds());
 264
 265   // Switch to Gaia.
 266   js_checker().Evaluate("$('add-user-button').click()");
 267   OobeScreenWaiter(OobeDisplay::SCREEN_GAIA_SIGNIN).Wait();
 268   CheckGaiaKeyboard();
 269
 270   // Switch back.
 271   js_checker().Evaluate("$('cancel-add-user-button').click()");
 272   OobeScreenWaiter(OobeDisplay::SCREEN_ACCOUNT_PICKER).Wait();
 273
 274   EXPECT_EQ(expected_input_methods,
 275             input_method::InputMethodManager::Get()
 276                 ->GetActiveIMEState()
 277                 ->GetActiveInputMethodIds());
 278 }
 279 }  // namespace chromeos