Hide handwriting and voice buttons when keyboard is in restricted state

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>
@@ skipping 1488 matching lines @@
             ime_controller.current_ime_id_);
 
   // Unlock screen. The original state, pinyin-dv, is restored.
   manager_->SetState(saved_ime_state);
   manager_->SetUISessionState(InputMethodManager::STATE_BROWSER_SCREEN);
   ime_controller_client.FlushMojoForTesting();
   ASSERT_EQ(3u, ime_controller.available_imes_.size());  // Dvorak and 2 IMEs.
   EXPECT_EQ(ImeIdFromEngineId(ids[1]), ime_controller.current_ime_id_);
 }
 
+TEST_F(InputMethodManagerImplTest, SetFeaturesDisabled) {
+  // All features are enabled by default.
+  EXPECT_TRUE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_ALL));
+  EXPECT_TRUE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_EMOJI));
+  EXPECT_TRUE(manager_->GetImeMenuFeatureEnabled(
+      InputMethodManager::FEATURE_HANDWRITING));
+  EXPECT_TRUE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_VOICE));
+
+  // Sets emoji disabled and then enabled.
+  manager_->SetImeMenuFeatureEnabled(InputMethodManager::FEATURE_EMOJI, false);
+  EXPECT_FALSE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_EMOJI));
+  manager_->SetImeMenuFeatureEnabled(InputMethodManager::FEATURE_EMOJI, true);
+  EXPECT_TRUE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_EMOJI));
+
+  // Sets voice disabled and then enabled.
+  manager_->SetImeMenuFeatureEnabled(InputMethodManager::FEATURE_VOICE, false);
+  EXPECT_FALSE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_VOICE));
+  manager_->SetImeMenuFeatureEnabled(InputMethodManager::FEATURE_VOICE, true);
+  EXPECT_TRUE(
+      manager_->GetImeMenuFeatureEnabled(InputMethodManager::FEATURE_VOICE));
+
+  // Sets handwriting disabled and then enabled.
+  manager_->SetImeMenuFeatureEnabled(InputMethodManager::FEATURE_HANDWRITING,
+                                     false);
+  EXPECT_FALSE(manager_->GetImeMenuFeatureEnabled(
+      InputMethodManager::FEATURE_HANDWRITING));
+  manager_->SetImeMenuFeatureEnabled(InputMethodManager::FEATURE_HANDWRITING,
+                                     true);
+  EXPECT_TRUE(manager_->GetImeMenuFeatureEnabled(
+      InputMethodManager::FEATURE_HANDWRITING));
+}
+
 }  // namespace input_method
 }  // namespace chromeos