Fetch a bunch of info from the browser

This includes info about the browser/renderer/extension/other PIDs, window size.

Add a method to set window size.
Add tests to verify window size, and verify flash loading.

BUG=43234
TEST=python chrome/test/functional/browser.py

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

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

3 files changed, 103 insertions, 15 deletions

diff --git a/chrome/test/functional/PYAUTO_TESTS b/chrome/test/functional/PYAUTO_TESTS
index a0c80c0..656f2b99 100644
--- a/chrome/test/functional/PYAUTO_TESTS
+++ b/chrome/test/functional/PYAUTO_TESTS
@@ -44,6 +44,7 @@
 
   'linux': [
     '-omnibox',  # http://crbug.com/44203
+    '-browser.BrowserTest.testWindowResize',  # crbug.com/44963
   ],
 
   # TODO(nirnimesh): Add a ChromeOS section
diff --git a/chrome/test/functional/browser.py b/chrome/test/functional/browser.py
index 7d2b08b..05aea1e 100644
--- a/chrome/test/functional/browser.py
+++ b/chrome/test/functional/browser.py
@@ -3,6 +3,7 @@
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
+import os
 import re
 
 import pyauto_functional  # Must be imported before pyauto
@@ -21,14 +22,38 @@ class BrowserTest(pyauto.PyUITest):
     pp = pprint.PrettyPrinter(indent=2)
     while True:
       raw_input('Hit <enter> to dump info.. ')
-      properties = self.GetBrowserInfo()
-      self.assertTrue(properties)
-      pp.pprint(properties)
+      info = self.GetBrowserInfo()
+      pp.pprint(info)
 
   def testGotVersion(self):
     """Verify there's a valid version string."""
-    version_string = self.GetBrowserInfo()['ChromeVersion']
-    self.assertTrue(re.match("\d+\.\d+\.\d+.\.\d+", version_string))
+    version_string = self.GetBrowserInfo()['properties']['ChromeVersion']
+    self.assertTrue(re.match('\d+\.\d+\.\d+.\.\d+', version_string))
+
+  def testWindowResize(self):
+    """Verify window resizing and persistence after restart."""
+    def _VerifySize(x, y, width, height):
+      info = self.GetBrowserInfo()
+      self.assertEqual(x, info['windows'][0]['x'])
+      self.assertEqual(y, info['windows'][0]['y'])
+      self.assertEqual(width, info['windows'][0]['width'])
+      self.assertEqual(height, info['windows'][0]['height'])
+    self.SetWindowDimensions(x=20, y=40, width=600, height=300)
+    _VerifySize(20, 40, 600, 300)
+    self.RestartBrowser(clear_profile=False)
+    _VerifySize(20, 40, 600, 300)
+
+  def testCanLoadFlash(self):
+    """Verify that we can play Flash.
+
+    We merely check that the flash process kicks in.
+    """
+    flash_url = self.GetFileURLForPath(os.path.join(self.DataDir(),
+                                                    'plugin', 'flash.swf'))
+    self.NavigateToURL(flash_url)
+    child_processes = self.GetBrowserInfo()['child_processes']
+    self.assertTrue([x for x in child_processes
+        if x['type'] == 'Plug-in' and x['name'] == 'Shockwave Flash'])
 
 
 if __name__ == '__main__':
diff --git a/chrome/test/pyautolib/pyauto.py b/chrome/test/pyautolib/pyauto.py
index 42102402..3ef206c 100644
--- a/chrome/test/pyautolib/pyauto.py
+++ b/chrome/test/pyautolib/pyauto.py
@@ -434,24 +434,86 @@ class PyUITest(pyautolib.PyUITestBase, unittest.TestCase):
     self.assertTrue(self.WaitUntil(
         lambda: len(self.GetDownloadsInfo().Downloads()) == num_downloads + 1))
 
+  def SetWindowDimensions(
+      self, x=None, y=None, width=None, height=None, windex=0):
+    """Set window dimensions.
+
+    All args are optional and current values will be preserved.
+    Arbitrarily large values will be handled gracefully by the browser.
+
+    Args:
+      x: window origin x
+      y: window origin y
+      width: window width
+      height: window height
+      windex: window index to work on. Defaults to 0 (first window)
+    """
+    cmd_dict = {  # Prepare command for the json interface
+      'command': 'SetWindowDimensions',
+    }
+    if x:
+      cmd_dict['x'] = x
+    if y:
+      cmd_dict['y'] = y
+    if width:
+      cmd_dict['width'] = width
+    if height:
+      cmd_dict['height'] = height
+    ret_dict = json.loads(self._SendJSONRequest(0, json.dumps(cmd_dict)))
+    if ret_dict.has_key('error'):
+      raise JSONInterfaceError(ret_dict['error'])
+    return ret_dict
+
   def GetBrowserInfo(self):
     """Return info about the browser.
 
     This includes things like the version number, the executable name,
-    executable path, and so on.
+    executable path, pid info about the renderer/plugin/extension processes,
+    window dimensions. (See sample below)
 
     Returns:
-      a dictionary of properties about the browser
+      a dictionary
+
       Sample:
-        { u'BrowserProcessExecutableName': u'Chromium',
-          u'BrowserProcessExecutablePath': u'Chromium.app/Contents/'
-                                            'MacOS/Chromium',
-          u'ChromeVersion': u'6.0.401.0',
+      { u'browser_pid': 93737,
+        # Child processes are the processes for plugins and other workers.
+        u'child_process_path': u'.../Chromium.app/Contents/'
+                                'Versions/6.0.412.0/Chromium Helper.app/'
+                                'Contents/MacOS/Chromium Helper',
+        u'child_processes': [ { u'name': u'Shockwave Flash',
+                                u'pid': 93766,
+                                u'type': u'Plug-in'}],
+        # There's one extension process per extension.
+        u'extension_processes': [
+          { u'name': u'Webpage Screenshot', u'pid': 93938},
+          { u'name': u'Google Voice (by Google)', u'pid': 93852}],
+        u'properties': {
+          u'BrowserProcessExecutableName': u'Chromium',
+          u'BrowserProcessExecutablePath': u'Chromium.app/Contents/MacOS/'
+                                            'Chromium',
+          u'ChromeVersion': u'6.0.412.0',
           u'HelperProcessExecutableName': u'Chromium Helper',
           u'HelperProcessExecutablePath': u'Chromium Helper.app/Contents/'
-                                           'MacOS/Chromium Helper',
-          u'command_line_string': "command_line_string --with-flags"
-        }
+                                            'MacOS/Chromium Helper',
+          u'command_line_string': "COMMAND_LINE_STRING --WITH-FLAGS"},
+        # The order of the windows and tabs listed here will be the same as
+        # what shows up on screen.
+        u'windows': [ { u'index': 0,
+                        u'height': 1134,
+                        u'incognito': False,
+                        u'is_fullscreen': False,
+                        u'selected_tab': 0,
+                        u'tabs': [ { u'index': 0,
+                                     u'num_infobars': 0,
+                                     u'renderer_pid': 93747,
+                                     u'url': u'http://www.google.com/'},
+                                   { u'index': 1,
+                                     u'num_infobars': 0,
+                                     u'renderer_pid': 93919,
+                                     u'url': u'https://chrome.google.com/'}],
+                        u'width': 925,
+                        u'x': 26,
+                        u'y': 44}]}
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
@@ -462,7 +524,7 @@ class PyUITest(pyautolib.PyUITestBase, unittest.TestCase):
     ret_dict = json.loads(self._SendJSONRequest(0, json.dumps(cmd_dict)))
     if ret_dict.has_key('error'):
       raise JSONInterfaceError(ret_dict['error'])
-    return ret_dict['properties']
+    return ret_dict
 
   def GetHistoryInfo(self, search_text=''):
     """Return info about browsing history.