src/chrome/test/functional/prefs.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import logging
   7 import os
   8 import shutil
   9
  10 import pyauto_functional  # Must be imported before pyauto
  11 import pyauto
  12 import test_utils
  13
  14 from webdriver_pages import settings
  15 from webdriver_pages.settings import Behaviors, ContentTypes
  16
  17
  18 class PrefsTest(pyauto.PyUITest):
  19   """TestCase for Preferences."""
  20
  21   INFOBAR_TYPE = 'rph_infobar'
  22
  23   def setUp(self):
  24     pyauto.PyUITest.setUp(self)
  25     self._driver = self.NewWebDriver()
  26
  27   def Debug(self):
  28     """Test method for experimentation.
  29
  30     This method will not run automatically.
  31     """
  32     while True:
  33       raw_input('Interact with the browser and hit <enter> to dump prefs... ')
  34       self.pprint(self.GetPrefsInfo().Prefs())
  35
  36   def testSessionRestore(self):
  37     """Test session restore preference."""
  38     url1 = 'http://www.google.com/'
  39     url2 = 'http://news.google.com/'
  40     self.NavigateToURL(url1)
  41     self.AppendTab(pyauto.GURL(url2))
  42     num_tabs = self.GetTabCount()
  43     # Set pref to restore session on startup.
  44     self.SetPrefs(pyauto.kRestoreOnStartup, 1)
  45     logging.debug('Setting %s to 1' % pyauto.kRestoreOnStartup)
  46     self.RestartBrowser(clear_profile=False)
  47     self.assertEqual(self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup), 1)
  48     self.assertEqual(num_tabs, self.GetTabCount())
  49     self.ActivateTab(0)
  50     self.assertEqual(url1, self.GetActiveTabURL().spec())
  51     self.ActivateTab(1)
  52     self.assertEqual(url2, self.GetActiveTabURL().spec())
  53
  54   def testNavigationStateOnSessionRestore(self):
  55     """Verify navigation state is preserved on session restore."""
  56     urls = ('http://www.google.com/',
  57             'http://news.google.com/',
  58             'http://dev.chromium.org/',)
  59     for url in urls:
  60       self.NavigateToURL(url)
  61     self.TabGoBack()
  62     self.assertEqual(self.GetActiveTabURL().spec(), urls[-2])
  63     self.SetPrefs(pyauto.kRestoreOnStartup, 1)  # set pref to restore session
  64     self.RestartBrowser(clear_profile=False)
  65     # Verify that navigation state (forward/back state) is restored.
  66     self.TabGoBack()
  67     self.assertEqual(self.GetActiveTabURL().spec(), urls[0])
  68     for i in (-2, -1):
  69       tab.GoForward()
  70       self.assertEqual(self.GetActiveTabURL().spec(), urls[i])
  71
  72   def testSessionRestoreURLs(self):
  73     """Verify restore URLs preference."""
  74     url1 = self.GetFileURLForPath(os.path.join(self.DataDir(), 'title1.html'))
  75     url2 = self.GetFileURLForPath(os.path.join(self.DataDir(), 'title2.html'))
  76     # Set pref to restore given URLs on startup
  77     self.SetPrefs(pyauto.kRestoreOnStartup, 4)  # 4 is for restoring URLs
  78     self.SetPrefs(pyauto.kURLsToRestoreOnStartup, [url1, url2])
  79     self.RestartBrowser(clear_profile=False)
  80     # Verify
  81     self.assertEqual(self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup), 4)
  82     self.assertEqual(2, self.GetTabCount())
  83     self.ActivateTab(0)
  84     self.assertEqual(url1, self.GetActiveTabURL().spec())
  85     self.ActivateTab(1)
  86     self.assertEqual(url2, self.GetActiveTabURL().spec())
  87
  88   def testGeolocationPref(self):
  89     """Verify geolocation pref.
  90
  91     Checks for the geolocation infobar.
  92     """
  93     # GetBrowserInfo() call seems to fail later on in this test. Call it early.
  94     # crbug.com/89000
  95     branding = self.GetBrowserInfo()['properties']['branding']
  96     url = self.GetFileURLForPath(os.path.join(  # triggers geolocation
  97         self.DataDir(), 'geolocation', 'geolocation_on_load.html'))
  98     self.assertEqual(3,  # default state
  99         self.GetPrefsInfo().Prefs(pyauto.kGeolocationDefaultContentSetting))
 100     self.NavigateToURL(url)
 101     self.assertTrue(self.WaitForInfobarCount(1))
 102     self.assertTrue(self.GetBrowserInfo()['windows'][0]['tabs'][0]['infobars'])
 103     # Disable geolocation
 104     self.SetPrefs(pyauto.kGeolocationDefaultContentSetting, 2)
 105     self.assertEqual(2,
 106         self.GetPrefsInfo().Prefs(pyauto.kGeolocationDefaultContentSetting))
 107     self.ReloadTab()
 108     # Fails on Win7/Vista Chromium bots.  crbug.com/89000
 109     if (self.IsWin7() or self.IsWinVista()) and branding == 'Chromium':
 110       return
 111     behavior = self._driver.execute_async_script(
 112         'triggerGeoWithCallback(arguments[arguments.length - 1]);')
 113     self.assertEqual(
 114         behavior, Behaviors.BLOCK,
 115         msg='Behavior is "%s" when it should be BLOCKED.'  % behavior)
 116
 117   def testAllowSelectedGeoTracking(self):
 118     """Verify hostname pattern and behavior for allowed tracking."""
 119     # Default location tracking option "Ask me".
 120     self.SetPrefs(pyauto.kGeolocationDefaultContentSetting, 3)
 121     self.NavigateToURL(
 122         self.GetHttpURLForDataPath('geolocation', 'geolocation_on_load.html'))
 123     self.assertTrue(self.WaitForInfobarCount(1))
 124     self.PerformActionOnInfobar('accept', infobar_index=0)  # Allow tracking.
 125     # Get the hostname pattern (e.g. http://127.0.0.1:57622).
 126     hostname_pattern = (
 127         '/'.join(self.GetHttpURLForDataPath('').split('/')[0:3]))
 128     self.assertEqual(
 129         # Allow the hostname.
 130         {hostname_pattern+','+hostname_pattern: {'geolocation': 1}},
 131         self.GetPrefsInfo().Prefs(pyauto.kContentSettingsPatternPairs))
 132
 133   def testDismissedInfobarSavesNoEntry(self):
 134     """Verify dismissing infobar does not save an exception entry."""
 135     # Default location tracking option "Ask me".
 136     self.SetPrefs(pyauto.kGeolocationDefaultContentSetting, 3)
 137     self.NavigateToURL(
 138         self.GetFileURLForDataPath('geolocation', 'geolocation_on_load.html'))
 139     self.assertTrue(self.WaitForInfobarCount(1))
 140     self.PerformActionOnInfobar('dismiss', infobar_index=0)
 141     self.assertEqual(
 142         {}, self.GetPrefsInfo().Prefs(pyauto.kContentSettingsPatternPairs))
 143
 144   def testGeolocationBlockedWhenTrackingDenied(self):
 145     """Verify geolocations is blocked when tracking is denied.
 146
 147     The test verifies the blocked hostname pattern entry on the Geolocations
 148     exceptions page.
 149     """
 150     # Ask for permission when site wants to track.
 151     self.SetPrefs(pyauto.kGeolocationDefaultContentSetting, 3)
 152     self.NavigateToURL(
 153         self.GetHttpURLForDataPath('geolocation', 'geolocation_on_load.html'))
 154     self.assertTrue(self.WaitForInfobarCount(1))
 155     self.PerformActionOnInfobar('cancel', infobar_index=0)  # Deny tracking.
 156     behavior = self._driver.execute_async_script(
 157         'triggerGeoWithCallback(arguments[arguments.length - 1]);')
 158     self.assertEqual(
 159         behavior, Behaviors.BLOCK,
 160         msg='Behavior is "%s" when it should be BLOCKED.'  % behavior)
 161     # Get the hostname pattern (e.g. http://127.0.0.1:57622).
 162     hostname_pattern = (
 163         '/'.join(self.GetHttpURLForDataPath('').split('/')[0:3]))
 164     self.assertEqual(
 165         # Block the hostname.
 166         {hostname_pattern+','+hostname_pattern: {'geolocation': 2}},
 167         self.GetPrefsInfo().Prefs(pyauto.kContentSettingsPatternPairs))
 168
 169   def _CheckForVisibleImage(self, tab_index=0, windex=0):
 170     """Checks whether or not an image is visible on the webpage.
 171
 172     Args:
 173       tab_index: Tab index. Defaults to 0 (first tab).
 174       windex: Window index. Defaults to 0 (first window).
 175
 176     Returns:
 177       True if image is loaded, otherwise returns False if image is not loaded.
 178     """
 179     # Checks whether an image is loaded by checking the area (width
 180     # and height) of the image. If the area is non zero then the image is
 181     # visible. If the area is zero then the image is not loaded.
 182     # Chrome zeros the |naturalWidth| and |naturalHeight|.
 183     script = """
 184       for (i=0; i < document.images.length; i++) {
 185         if ((document.images[i].naturalWidth != 0) &&
 186             (document.images[i].naturalHeight != 0)) {
 187           window.domAutomationController.send(true);
 188         }
 189       }
 190       window.domAutomationController.send(false);
 191     """
 192     return self.ExecuteJavascript(script, windex=windex, tab_index=tab_index)
 193
 194   def testBlockImagesForHostname(self):
 195     """Verify images blocked for defined hostname pattern."""
 196     url = 'http://www.google.com'
 197     page = settings.ManageExceptionsPage.FromNavigation(
 198         self._driver, ContentTypes.IMAGES)
 199     pattern, behavior = (url, Behaviors.BLOCK)
 200     # Add an exception BLOCK for hostname pattern 'www.google.com'.
 201     page.AddNewException(pattern, behavior)
 202     self.NavigateToURL(url)
 203     self.assertFalse(self._CheckForVisibleImage(),
 204                      msg='At least one visible image found.')
 205
 206   def testAllowImagesForHostname(self):
 207     """Verify images allowed for defined hostname pattern."""
 208     url = 'http://www.google.com'
 209     page = settings.ManageExceptionsPage.FromNavigation(
 210         self._driver, ContentTypes.IMAGES)
 211     pattern, behavior = (url, Behaviors.ALLOW)
 212     # Add an exception ALLOW for hostname pattern 'www.google.com'.
 213     page.AddNewException(pattern, behavior)
 214     self.NavigateToURL(url)
 215     self.assertTrue(self._CheckForVisibleImage(),
 216                     msg='No visible images found.')
 217
 218   def testProtocolHandlerRegisteredCorrectly(self):
 219     """Verify sites that ask to be default handlers registers correctly."""
 220     url = self.GetHttpURLForDataPath('settings', 'protocol_handler.html')
 221     self.NavigateToURL(url)
 222     # Returns a dictionary with the custom handler.
 223     asked_handler_dict = self._driver.execute_script(
 224         'return registerCustomHandler()')
 225     self.PerformActionOnInfobar(
 226         'accept', infobar_index=test_utils.WaitForInfobarTypeAndGetIndex(
 227             self, self.INFOBAR_TYPE))
 228     self._driver.find_element_by_id('test_protocol').click()
 229     self.assertTrue(
 230         self._driver.execute_script(
 231             'return doesQueryConformsToProtocol("%s", "%s")'
 232             % (asked_handler_dict['query_key'],
 233                asked_handler_dict['query_value'])),
 234         msg='Protocol did not register correctly.')
 235
 236
 237 if __name__ == '__main__':
 238   pyauto_functional.Main()