Update pyauto tests to work with the uber page.

chrome/test/functional/policy_prefs_ui.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.
 
 # If this test is failing, please check these steps.
 #
 # - You introduced a new policy:
 #   Cool! Edit |policies| in policy_test_cases.py and add an entry for it.
 #   See the comment above it for the format.
@@ skipping 13 matching lines @@
 import policy_base
 import pyauto
 
 from policy_test_cases import PolicyPrefsTestCases
 
 
 class PolicyPrefsUITest(policy_base.PolicyTestBase):
   """Tests policies and their impact on the prefs UI."""
 
   settings_pages = [
-      'chrome://settings/browser',
-      'chrome://settings/personal',
-      'chrome://settings/advanced',
-      'chrome://settings/searchEngines',
-      'chrome://settings/passwords',
-      'chrome://settings/autofill',
-      'chrome://settings/content',
-      'chrome://settings/languages',
-      'chrome://settings/extensions',
+      'chrome://settings-frame',
+      'chrome://settings-frame/searchEngines',
+      'chrome://settings-frame/passwords',
+      'chrome://settings-frame/autofill',
+      'chrome://settings-frame/content',
+      'chrome://settings-frame/languages',
   ]
   if pyauto.PyUITest.IsChromeOS():
     settings_pages += [
-        'chrome://settings/system',
-        'chrome://settings/internet',
-        'chrome://settings/accounts',
+        'chrome://settings-frame/accounts',
     ]
 
   def GetPlatform(self):
     if self.IsChromeOS():
       return 'chromeos'
     elif self.IsLinux():
       return 'linux'
     elif self.IsMac():
       return 'mac'
     elif self.IsWin():
       return 'win'
     else:
       self.fail(msg='Unknown platform')
 
   def IsBannerVisible(self):
     """Returns true if the managed-banner is visible in the current page."""
+    # TODO(csilv): This logic assumes there is only one banner, it needs to be
+    # updated to work with uber page.
     ret = self.ExecuteJavascript("""
         var visible = false;
         var banner = document.getElementById('managed-prefs-banner');
         if (banner)
           visible = !banner.hidden;
         domAutomationController.send(visible.toString());
     """)
     return ret == 'true'
 
   def testNoPoliciesNoBanner(self):
@@ skipping 95 matching lines @@
 
     self.SetPolicies(policy_dict)
     self.assertTrue(self.IsBannerVisible())
 
     self.SetPolicies({})
     self.assertFalse(self.IsBannerVisible())
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()