Add RESTORE_MODE_DISABLED to RestoreMode

chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.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/chromeos/login/screens/device_disabled_screen.h"
 
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/scoped_user_pref_update.h"
@@ skipping 84 matching lines @@
 }
 
 void DeviceDisabledScreenTest::ShowErrorScreen() {
 }
 
 void DeviceDisabledScreenTest::HideErrorScreen(BaseScreen* parent_screen) {
 }
 
 void DeviceDisabledScreenTest::SetDeviceDisabled(bool disabled) {
   DictionaryPrefUpdate dict(&local_state_, prefs::kServerBackedDeviceState);
-  dict->SetBoolean(policy::kDeviceStateDisabled, disabled);
-  if (disabled)
-    dict->SetString(policy::kDeviceStateDisabledMessage, kDisabledMessage);
+  if (disabled) {
+    dict->SetString(policy::kDeviceStateRestoreMode,
+                    policy::kDeviceStateRestoreModeDisabled);
+  } else {
+    dict->Remove(policy::kDeviceStateRestoreMode, nullptr);
+  }
+  dict->SetString(policy::kDeviceStateDisabledMessage, kDisabledMessage);
 }
 
 void DeviceDisabledScreenTest::SetDeviceMode(policy::DeviceMode device_mode) {
   reinterpret_cast<policy::StubEnterpriseInstallAttributes*>(
       TestingBrowserProcess::GetGlobal()->platform_part()->
           browser_policy_connector_chromeos()->GetInstallAttributes())->
               SetMode(device_mode);
 }
 
 void DeviceDisabledScreenTest::ExpectScreenToNotShow() {
@@ skipping 55 matching lines @@
 // Verifies that the device disabled screen is shown when the device is marked
 // as disabled, device disabling is not turned off by flag and the device is not
 // owned yet.
 TEST_F(DeviceDisabledScreenTest, ShowWhenDisabledAndNotOwned) {
   SetDeviceDisabled(true);
   ExpectScreenToShow();
   TryToShowScreen();
 }
 
 }  // namespace chromeos