Upstream version 10.39.225.0

[platform/framework/web/crosswalk.git] / src / tools / telemetry / telemetry / core / platform / profiler / tcmalloc_heap_profiler.py

diff --git a/src/tools/telemetry/telemetry/core/platform/profiler/tcmalloc_heap_profiler.py b/src/tools/telemetry/telemetry/core/platform/profiler/tcmalloc_heap_profiler.py
index 3bbe6bf..14cf6f7 100644 (file)
--- a/src/tools/telemetry/telemetry/core/platform/profiler/tcmalloc_heap_profiler.py
+++ b/src/tools/telemetry/telemetry/core/platform/profiler/tcmalloc_heap_profiler.py
@@ -122,5 +122,19 @@ class TCMallocHeapProfiler(profiler.Profiler):
     options.AppendExtraBrowserArgs('--no-sandbox')
     options.AppendExtraBrowserArgs('--enable-memory-benchmarking')
 
+  @classmethod
+  def WillCloseBrowser(cls, browser_backend, platform_backend):
+    # The tcmalloc_heap_profiler dumps files at regular
+    # intervals (~20 secs).
+    # This is a minor optimization to ensure it'll dump the last file when
+    # the test completes.
+    for i in xrange(len(browser_backend.browser.tabs)):
+      browser_backend.browser.tabs[i].ExecuteJavaScript("""
+        if (chrome && chrome.memoryBenchmarking) {
+          chrome.memoryBenchmarking.heapProfilerDump('renderer', 'final');
+          chrome.memoryBenchmarking.heapProfilerDump('browser', 'final');
+        }
+      """)
+
   def CollectProfile(self):
     return self._platform_profiler.CollectProfile()