Update pyauto tests to work with the uber page.

chrome/test/functional/extensions.py

 #!/usr/bin/env 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.
 
 """
 This module is a simple qa tool that installs extensions and tests whether the
 browser crashes while visiting a list of urls.
 
 Usage: python extensions.py -v
 
 Note: This assumes that there is a directory of extensions called
 'extensions-tool' and that there is a file of newline-separated urls to visit
 called 'urls.txt' in the data directory.
 """
 
 import glob
 import logging
 import os
 import sys
 
 import pyauto_functional # must be imported before pyauto
 import pyauto
 
 
 class ExtensionsPage(object):
-  """Access options in extensions page (chrome://settings/extensions)."""
+  """Access options in extensions page (chrome://extensions-frame)."""
 
-  _URL = 'chrome://settings/extensions'
+  _URL = 'chrome://extensions-frame'
 
   def __init__(self, driver):
     self._driver = driver
     self._driver.get(ExtensionsPage._URL)
 
   def CheckExtensionVisible(self, ext_id):
-    """Returns True if |ext_id| Enabled checkbox exists on page."""
-    return len(self._driver.find_elements_by_id('toggle-%s' % ext_id)) == 1
+    """Returns True if |ext_id| exists on page."""
+    return len(self._driver.find_elements_by_id(ext_id)) == 1
 
   def SetEnabled(self, ext_id, enabled):
     """Clicks on 'Enabled' checkbox for specified extension.
 
     Args:
       ext_id: Extension ID to be enabled or disabled.
       enabled: Boolean indicating whether |ext_id| is to be enabled or disabled.
     """
-    checkbox = self._driver.find_element_by_id('toggle-%s' % ext_id)
+    checkbox = self._driver.find_element_by_xpath(
+        '//*[@id="%s"]//*[@class="enable-controls"]//*[@type="checkbox"]' %
+        ext_id)
     if checkbox != enabled:
       checkbox.click()
     # Reload page to ensure that the UI is recreated.
     self._driver.get(ExtensionsPage._URL)
 
   def SetAllowInIncognito(self, ext_id, allowed):
     """Clicks on 'Allow in incognito' checkbox for specified extension.
 
     Args:
       ext_id: Extension ID to be enabled or disabled.
       allowed: Boolean indicating whether |ext_id| is to be allowed or
           disallowed in incognito.
     """
     checkbox = self._driver.find_element_by_xpath(
-        '//*[@id="%s"][@type="checkbox"]' % ext_id)
-    # Expand extension and click on 'Allow in incognito'.
-    if not checkbox.is_displayed():
-      self._driver.find_element_by_id('%s_zippy' % ext_id).click()
+        '//*[@id="%s"]//*[@class="incognito-control"]//*[@type="checkbox"]' %
+        ext_id)
     if checkbox.is_selected() != allowed:
       checkbox.click()
     # Reload page to ensure that the UI is recreated.
     self._driver.get(ExtensionsPage._URL)
 
   def SetAllowAccessFileURLs(self, ext_id, allowed):
     """Clicks on 'Allow access to file URLs' checkbox for specified extension.
 
     Args:
       ext_id: Extension ID to be enabled or disabled.
       allowed: Boolean indicating whether |ext_id| is to be allowed access to
           file URLs.
     """
     checkbox = self._driver.find_element_by_xpath(
-        '(//*[@id="%s"][@type="checkbox"])[2]' % ext_id)
-    # Expand extension and click on 'Allow access to file URLs'.
-    if not checkbox.is_displayed():
-      self._driver.find_element_by_id('%s_zippy' % ext_id).click()
+        '//*[@id="%s"]//*[@class="file-access-control"]//*[@type="checkbox"]' %
+        ext_id)
     if checkbox.is_selected() != allowed:
       checkbox.click()
 
 
 class ExtensionsTest(pyauto.PyUITest):
   """Test of extensions."""
 
   def Debug(self):
     """Test method for experimentation.
 
@@ skipping 367 matching lines @@
     ext_page.SetAllowAccessFileURLs(ext_id, True)
 
     # Check that extension now has access to file URLs.
     self.WaitUntil(self._FileAccess, args=[ext_id], expect_retval=True)
     self.assertTrue(self._FileAccess(ext_id),
                     msg='Extension did not have access to file URLs granted.')
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()