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 135 matching lines @@
   void CandidateWindowClosed(InputMethodManager* manager) override {
     ++candidate_window_closed_count_;
   }
 
   int candidate_window_opened_count_;
   int candidate_window_closed_count_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestCandidateWindowObserver);
 };
+
 }  // namespace
 
 class InputMethodManagerImplTest :  public BrowserWithTestWindowTest {
  public:
   InputMethodManagerImplTest()
       : delegate_(NULL),
         candidate_window_controller_(NULL),
         keyboard_(NULL) {
   }
   ~InputMethodManagerImplTest() override {}
@@ skipping 1472 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, SetOneFeatureRestrictedState) {
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,

[James Cook, 2017/06/26 17:21:19]

nit: EXPECT_EQ here and below (unless the test will crash when the assertion
fails, in which case ASSERT_EQ is preferred)

[Azure Wei, 2017/06/27 14:27:15]

Done.

+            manager_->GetFeaturesRestrictedState());
+
+  // Sets emoji restricted and not restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI, true);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI,
+            manager_->GetFeaturesRestrictedState());
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI, false);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+
+  // Sets voice restricted and not restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, true);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE,
+            manager_->GetFeaturesRestrictedState());
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, false);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+
+  // Sets handwriting restricted and not restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_HANDWRITING,
+      true);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_HANDWRITING,
+            manager_->GetFeaturesRestrictedState());
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_HANDWRITING,
+      false);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+}
+
+TEST_F(InputMethodManagerImplTest, SetMoreThanOneFeaturesRestrictedState) {
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+
+  // Sets emoji and voice restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI, true);
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, true);
+  uint32_t emojiAndVoice =
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI |
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE;
+  ASSERT_EQ(emojiAndVoice, manager_->GetFeaturesRestrictedState());
+
+  // Sets emoji not restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI, false);
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, true);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE,
+            manager_->GetFeaturesRestrictedState());
+
+  // Sets voice not restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, false);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+}
+
+TEST_F(InputMethodManagerImplTest, SetAllFeaturesRestrictedState) {
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+
+  // Sets all restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI, true);
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, true);
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_HANDWRITING,
+      true);
+  uint32_t allRestricted =
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI |
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE |
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_HANDWRITING;
+  ASSERT_EQ(allRestricted, manager_->GetFeaturesRestrictedState());
+
+  // Sets all not restricted
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_EMOJI, false);
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_VOICE, false);
+  manager_->SetFeaturesRestrictedState(
+      InputMethodManager::FeaturesRestrictedState::RESTRICTED_HANDWRITING,
+      false);
+  ASSERT_EQ(InputMethodManager::FeaturesRestrictedState::RESTRICTED_NONE,
+            manager_->GetFeaturesRestrictedState());
+}
+
 }  // namespace input_method
 }  // namespace chromeos