The consumer management enroll and unenroll buttons should only be visible to the device owner.

chrome/browser/ui/webui/options/chromeos/consumer_management_handler.cc

 // Copyright 2014 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/options/chromeos/consumer_management_handler.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/logging.h"
+#include "base/prefs/pref_service.h"
 #include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/chromeos/login/users/user_manager.h"
+#include "chrome/common/pref_names.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/power_manager_client.h"
 #include "content/public/browser/web_ui.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace chromeos {
 namespace options {
 
 ConsumerManagementHandler::ConsumerManagementHandler() {
 }
 
@@ skipping 41 matching lines @@
   web_ui()->RegisterMessageCallback(
       "enrollConsumerManagement",
       base::Bind(&ConsumerManagementHandler::HandleEnrollConsumerManagement,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback(
       "unenrollConsumerManagement",
       base::Bind(&ConsumerManagementHandler::HandleUnenrollConsumerManagement,
                  base::Unretained(this)));
 }
 
-void ConsumerManagementHandler::InitializeHandler() {
-  // TODO(davidyu): Get the real enrollment status. http://crbug.com/353050.
-  SetEnrollmentStatus(false);
-}
-
 void ConsumerManagementHandler::HandleEnrollConsumerManagement(
     const base::ListValue* args) {
-  StartEnrollment();
-  SetEnrollmentStatus(true);
+  UserManager* user_manager = chromeos::UserManager::Get();
+  if (!user_manager) {

[bartfab (slow), 2014/07/23 19:27:13]

Nit: Just like the other singletons, the UserManager will always be there.

[davidyu, 2014/07/24 04:28:30]

Hmm... the comment of UserManager::Get() is a little bit misleading. It says it
may return NULL on some cases, but in the implementation it actually CHECK()s
before returning the value. I removed the check statement in my change.

[bartfab (slow), 2014/07/24 11:54:53]

The UserManager might not exist in three cases:
* Early on in browser start-up, before enrollment can even start.
* Late in browser shut-down, when no enrollment should be happening anymore.
* In tests that do not instantiate it.

You should not be running into any of these.

+    LOG(ERROR) << "UserManager is not available.";
+    return;
+  }
+  if (!user_manager->IsCurrentUserOwner()) {
+    LOG(ERROR) << "Received enrollConsumerManagement, but the current user is "
+               << "not the owner.";
+    return;
+  }
+
+  PrefService* prefs = g_browser_process->local_state();
+  prefs->SetBoolean(prefs::kConsumerManagementEnrollmentRequested, true);
+  prefs->CommitPendingWrite();
+  chromeos::DBusThreadManager::Get()->GetPowerManagerClient()->RequestRestart();
 }
 
 void ConsumerManagementHandler::HandleUnenrollConsumerManagement(
     const base::ListValue* args) {
-  StartUnenrollment();
-  SetEnrollmentStatus(false);
-}
-
-void ConsumerManagementHandler::StartEnrollment() {
   NOTIMPLEMENTED();
 }
 
-void ConsumerManagementHandler::StartUnenrollment() {
-  NOTIMPLEMENTED();
-}
-
-void ConsumerManagementHandler::SetEnrollmentStatus(
-    bool is_enrolled) {
-  base::FundamentalValue is_enrolled_value(is_enrolled);
-  web_ui()->CallJavascriptFunction(
-      "BrowserOptions.setConsumerManagementEnrollmentStatus",
-      is_enrolled_value);
-}
-
 }  // namespace options
 }  // namespace chromeos