PyAuto automation support for blocked popups

Add PyAuto automation hooks for fetching info about blocked popups in a tab,
and launching a blocked popup.

BUG=55889
TEST=popups.py

Review URL: http://codereview.chromium.org/3688004

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@62435 0039d316-1c4b-4281-b951-d872f2087c98

3 files changed, 86 insertions, 0 deletions

diff --git a/chrome/test/functional/PYAUTO_TESTS b/chrome/test/functional/PYAUTO_TESTS
index e7b3f7a..b4078b0 100644
--- a/chrome/test/functional/PYAUTO_TESTS
+++ b/chrome/test/functional/PYAUTO_TESTS
@@ -40,6 +40,7 @@
     'omnibox',
     'passwords',
     'plugins',
+    'popups',
     'prefs',
     'search_engines',
     'special_tabs',
diff --git a/chrome/test/functional/popups.py b/chrome/test/functional/popups.py
new file mode 100644
index 0000000..48b9831
--- /dev/null
+++ b/chrome/test/functional/popups.py
@@ -0,0 +1,54 @@
+#!/usr/bin/python
+
+# Copyright (c) 2010 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 os
+
+import pyauto_functional  # Must be imported before pyauto
+import pyauto
+
+
+class PopupsTest(pyauto.PyUITest):
+  """TestCase for Popup blocking."""
+
+  def Debug(self):
+    """Test method for experimentation.
+
+    This method will not run automatically.
+    """
+    import pprint
+    pp = pprint.PrettyPrinter(indent=2)
+    while True:
+      raw_input('Interact with the browser and hit <enter>')
+      pp.pprint(self.GetBlockedPopupsInfo())
+
+  def testPopupBlockerEnabled(self):
+    """Verify popup blocking is enabled."""
+    self.assertFalse(self.GetBlockedPopupsInfo(),
+                     msg='Should have no blocked popups on startup')
+    file_url = self.GetFileURLForPath(os.path.join(
+        self.DataDir(), 'popup_blocker', 'popup-blocked-to-post-blank.html'))
+    self.NavigateToURL(file_url)
+    blocked_popups = self.GetBlockedPopupsInfo()
+    self.assertEqual(1, len(blocked_popups), msg='Popup not blocked')
+    self.assertEqual('Popup Success!', blocked_popups[0]['title'])
+
+  def testLaunchBlockedPopup(self):
+    """Verify that a blocked popup can be unblocked."""
+    file_url = self.GetFileURLForPath(os.path.join(
+        self.DataDir(), 'popup_blocker', 'popup-blocked-to-post-blank.html'))
+    self.NavigateToURL(file_url)
+    self.assertEqual(1, len(self.GetBlockedPopupsInfo()))
+    self.UnblockAndLaunchBlockedPopup(0)
+    # Verify that no more popups are blocked
+    self.assertFalse(self.GetBlockedPopupsInfo())
+    # Verify that popup window was created
+    self.assertEqual(2, self.GetBrowserWindowCount(),
+                     msg='Popup could not be launched');
+    self.assertEqual('Popup Success!', self.GetActiveTabTitle(1))
+
+
+if __name__ == '__main__':
+  pyauto_functional.Main()
diff --git a/chrome/test/pyautolib/pyauto.py b/chrome/test/pyautolib/pyauto.py
index 80d08c76..8103795 100644
--- a/chrome/test/pyautolib/pyauto.py
+++ b/chrome/test/pyautolib/pyauto.py
@@ -1392,6 +1392,37 @@ class PyUITest(pyautolib.PyUITestBase, unittest.TestCase):
     }
     return self._GetResultFromJSONRequest(cmd_dict)['passwords']
 
+  def GetBlockedPopupsInfo(self, tab_index=0, windex=0):
+    """Get info about blocked popups in a tab.
+
+    Args:
+      tab_index: 0-based tab index. Default: 0
+      windex: 0-based window index. Default: 0
+
+    Returns:
+      [a list of property dictionaries for each blocked popup]
+      Property dictionary contains: title, url
+    """
+    cmd_dict = {
+      'command': 'GetBlockedPopupsInfo',
+      'tab_index': tab_index,
+    }
+    return self._GetResultFromJSONRequest(cmd_dict,
+                                          windex=windex)['blocked_popups']
+
+  def UnblockAndLaunchBlockedPopup(self, popup_index, tab_index=0, windex=0):
+    """Unblock/launch a poup at the given index.
+
+    This is equivalent to clicking on a blocked popup in the UI available
+    from the omnibox.
+    """
+    cmd_dict = {
+      'command': 'UnblockAndLaunchBlockedPopup',
+      'popup_index': popup_index,
+      'tab_index': tab_index,
+    }
+    self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+
   def SetTheme(self, crx_file_path):
     """Installs the given theme synchronously.