tools/perf/page_sets/memory_health_story.py

   1 # Copyright 2015 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import logging
   6 import re
   7
   8 from telemetry.core import util
   9 from telemetry.page import page as page_module
  10 from telemetry.page import shared_page_state
  11 from telemetry import story
  12
  13 util.AddDirToPythonPath(util.GetChromiumSrcDir(), 'build', 'android')
  14 from pylib.constants import keyevent # pylint: disable=import-error
  15 from pylib.device import intent # pylint: disable=import-error
  16
  17
  18 DUMP_WAIT_TIME = 3
  19
  20 URL_LIST = [
  21     'http://google.com',
  22     'http://vimeo.com',
  23     'http://yahoo.com',
  24     'http://baidu.com',
  25     'http://cnn.com',
  26     'http://yandex.ru',
  27     'http://yahoo.co.jp',
  28     'http://amazon.com',
  29     'http://ebay.com',
  30     'http://bing.com',
  31 ]
  32
  33
  34 class ForegroundPage(page_module.Page):
  35   """Take a measurement after loading a regular webpage."""
  36
  37   def __init__(self, story_set, name, url):
  38     super(ForegroundPage, self).__init__(
  39         url=url, page_set=story_set, name=name,
  40         shared_page_state_class=shared_page_state.SharedMobilePageState)
  41     self.archive_data_file = story_set.archive_data_file
  42
  43   def _TakeMemoryMeasurement(self, action_runner, phase):
  44     android_platform = action_runner.tab.browser.platform
  45     with action_runner.CreateInteraction(phase):
  46       action_runner.Wait(DUMP_WAIT_TIME)
  47       action_runner.ForceGarbageCollection()
  48       android_platform.RelaxMemory()
  49       action_runner.Wait(DUMP_WAIT_TIME)
  50       if not action_runner.tab.browser.DumpMemory():
  51         logging.error('Unable to get a memory dump for %s.', self.name)
  52
  53   def RunPageInteractions(self, action_runner):
  54     action_runner.tab.WaitForDocumentReadyStateToBeComplete()
  55     self._TakeMemoryMeasurement(action_runner, 'foreground')
  56
  57
  58 class BackgroundPage(ForegroundPage):
  59   """Take a measurement while Chrome is in the background."""
  60
  61   def __init__(self, story_set, name):
  62     super(BackgroundPage, self).__init__(story_set, name, 'about:blank')
  63
  64   def RunPageInteractions(self, action_runner):
  65     action_runner.tab.WaitForDocumentReadyStateToBeComplete()
  66
  67     # Launch clock app, pushing Chrome to the background.
  68     android_platform = action_runner.tab.browser.platform
  69     android_platform.LaunchAndroidApplication(
  70         intent.Intent(package='com.google.android.deskclock',
  71                       activity='com.android.deskclock.DeskClock'),
  72         app_has_webviews=False)
  73
  74     # Take measurement.
  75     self._TakeMemoryMeasurement(action_runner, 'background')
  76
  77     # Go back to Chrome.
  78     android_platform.android_action_runner.InputKeyEvent(keyevent.KEYCODE_BACK)
  79
  80
  81 class MemoryHealthStory(story.StorySet):
  82   """User story for the Memory Health Plan."""
  83
  84   def __init__(self):
  85     super(MemoryHealthStory, self).__init__(
  86         archive_data_file='data/memory_health_plan.json',
  87         cloud_storage_bucket=story.PARTNER_BUCKET)
  88
  89     for url in URL_LIST:
  90       # We name pages so their foreground/background counterparts are easy
  91       # to identify. For example 'http://google.com' becomes
  92       # 'http_google_com' and 'after_http_google_com' respectively.
  93       name = re.sub('\W+', '_', url)
  94       self.AddStory(ForegroundPage(self, name, url))
  95       self.AddStory(BackgroundPage(self, 'after_' + name))