Settings People Revamp: Add 'Manage Supervised Users' row

chrome/browser/ui/webui/settings/profile_info_handler_unittest.cc

 // Copyright 2016 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/ui/webui/settings/profile_info_handler.h"
 
 #include <memory>
 
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
@@ skipping 39 matching lines @@
 #if defined(OS_CHROMEOS)
     profile_ = profile_manager_.CreateTestingProfile("fake_id@gmail.com");
 #else
     profile_ = profile_manager_.CreateTestingProfile("Profile 1");
 #endif
 
     handler_.reset(new TestProfileInfoHandler(profile_));
     handler_->set_web_ui(&web_ui_);
   }
 
-  void VerifyResponse(const base::Value* call_argument) {
+  void VerifyProfileInfo(const base::Value* call_argument) {
     const base::DictionaryValue* response = nullptr;
     ASSERT_TRUE(call_argument->GetAsDictionary(&response));
 
     std::string name;
     std::string icon_url;
     ASSERT_TRUE(response->GetString("name", &name));
     ASSERT_TRUE(response->GetString("iconUrl", &icon_url));
 
 #if defined(OS_CHROMEOS)
     EXPECT_EQ("fake_id@gmail.com", name);
@@ skipping 34 matching lines @@
   EXPECT_EQ("cr.webUIResponse", data.function_name());
 
   std::string callback_id;
   ASSERT_TRUE(data.arg1()->GetAsString(&callback_id));
   EXPECT_EQ("get-profile-info-callback-id", callback_id);
 
   bool success = false;
   ASSERT_TRUE(data.arg2()->GetAsBoolean(&success));
   EXPECT_TRUE(success);
 
-  VerifyResponse(data.arg3());
+  VerifyProfileInfo(data.arg3());
 }
 
 TEST_F(ProfileInfoHandlerTest, PushProfileInfo) {
-  base::ListValue list_args;
-  list_args.AppendString("get-profile-info-callback-id");
-  handler()->HandleGetProfileInfo(&list_args);
+  handler()->AllowJavascript();
 
   handler()->OnProfileAvatarChanged(base::FilePath());
 
-  EXPECT_EQ(2U, web_ui()->call_data().size());
+  EXPECT_EQ(1U, web_ui()->call_data().size());
 
   const content::TestWebUI::CallData& data = *web_ui()->call_data().back();
   EXPECT_EQ("cr.webUIListenerCallback", data.function_name());
 
   std::string event_id;
   ASSERT_TRUE(data.arg1()->GetAsString(&event_id));
   EXPECT_EQ(ProfileInfoHandler::kProfileInfoChangedEventName, event_id);
 
-  VerifyResponse(data.arg2());
+  VerifyProfileInfo(data.arg2());
+}
+
+TEST_F(ProfileInfoHandlerTest, GetProfileManagesSupervisedUsers) {
+  base::ListValue list_args;
+  list_args.AppendString("get-profile-manages-supervised-users-callback-id");
+  handler()->HandleGetProfileManagesSupervisedUsers(&list_args);
+
+  EXPECT_EQ(1U, web_ui()->call_data().size());
+
+  const content::TestWebUI::CallData& data = *web_ui()->call_data().back();
+  EXPECT_EQ("cr.webUIResponse", data.function_name());
+
+  std::string callback_id;
+  ASSERT_TRUE(data.arg1()->GetAsString(&callback_id));
+  EXPECT_EQ("get-profile-manages-supervised-users-callback-id", callback_id);
+
+  bool success = false;
+  ASSERT_TRUE(data.arg2()->GetAsBoolean(&success));
+  EXPECT_TRUE(success);
+
+  bool has_supervised_users = false;
+  ASSERT_TRUE(data.arg3()->GetAsBoolean(&has_supervised_users));
+  EXPECT_FALSE(has_supervised_users);
+}
+
+TEST_F(ProfileInfoHandlerTest, PushProfileManagesSupervisedUsers) {
+  handler()->AllowJavascript();
+
+  handler()->PushProfileManagesSupervisedUsersStatus();
+
+  EXPECT_EQ(1U, web_ui()->call_data().size());
+
+  const content::TestWebUI::CallData& data = *web_ui()->call_data().back();
+  EXPECT_EQ("cr.webUIListenerCallback", data.function_name());
+
+  std::string event_id;
+  ASSERT_TRUE(data.arg1()->GetAsString(&event_id));
+  EXPECT_EQ(ProfileInfoHandler::kProfileManagesSupervisedUsersChangedEventName,
+            event_id);
+
+  bool has_supervised_users = false;
+  ASSERT_TRUE(data.arg2()->GetAsBoolean(&has_supervised_users));
+  EXPECT_FALSE(has_supervised_users);

[hcarmona, 2016/06/10 22:58:08]

Is it possible to add a test where |has_supervised_users| is expected to be
true?

[tommycli, 2016/06/10 23:28:45]

Done. Yes, I was able to enhance the test using this feedback.

 }
 
 }  // namespace settings