Hide handwriting and voice buttons when keyboard is in restricted state

chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc

Index: chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc
diff --git a/chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc b/chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc
index 3e657f07336b6bdd8e1f1bf5a2e6bd1bc87aa319..642c1becd154a716bc12e41b4c8598a99bf42b7c 100644
--- a/chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc
+++ b/chrome/browser/chromeos/input_method/input_method_manager_impl_unittest.cc
@@ -1645,5 +1645,104 @@ TEST_F(InputMethodManagerImplTest, IntegrationWithAsh) {
   EXPECT_EQ(ImeIdFromEngineId(ids[1]), ime_controller.current_ime_id_);
 }
 
+TEST_F(InputMethodManagerImplTest, SetOneFeatureRestricted) {
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+
+  // Sets emoji restricted and not restricted

[James Cook, 2017/06/27 16:31:11]

super-nit: Comments should end with "." here and below

[Azure Wei, 2017/06/29 16:37:03]

Done.

+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::EMOJI,
+                                  true);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::EMOJI));
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::EMOJI,
+                                  false);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+
+  // Sets voice restricted and not restricted
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  true);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::VOICE));
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  false);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+
+  // Sets handwriting restricted and not restricted
+  manager_->SetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING, true);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING));
+  manager_->SetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING, false);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+}
+
+TEST_F(InputMethodManagerImplTest, SetMoreThanOneFeaturesRestricted) {
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+
+  // Sets emoji and voice restricted
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::EMOJI,
+                                  true);
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  true);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::EMOJI));
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::VOICE));
+  // Sets emoji not restricted
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::EMOJI,
+                                  false);
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  true);
+  EXPECT_FALSE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::EMOJI));
+
+  // Sets voice not restricted
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  false);
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+}
+
+TEST_F(InputMethodManagerImplTest, SetAllFeaturesRestricted) {
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+
+  // Sets all restricted
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::EMOJI,
+                                  true);
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  true);
+  manager_->SetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING, true);
+
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::EMOJI));
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::VOICE));
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING));
+
+  // Sets all not restricted
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::EMOJI,
+                                  false);
+  manager_->SetFeaturesRestricted(InputMethodManager::FeaturesRestricted::VOICE,
+                                  false);
+  manager_->SetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING, false);
+  EXPECT_FALSE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::EMOJI));
+  EXPECT_FALSE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::VOICE));
+  EXPECT_FALSE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::HANDWRITING));
+  EXPECT_TRUE(manager_->GetFeaturesRestricted(
+      InputMethodManager::FeaturesRestricted::NONE));
+}
+
 }  // namespace input_method
 }  // namespace chromeos