Add login screen locale and input method device policies

chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/chromeos/input_method/input_method_manager_impl.h"
 
 #include <stddef.h>
 
 #include <algorithm>
 #include <memory>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "chrome/browser/chromeos/input_method/mock_candidate_window_controller.h"
 #include "chrome/browser/chromeos/input_method/mock_input_method_engine.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/test/base/browser_with_test_window_test.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/testing_profile_manager.h"
+#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/accelerators/accelerator.h"
 #include "ui/base/ime/chromeos/extension_ime_util.h"
 #include "ui/base/ime/chromeos/fake_ime_keyboard.h"
 #include "ui/base/ime/chromeos/fake_input_method_delegate.h"
 #include "ui/base/ime/chromeos/mock_component_extension_ime_manager_delegate.h"
 #include "ui/base/ime/chromeos/mock_ime_engine_handler.h"
 #include "ui/base/ime/ime_bridge.h"
 #include "ui/base/ime/input_method_initializer.h"
 #include "ui/chromeos/ime/input_method_menu_item.h"
@@ skipping 1411 matching lines @@
 TEST_F(InputMethodManagerImplTest, OverrideDefaultKeyboardUrlRef) {
   const GURL default_url("chrome://inputview.html");
   keyboard::SetOverrideContentUrl(default_url);
 
   EXPECT_EQ(default_url, keyboard::GetOverrideContentUrl());
 
   manager_->OverrideKeyboardUrlRef("emoji");
   EXPECT_EQ(default_url, keyboard::GetOverrideContentUrl());
 }
 
+TEST_F(InputMethodManagerImplTest, AllowedKeyboardLayoutsValid) {
+  InitComponentExtension();
+
+  // First, setup xkb:fr::fra input method
+  std::string original_input_method(ImeIdFromEngineId("xkb:fr::fra"));
+  ASSERT_TRUE(
+      manager_->GetActiveIMEState()->EnableInputMethod(original_input_method));
+  manager_->GetActiveIMEState()->ChangeInputMethod(original_input_method,
+                                                   false);
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetCurrentInputMethod().id(),
+              original_input_method);
+
+  // Only allow xkb:us::eng
+  std::vector<std::string> allowed = {"xkb:us::eng"};
+  EXPECT_TRUE(manager_->GetActiveIMEState()->SetAllowedInputMethods(allowed));
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetActiveInputMethodIds(),
+              testing::ElementsAre(ImeIdFromEngineId("xkb:us::eng")));
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetCurrentInputMethod().id(),
+              ImeIdFromEngineId("xkb:us::eng"));
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetAllowedInputMethods(),
+              testing::ElementsAre(ImeIdFromEngineId("xkb:us::eng")));
+}
+
+TEST_F(InputMethodManagerImplTest, AllowedKeyboardLayoutsInvalid) {
+  InitComponentExtension();
+
+  // First, setup xkb:fr::fra input method
+  std::string original_input_method(ImeIdFromEngineId("xkb:fr::fra"));
+  ASSERT_TRUE(
+      manager_->GetActiveIMEState()->EnableInputMethod(original_input_method));
+  manager_->GetActiveIMEState()->ChangeInputMethod(original_input_method,
+                                                   false);
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetCurrentInputMethod().id(),
+              original_input_method);
+
+  // Only allow xkb:us::eng
+  std::vector<std::string> allowed = {"invalid_input_method"};
+  EXPECT_FALSE(manager_->GetActiveIMEState()->SetAllowedInputMethods(allowed));
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetCurrentInputMethod().id(),
+              original_input_method);
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetAllowedInputMethods(),
+              testing::IsEmpty());
+}
+
+TEST_F(InputMethodManagerImplTest, AllowedKeyboardLayoutsValidAndInvalid) {
+  InitComponentExtension();
+
+  // First, enable xkb:fr::fra and xkb:de::ger
+  std::string original_input_method_1(ImeIdFromEngineId("xkb:fr::fra"));
+  std::string original_input_method_2(ImeIdFromEngineId("xkb:de::ger"));
+  ASSERT_TRUE(manager_->GetActiveIMEState()->EnableInputMethod(
+      original_input_method_1));
+  ASSERT_TRUE(manager_->GetActiveIMEState()->EnableInputMethod(
+      original_input_method_2));
+  manager_->GetActiveIMEState()->ChangeInputMethod(original_input_method_1,
+                                                   false);
+
+  // Allow xkb:fr::fra and an invalid input method id. The invalid id should be
+  // ignored.
+  std::vector<std::string> allowed = {original_input_method_1,
+                                      "invalid_input_method"};
+  EXPECT_TRUE(manager_->GetActiveIMEState()->SetAllowedInputMethods(allowed));
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetCurrentInputMethod().id(),
+              original_input_method_1);
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetAllowedInputMethods(),
+              testing::ElementsAre(original_input_method_1));
+
+  // Try to re-enable xkb:de::ger
+  EXPECT_FALSE(manager_->GetActiveIMEState()->EnableInputMethod(
+      original_input_method_2));
+}
+
+TEST_F(InputMethodManagerImplTest, AllowedKeyboardLayoutsAndExtensions) {
+  InitComponentExtension();
+
+  EXPECT_TRUE(manager_->GetActiveIMEState()->EnableInputMethod(
+      ImeIdFromEngineId(kNaclMozcJpId)));
+  EXPECT_TRUE(manager_->GetActiveIMEState()->EnableInputMethod(
+      ImeIdFromEngineId("xkb:fr::fra")));
+
+  std::vector<std::string> allowed = {"xkb:us::eng"};
+  EXPECT_TRUE(manager_->GetActiveIMEState()->SetAllowedInputMethods(allowed));
+
+  EXPECT_TRUE(manager_->GetActiveIMEState()->EnableInputMethod(
+      ImeIdFromEngineId(kNaclMozcUsId)));
+  EXPECT_THAT(manager_->GetActiveIMEState()->GetActiveInputMethodIds(),
+              testing::ElementsAre(ImeIdFromEngineId("xkb:us::eng"),
+                                   ImeIdFromEngineId(kNaclMozcJpId),
+                                   ImeIdFromEngineId(kNaclMozcUsId)));
+}
+
 }  // namespace input_method
 }  // namespace chromeos