Add RESTORE_MODE_DISABLED to RestoreMode

chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.cc

Index: chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.cc
diff --git a/chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.cc b/chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.cc
index 31a287b489db7ac9571a0795cd67297b611c2588..58409e10cb6c3b538d04c024c06e82e0c836df79 100644
--- a/chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.cc
+++ b/chrome/browser/chromeos/login/screens/device_disabled_screen_unittest.cc
@@ -102,9 +102,13 @@ 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) {