Hide handwriting and voice buttons when keyboard is in restricted state

ui/base/ime/chromeos/mock_input_method_manager.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 "ui/base/ime/chromeos/mock_input_method_manager.h"
 
 #include <utility>
 
 namespace chromeos {
 namespace input_method {
@@ skipping 81 matching lines @@
   return true;
 }
 
 const std::vector<std::string>&
 MockInputMethodManager::State::GetAllowedInputMethods() {
   return allowed_input_method_ids_;
 }
 
 MockInputMethodManager::State::~State() {}
 
-MockInputMethodManager::MockInputMethodManager() {}
+MockInputMethodManager::MockInputMethodManager()
+    : features_enabled_state_(InputMethodManager::FEATURE_ALL) {}
 
 MockInputMethodManager::~MockInputMethodManager() {}
 
 InputMethodManager::UISessionState MockInputMethodManager::GetUISessionState() {
   return InputMethodManager::STATE_BROWSER_SCREEN;
 }
 
 void MockInputMethodManager::AddObserver(
     InputMethodManager::Observer* observer) {}
 
@@ skipping 70 matching lines @@
 
 void MockInputMethodManager::MaybeNotifyImeMenuActivationChanged() {}
 
 void MockInputMethodManager::OverrideKeyboardUrlRef(const std::string& keyset) {
 }
 
 bool MockInputMethodManager::IsEmojiHandwritingVoiceOnImeMenuEnabled() {
   return true;
 }
 
+void MockInputMethodManager::SetImeMenuFeatureEnabled(ImeMenuFeature feature,
+                                                      bool enabled) {
+  if (enabled)
+    features_enabled_state_ |= feature;
+  else
+    features_enabled_state_ &= ~feature;
+}
+
+bool MockInputMethodManager::GetImeMenuFeatureEnabled(
+    ImeMenuFeature feature) const {
+  return features_enabled_state_ & feature;
+}
+
 }  // namespace input_method
 }  // namespace chromeos