Modify session restore test to set restore setting via UI and verify UI settings using web driver.

functional/prefs.py

 #!/usr/bin/python
 # Copyright (c) 2011 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 shutil
 import sys
 
 import pyauto_functional  # Must be imported before pyauto
 import pyauto
+import test_utils
 
 
 class PrefsTest(pyauto.PyUITest):
   """TestCase for Preferences."""
 
   def Debug(self):
     """Test method for experimentation.
 
     This method will not run automatically.
     """
     while True:
       raw_input('Interact with the browser and hit <enter> to dump prefs... ')
       self.pprint(self.GetPrefsInfo().Prefs())
 
   def testSessionRestore(self):
     """Test session restore preference."""
+
+    pref_url = 'chrome://settings/browser'
     url1 = 'http://www.google.com/'
     url2 = 'http://news.google.com/'
+
+    self.NavigateToURL(pref_url)
+    # Set pref to restore session on startup.
+    driver = self.NewWebDriver()
+    restore_elem = driver.find_element_by_xpath(
+        '//input[@metric="Options_Startup_LastSession"]')
+    restore_elem.click()
+    self.assertTrue(restore_elem.is_selected())
+    self.RestartBrowser(clear_profile=False)
     self.NavigateToURL(url1)
     self.AppendTab(pyauto.GURL(url2))
     num_tabs = self.GetTabCount()
-    # Set pref to restore session on startup
-    self.SetPrefs(pyauto.kRestoreOnStartup, 1)
-    logging.debug('Setting %s to 1' % pyauto.kRestoreOnStartup)
     self.RestartBrowser(clear_profile=False)
-    # Verify
+    # Verify tabs are properly restored.
     self.assertEqual(self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup), 1)
     self.assertEqual(num_tabs, self.GetTabCount())
     self.ActivateTab(0)
     self.assertEqual(url1, self.GetActiveTabURL().spec())
     self.ActivateTab(1)
     self.assertEqual(url2, self.GetActiveTabURL().spec())
+    # Verify session restore option is still selected.
+    self.NavigateToURL(pref_url, 0, 0)
+    driver = self.NewWebDriver()
+    self.assertTrue(driver.find_element_by_xpath(
+        '//input[@metric="Options_Startup_LastSession"]').is_selected())
 
   def testNavigationStateOnSessionRestore(self):
     """Verify navigation state is preserved on session restore."""
     urls = ('http://www.google.com/',
             'http://news.google.com/',
             'http://dev.chromium.org/',)
     for url in urls:
       self.NavigateToURL(url)
     tab = self.GetBrowserWindow(0).GetTab(0)
     tab.GoBack()
@@ skipping 134 matching lines @@
     self.assertEqual(title, self.GetActiveTabTitle())
 
   def testHaveLocalStatePrefs(self):
     """Verify that we have some Local State prefs."""
     self.assertTrue(self.GetLocalStatePrefsInfo())
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()