Enable device disabling during normal operation

chrome/browser/chromeos/system/device_disabling_browsertest.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 "base/command_line.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/run_loop.h"
+#include "chrome/browser/chromeos/login/wizard_controller.h"
+#include "chrome/browser/chromeos/policy/device_policy_builder.h"
+#include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h"
+#include "chrome/browser/chromeos/settings/cros_settings.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "chromeos/chromeos_switches.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/fake_session_manager_client.h"
+#include "chromeos/settings/cros_settings_names.h"
+#include "policy/proto/device_management_backend.pb.h"
+
+namespace chromeos {
+namespace system {
+
+class DeviceDisablingTest : public InProcessBrowserTest {
+ public:
+  DeviceDisablingTest();
+
+  // Device policy is updated in two steps:
+  // - First, set up the policy builder that GetDevicePolicyBuilder() returns.
+  // - Second, call SimulatePolicyFetch() to build and flush the resulting
+  //   policy blob to the browser.
+  policy::DevicePolicyBuilder* GetDevicePolicyBuilder();
+  void SimulatePolicyFetch();
+
+ private:
+  // InProcessBrowserTest:
+  void SetUpInProcessBrowserTestFixture() override;
+  void SetUpCommandLine(CommandLine* command_line) override;
+
+  FakeSessionManagerClient* fake_session_manager_client_;
+  policy::DevicePolicyCrosTestHelper test_helper_;
+
+  DISALLOW_COPY_AND_ASSIGN(DeviceDisablingTest);
+};
+
+
+DeviceDisablingTest::DeviceDisablingTest()
+    : fake_session_manager_client_(new FakeSessionManagerClient) {
+}
+
+policy::DevicePolicyBuilder* DeviceDisablingTest::GetDevicePolicyBuilder() {
+  return test_helper_.device_policy();
+}
+
+void DeviceDisablingTest::SimulatePolicyFetch() {
+  GetDevicePolicyBuilder()->Build();
+  fake_session_manager_client_->set_device_policy(
+      GetDevicePolicyBuilder()->GetBlob());
+  fake_session_manager_client_->OnPropertyChangeComplete(true);
+}
+
+void DeviceDisablingTest::SetUpInProcessBrowserTestFixture() {
+  DBusThreadManager::GetSetterForTesting()->SetSessionManagerClient(
+      scoped_ptr<SessionManagerClient>(fake_session_manager_client_));
+
+  test_helper_.InstallOwnerKey();
+  test_helper_.MarkAsEnterpriseOwned();
+}
+
+void DeviceDisablingTest::SetUpCommandLine(CommandLine* command_line) {
+  command_line->AppendSwitch(switches::kLoginManager);
+  command_line->AppendSwitch(switches::kForceLoginManagerInTests);
+}
+
+IN_PROC_BROWSER_TEST_F(DeviceDisablingTest, DisableDuringNormalOperation) {
+  // Mark the device as disabled and wait until cros settings update.
+  base::RunLoop run_loop;
+  scoped_ptr<CrosSettings::ObserverSubscription> observer =
+      CrosSettings::Get()->AddSettingsObserver(
+           kDeviceDisabled,
+           run_loop.QuitClosure());
+  GetDevicePolicyBuilder()->policy_data().mutable_device_state()->
+      set_device_mode(enterprise_management::DeviceState::DEVICE_MODE_DISABLED);
+  SimulatePolicyFetch();
+  run_loop.Run();
+
+  // Verify that the device disabled screen is being shown.
+  WizardController* wizard_controller = WizardController::default_controller();
+  ASSERT_TRUE(wizard_controller);
+  EXPECT_EQ(wizard_controller->GetScreen(
+                WizardController::kDeviceDisabledScreenName),
+            wizard_controller->current_screen());
+}
+
+}  // namespace system
+}  // namespace chromeos