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)

[kkania, 2012/03/03 02:02:27]

bad indentation and several places below

[dyu1, 2012/03/03 02:16:45]

Done.

+    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)
+    plugins_exception = {
+          'pattern': 'http://maps.google.com:80', 'behavior': Behaviors.BLOCK,
+          'incognito': True}
+    page.AddNewException(**plugins_exception)

[kkania, 2012/03/03 02:02:27]

this is pretty fancy, but can you just do it the normal way

[dyu1, 2012/03/03 02:16:45]

Done.

+    self._VerifyContentExceptionUI(
+        ContentTypes.PLUGINS, 'http://maps.google.com:80',
+        Behaviors.BLOCK, incognito=True)
+
+  def testCorrectBehaviorSetInIncognitoUI(self):

[kkania, 2012/03/03 02:02:27]

remove?

[dyu1, 2012/03/03 02:16:45]

Removing since SetPrefs() function doesn't offer the capability of setting
behavior or patterns within Incognito mode. This test just happens to work based
on the behavior, thus removing until capability is added. 

On 2012/03/03 02:02:27, kkania wrote:

+    """Verify behavior is correctly set for initial Incognito list entry."""
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)  # Display incognito list.
+    # Ask for permission when site wants to track (Click to play).
+    self.SetPrefs(pyauto.kDefaultContentSettings, {'plugins': 3}, windex=1)
+    self._VerifyContentExceptionUI(
+        ContentTypes.PLUGINS, '*', Behaviors.ASK, incognito=True)
+    # Block all sites.
+    self.SetPrefs(pyauto.kDefaultContentSettings, {'plugins': 2}, windex=1)
+    self._VerifyContentExceptionUI(
+        ContentTypes.PLUGINS, '*', Behaviors.BLOCK, incognito=True)
+    # Allow all sites.
+    self.SetPrefs(pyauto.kDefaultContentSettings, {'plugins': 1}, windex=1)
+    self._VerifyContentExceptionUI(
+        ContentTypes.PLUGINS, '*', Behaviors.ALLOW, incognito=True)
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()