Pyauto process count tests now get browser executable name via an automation hook.

chrome/test/functional/process_count.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 os
 
 import pyauto_functional
 import pyauto
 
 
 class ProcessCountTest(pyauto.PyUITest):
   """Tests to ensure the number of Chrome-related processes is as expected."""
 
   FRESH_PROFILE_PROC_COUNT = {
     'win': 2,  # Processes: browser, tab.
     'mac': 2,  # Processes: browser, tab.
     'linux': 4,  # Processes: browser, tab, sandbox helper, zygote.
     'chromeos': 4,  # Processes: browser, tab, sandbox helper, zygote.
   }
 
-  CHROME_PROCESS_NAME = {
-    'win': 'chrome.exe',
-    'mac': 'Chromium',
-    'linux': 'chrome',
-    'chromeos': 'chrome',
-  }
-
   def Debug(self):
     """Test method for experimentation.
 
     This method will not run automatically.
     """
     while True:
       raw_input('Hit <enter> to dump process info...')
       self.pprint(self.GetProcessInfo())
 
   def setUp(self):
     self.proc_count_fresh_profile = 0
-    self.chrome_proc_name = ''
     if self.IsChromeOS():
       self.proc_count_fresh_profile = self.FRESH_PROFILE_PROC_COUNT['chromeos']
-      self.chrome_proc_name = self.CHROME_PROCESS_NAME['chromeos']
     elif self.IsWin():
       self.proc_count_fresh_profile = self.FRESH_PROFILE_PROC_COUNT['win']
-      self.chrome_proc_name = self.CHROME_PROCESS_NAME['win']
     elif self.IsMac():
       self.proc_count_fresh_profile = self.FRESH_PROFILE_PROC_COUNT['mac']
-      self.chrome_proc_name = self.CHROME_PROCESS_NAME['mac']
     elif self.IsLinux():
       self.proc_count_fresh_profile = self.FRESH_PROFILE_PROC_COUNT['linux']
-      self.chrome_proc_name = self.CHROME_PROCESS_NAME['linux']
 
     pyauto.PyUITest.setUp(self)
 
     self.NavigateToURL('about:blank')
 
   def _VerifyProcessCount(self, num_expected):
     """Verifies the number of Chrome-related processes is as expected.
 
     Args:
       num_expected: An integer representing the expected number of Chrome-
                     related processes that should currently exist.
     """
+    browser_process_name = (
+        self.GetBrowserInfo()['properties']['BrowserProcessExecutableName'])
     proc_info = self.GetProcessInfo()
     browser_info = [x for x in proc_info['browsers']
-                    if x['process_name'] == self.chrome_proc_name]
+                    if x['process_name'] == browser_process_name]
     assert len(browser_info) == 1
     # Utility & GPU processes may show up any time.  Ignore them.
     processes = [x for x in browser_info[0]['processes']
                  if x['child_process_type'] not in ('Utility', 'GPU')]
     num_actual = len(processes)
 
     self.assertEqual(num_actual, num_expected,
                      msg='Number of processes (ignoring Utility/GPU processes) '
                          'should be %d, but was %d.\n'
                          'Actual processes:\n%s' % (
@@ skipping 53 matching lines @@
     self.OpenNewBrowserWindow(True)
 
     for _ in xrange(3):
       self.AppendTab(pyauto.GURL('about:blank'), 1)
 
     self._VerifyProcessCount(self.proc_count_fresh_profile + 8)
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()