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 24 matching lines @@
     """
     behavior_key = self.GetPrefsInfo().Prefs(
         pyauto.kGeolocationDefaultContentSetting)
     behaviors_dict = {1: 'ALLOW', 2: 'BLOCK', 3: 'ASK'}
     self.assertTrue(
         behavior_key in behaviors_dict,
         msg=('Invalid default behavior key "%s" for "geolocation" content' %
              behavior_key))
     return behaviors_dict[behavior_key]
 
-  def _VerifyContentExceptionUI(self, content_type, hostname_pattern, behavior):
+  def _VerifyContentExceptionUI(self, content_type, hostname_pattern, behavior,
+                                incognito=False):
     """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.
+      incognito: Incognito list displayed on exceptions settings page.
+                 Default to False.
     """
     page = settings.ManageExceptionsPage.FromNavigation(
         self._driver, content_type)
-    self.assertTrue(page.GetExceptions().has_key(hostname_pattern),
+    self.assertTrue(page.GetExceptions(incognito).has_key(hostname_pattern),
                      msg=('No displayed host name matches pattern "%s"'
                           % hostname_pattern))
-    self.assertEqual(behavior, page.GetExceptions()[hostname_pattern],
+    self.assertEqual(behavior, page.GetExceptions(incognito)[hostname_pattern],
                      msg=('Displayed behavior "%s" does not match behavior "%s"'
-                          % (page.GetExceptions()[hostname_pattern], behavior)))
+                          % (page.GetExceptions(incognito)[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(
@@ skipping 87 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 testCorrectCookiesSessionInUI(self):
+    """Verify exceptions for cookies in UI list entry."""
+    # Block cookies for for a session for google.com.
+    self.SetPrefs(pyauto.kContentSettingsPatternPairs,
+                  {'http://google.com:80': {'cookies': 2}})
+    self._VerifyContentExceptionUI(
+        ContentTypes.COOKIES, 'http://google.com:80', Behaviors.BLOCK)
+
+  def testInitialLineEntryInIncognitoUI(self):
+    """Verify initial line entry is displayed in Incognito UI."""
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)  # Display incognito list.
+    page = settings.ManageExceptionsPage.FromNavigation(
+        self._driver, ContentTypes.PLUGINS)
+    self.assertEqual(1, len(page.GetExceptions(incognito=True)))
+
+  def testIncognitoExceptionsEntryCorrectlyDisplayed(self):
+    """Verify exceptions entry is correctly displayed in the incognito UI."""
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)  # Display incognito list.
+    page = settings.ManageExceptionsPage.FromNavigation(
+        self._driver, ContentTypes.PLUGINS)
+    pattern, behavior = ('http://maps.google.com:80', Behaviors.BLOCK)
+    page.AddNewException(pattern, behavior, incognito=True)
+    self._VerifyContentExceptionUI(
+        ContentTypes.PLUGINS, 'http://maps.google.com:80',
+        Behaviors.BLOCK, incognito=True)
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()