Update RestoreOnStartupPolicyHandler to translate to the correct preferences

chrome/test/functional/policy.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 import logging
 import os
 
 import pyauto_functional  # must come before pyauto.
 import policy_base
@@ skipping 248 matching lines @@
     self.assertTrue(self._IsWebGLEnabled())
 
     self.SetUserPolicy({
         'Disable3DAPIs': True
     })
     self.assertTrue(self.GetPrefsInfo().Prefs(pyauto.kDisable3DAPIs))
     # The Disable3DAPIs policy only applies updated values to new renderers.
     self._RestartRenderer()
     self.assertFalse(self._IsWebGLEnabled())
 
-  def testStartupOptions(self):
-    """Verify that user cannot modify the startup page options."""
+  def testStartupOptionsURLs(self):
+    """Verify that user cannot modify the startup page options if "Open the
+    following URLs" is set by a policy.
+    """
     policy = {
       'RestoreOnStartup': 4,
       'RestoreOnStartupURLs': ['http://chromium.org']
     }
     self.SetUserPolicy(policy)
     # Verify startup option
     self.assertEquals(4, self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup))
     self.assertRaises(
         pyauto.JSONInterfaceError,
         lambda: self.SetPrefs(pyauto.kRestoreOnStartup, 1))
-    policy['RestoreOnStartup'] = 0
+
+  def testStartupOptionsHomepage(self):
+    """Verify that user cannot modify the startup page options if the
+    deprecated "Open the homepage" option is set by a policy.
+    """
+    policy = {
+      'RestoreOnStartup': 0,
+      'HomepageLocation': 'http://chromium.org',
+      'HomepageIsNewTabPage': False,
+    }
     self.SetUserPolicy(policy)
-    self.assertEquals(0, self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup))
+    self.assertEquals(4, self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup))
     self.assertRaises(
         pyauto.JSONInterfaceError,
         lambda: self.SetPrefs(pyauto.kRestoreOnStartup, 1))
     # Verify URLs to open on startup
     self.assertEquals(
         ['http://chromium.org'],
         self.GetPrefsInfo().Prefs(pyauto.kURLsToRestoreOnStartup))
     self.assertRaises(
         pyauto.JSONInterfaceError,
         lambda: self.SetPrefs(pyauto.kURLsToRestoreOnStartup,
@@ skipping 353 matching lines @@
     # Give the system 30 seconds to go get this extension.  We are not sure how
     # long it will take the policy to take affect and download the extension.
     self.assertTrue(self.WaitUntil(lambda:
       self._CheckForExtensionByID(self._SCREEN_CAPTURE_CRX_ID),
       expect_retval=True),
       msg='The force install extension was never installed.')
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()