Add prefs UI tests for Incognito exceptions for new Settings UI.

chrome/test/functional/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.
 
 import pyauto_functional  # Must be imported before pyauto
 import pyauto
 from webdriver_pages import settings
 from webdriver_pages.settings import Behaviors, ContentTypes
 
@@ skipping 41 matching lines @@
     """
     page = settings.ManageExceptionsPage.FromNavigation(
         self._driver, content_type)
     self.assertTrue(page.GetExceptions().has_key(hostname_pattern),
                      msg=('No displayed host name matches pattern "%s"'
                           % hostname_pattern))
     self.assertEqual(behavior, page.GetExceptions()[hostname_pattern],
                      msg=('Displayed behavior "%s" does not match behavior "%s"'
                           % (page.GetExceptions()[hostname_pattern], behavior)))
 
+  def _VerifyIncognitoContentExceptionUI(
+      self, content_type, hostname_pattern, behavior):
+    """Find hostname pattern and behavior within UI on content exceptions page.
+
+    Args:
+      content_type: The string content settings type to manage.
+      hostname_pattern: The URL or pattern associated with the behavior.
+      behavior: The exception to allow or block the hostname.
+    """
+    page = settings.ManageExceptionsPage.FromNavigation(
+        self._driver, content_type)
+    self.assertTrue(
+        page.GetExceptions(incognito=True).has_key(hostname_pattern),
+        msg='No displayed host name matches pattern "%s"' % hostname_pattern)
+    self.assertEqual(
+        behavior, page.GetExceptions(incognito=True)[hostname_pattern],
+        msg=('Displayed behavior "%s" does not match behavior "%s"'
+             % (page.GetExceptions(incognito=True)[hostname_pattern],
+                behavior)))
+
   def testLocationSettingOptionsUI(self):
     """Verify the location options setting UI.
 
     Set the options through the UI using webdriver and verify the settings in
     pyAuto.
     """
     page = settings.ContentSettingsPage.FromNavigation(self._driver)
     page.SetContentTypeOption(ContentTypes.GEOLOCATION, Behaviors.ALLOW)
     self.assertEqual(
         1, self.GetPrefsInfo().Prefs(pyauto.kGeolocationDefaultContentSetting))
@@ skipping 35 matching lines @@
     self.SetPrefs(pyauto.kGeolocationContentSettings, geo_exception)
     self._VerifyContentExceptionUI(
         ContentTypes.GEOLOCATION, 'http://maps.google.com:80',
         Behaviors.ALLOW)
     geo_exception = (
         {'http://maps.google.com:80/': {'http://maps.google.com:80': 3}})
     self.SetPrefs(pyauto.kGeolocationContentSettings, geo_exception)
     self._VerifyContentExceptionUI(
         ContentTypes.GEOLOCATION, 'http://maps.google.com:80', Behaviors.ASK)
 
+  def testIncognitoExceptionsEntryCorrectlyDisplayed(self):
+    """Verify exceptions entry is correctly displayed in the incognito UI."""
+    geo_exception = (
+        {'http://maps.google.com:80/': {'http://maps.google.com': 2}})
+    self.SetPrefs(pyauto.kGeolocationContentSettings, geo_exception)
+    self._VerifyIncognitoContentExceptionUI(

[kkania, 2012/02/27 00:56:35]

how about do all this without loading the page thrice?

[dyu1, 2012/02/27 20:54:45]

I'm not sure what you mean since this test is similar to
testExceptionsEntryCorrectlyDisplayed. The only difference is using
self._VerifyContentExceptionUI vs self.VerifyIncognitoContentExceptionsUI. The
previous test was given a LGTM, so what has changed? 

[kkania, 2012/02/28 00:51:30]

I didn't think about it. How about something like this:

geo_exception = (
    {'http://maps.google.com:80/': {'http://maps.google.com': 2},
     'http://gmail.com:80/': {'h.....'},
     '...': '...'})
self.SetPrefs(pyauto.kGeolocationContentSettings, geo_exception)
page = settings.ManageExceptionsPage.FromNavigation(self._driver,
ContentTypes.GEOLOCATION)
exceptions = page.GetExceptions(incognito=True)
expected_exceptions = {
  'http://maps.google.com:80/': Behaviors.BLOCK,
  'http://gmail.com:80/': Behaviors.ASK,
  'http://yahoo.com:80': Behaviors.DENY]
for host, behavior in expected_exceptions.iteritems():
  self.assertTrue(exceptions.has_key(host),
                  msg='No displayed host name matches pattern "%s"' % host)
  self.assertEqual(expected_exceptions[host], exceptions[host])

+        ContentTypes.GEOLOCATION, 'http://maps.google.com:80',
+        Behaviors.BLOCK)
+    geo_exception = (
+        {'http://maps.google.com:80/': {'http://maps.google.com:80': 1}})
+    self.SetPrefs(pyauto.kGeolocationContentSettings, geo_exception)
+    self._VerifyIncognitoContentExceptionUI(
+        ContentTypes.GEOLOCATION, 'http://maps.google.com:80',
+        Behaviors.ALLOW)
+    geo_exception = (
+        {'http://maps.google.com:80/': {'http://maps.google.com:80': 3}})
+    self.SetPrefs(pyauto.kGeolocationContentSettings, geo_exception)
+    self._VerifyIncognitoContentExceptionUI(
+        ContentTypes.GEOLOCATION, 'http://maps.google.com:80', Behaviors.ASK)
+
   def testAddNewExceptionUI(self):
     """Verify new exception added for hostname pattern and behavior in UI."""
     content_type = ContentTypes.PLUGINS
     page = settings.ManageExceptionsPage.FromNavigation(
         self._driver, content_type)
 
     pattern, behavior = ('bing.com', Behaviors.BLOCK)
     page.AddNewException(pattern, behavior)
     self.assertEqual(page.GetExceptions()[pattern], Behaviors.BLOCK,
                      msg='The behavior "%s" was not added for pattern "%s"'
@@ skipping 31 matching lines @@
   def testNoInitialLineEntryInUI(self):
     """Verify no initial line entry is displayed in UI."""
     # Ask for permission when site wants to track.
     self.SetPrefs(pyauto.kGeolocationDefaultContentSetting, 3)
     self.assertEqual(
         3, self.GetPrefsInfo().Prefs(pyauto.kGeolocationDefaultContentSetting))
     page = settings.ManageExceptionsPage.FromNavigation(
         self._driver, ContentTypes.GEOLOCATION)
     self.assertEqual(0, len(page.GetExceptions()))
 
+  def testInitialLineEntryInIncognitoUI(self):
+    """Verify initial line entry is displayed in Incognito UI."""
+    # Ask for permission when site wants to track.
+    self.SetPrefs(pyauto.kGeolocationDefaultContentSetting, 3)
+    self.assertEqual(
+        3, self.GetPrefsInfo().Prefs(pyauto.kGeolocationDefaultContentSetting))
+    page = settings.ManageExceptionsPage.FromNavigation(
+        self._driver, ContentTypes.GEOLOCATION)
+    self.assertEqual(1, len(page.GetExceptions(incognito=True)))
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()