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 93 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_restricted_state_(FeaturesRestrictedState::RESTRICTED_NONE) {}
 
 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::SetFeaturesRestrictedState(
+    FeaturesRestrictedState feature,
+    bool restricted) {
+  if (restricted) {
+    features_restricted_state_ |= feature;
+  } else {
+    features_restricted_state_ &= (~feature);
+  }
+}
+
+uint32_t MockInputMethodManager::GetFeaturesRestrictedState() {
+  return features_restricted_state_;
+}
+
 }  // namespace input_method
 }  // namespace chromeos