PyAuto tests to measure memory usage of Chrome-related processes.

These tests currently run on Linux/ChromeOS only.

The first test measures memory usage for a renderer process in a tab, and
the second test measures memory usage of an extension process.  This
initial implementation hard-codes the first test to use the "Angry Birds"
Chrome URL, and the second test to use the "Google Talk" extension.

Each test periodically measures the current memory usage of the relevant
process, and logs the information to a file.  This occurs repeatedly until
the test is manually killed by a user.

BUG=chromium-os:15225
TEST=None

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

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

1 files changed, 23 insertions, 0 deletions

diff --git a/chrome/test/functional/test_utils.py b/chrome/test/functional/test_utils.py
index a6010c3..4454940 100644
--- a/chrome/test/functional/test_utils.py
+++ b/chrome/test/functional/test_utils.py
@@ -9,6 +9,7 @@ import email
 import logging
 import os
 import smtplib
+import subprocess
 import types
 
 import pyauto_functional
@@ -234,3 +235,25 @@ def CallFunctionWithNewTimeout(self, new_timeout, function):
                % new_timeout)
   function()
   del timeout_changer
+
+
+def GetMemoryUsageOfProcess(pid):
+  """Queries the system for the current memory usage of a specified process.
+
+  This function only works in Linux and ChromeOS.
+
+  Args:
+    pid: The integer process identifier for the process to use.
+
+  Returns:
+    The memory usage of the process in MB, given as a float.  If the process
+    doesn't exist on the machine, then the value 0 is returned.
+  """
+  assert pyauto.PyUITest.IsLinux() or pyauto.PyUITest.IsChromeOS()
+  process = subprocess.Popen('ps h -o rss -p %s' % pid, shell=True,
+                             stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+  stdout = process.communicate()[0]
+  if stdout:
+    return float(stdout.strip()) / 1024
+  else:
+    return 0