[telemetry_mini] Add UserStory class

experimental/telemetry_mini/android_go_stories.py

 #!/usr/bin/env python
 # Copyright 2017 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 argparse
 import logging
 import os
 import sys
 import time
@@ skipping 22 matching lines @@
 
 KEYCODE_BACK = 4
 
 BROWSERS = {
     'android-chrome': telemetry_mini.ChromeApp,
     'android-chromium': telemetry_mini.ChromiumApp,
     'android-system-chrome': telemetry_mini.SystemChromeApp,
 }
 
 
-class TwitterApp(telemetry_mini.AndroidApp):
-  PACKAGE_NAME = 'com.twitter.android'
-
-
 class ProcessWatcher(object):
   def __init__(self, device):
     self.device = device
     self._process_pid = {}
 
   def StartWatching(self, process_name):
     """Register a process or android app to keep track of its PID."""
     if isinstance(process_name, telemetry_mini.AndroidApp):
       process_name = process_name.PACKAGE_NAME
 
@@ skipping 55 matching lines @@
   # Finally check that only the selected browser is actually available.
   available_browsers = device.ListPackages('chrome', only_enabled=True)
   assert browser.PACKAGE_NAME in available_browsers, (
       'Unable to make %s available' % browser.PACKAGE_NAME)
   available_browsers.remove(browser.PACKAGE_NAME)
   assert not available_browsers, (
       'Other browsers may intefere with the test: %s' % available_browsers)
   return browser
 
 
-def RunStory(browser):
-  tracefile = 'trace.json'
-  device = browser.device
-  twitter = TwitterApp(device)
-  watcher = ProcessWatcher(device)
+class TwitterApp(telemetry_mini.AndroidApp):
+  PACKAGE_NAME = 'com.twitter.android'
 
-  with browser.Session(BROWSER_FLAGS, TRACE_CONFIG):
-    twitter.ForceStop()
-    try:
-      # Intent will launch Twitter app on Flipkart profile.
-      device.RunShellCommand(
-          'am', 'start', '-a', 'android.intent.action.VIEW',
-          '-d', 'https://twitter.com/flipkart')
-      watcher.StartWatching(twitter)
 
-      # Tapping on Flikpart link on Twitter app will launch Chrome.
-      device.TapUiNode(FLIPKART_TWITTER_LINK)
-      watcher.StartWatching(browser)
+class TwitterFlipkartStory(telemetry_mini.UserStory):
+  def __init__(self, *args, **kwargs):
+    super(TwitterFlipkartStory, self).__init__(*args, **kwargs)
+    self.watcher = ProcessWatcher(self.device)
+    self.twitter = TwitterApp(self.device)
 
-      time.sleep(4)
-      # Scroll content up a bit.
-      device.RunShellCommand(
-          'input', 'swipe', '240', '568', '240', '284', '400')
-      time.sleep(1)
+  def GetExtraStoryApps(self):
+    return (self.twitter,)
 
-      browser.CollectTrace(tracefile)
-      watcher.AssertAllAlive()
+  def RunStorySteps(self):
+    # Intent will launch Twitter app on Flipkart profile.
+    self.device.RunShellCommand(
+        'am', 'start', '-a', 'android.intent.action.VIEW',
+        '-d', 'https://twitter.com/flipkart')
+    self.watcher.StartWatching(self.twitter)
 
-      # Go "Back" and return to Twitter app.
-      device.RunShellCommand('input', 'keyevent', str(KEYCODE_BACK))
-      time.sleep(1)
-    finally:
-      twitter.ForceStop()
+    # Tapping on Flikpart link on Twitter app will launch Chrome.
+    self.device.TapUiNode(FLIPKART_TWITTER_LINK)
+    self.watcher.StartWatching(self.browser)
+
+    time.sleep(10)  # TODO: Replace with wait until page loaded.
+    # Scroll content up a bit.
+    self.device.RunShellCommand(
+        'input', 'swipe', '240', '568', '240', '284', '400')
+    time.sleep(1)
+
+    # Go "Back" and return to Twitter app.
+    self.device.RunShellCommand('input', 'keyevent', str(KEYCODE_BACK))
+    time.sleep(1)
+
+    self.watcher.AssertAllAlive()
 
 
 def main():
   browser_names = sorted(BROWSERS)
   default_browser = 'android-chrome'
   parser = argparse.ArgumentParser()
   parser.add_argument('--serial',
                       help='device serial on which to run user stories'
                       ' (defaults to first device found)')
   parser.add_argument('--adb-bin', default='adb', metavar='PATH',
@@ skipping 29 matching lines @@
     logging.warning('Connected to first device found: %s', device.serial)
   else:
     device = telemetry_mini.AdbMini(args.serial)
 
   # Some operations may require a rooted device.
   device.RunCommand('root')
   device.RunCommand('wait-for-device')
 
   browser = EnsureSingleBrowser(device, args.browser, args.force_install)
   browser.SetDevToolsLocalPort(args.port)
-  RunStory(browser)
+
+  story = TwitterFlipkartStory(browser)
+  story.Run(BROWSER_FLAGS, TRACE_CONFIG, 'trace.json')
+
 
 if __name__ == '__main__':
   sys.exit(main())