sandwich: Implements network condition on WPR server and browser.

tools/android/loading/trace_recorder.py

 #! /usr/bin/python
 # Copyright 2016 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.
 
 """Loading trace recorder."""
 
 import argparse
 import datetime
 import json
@@ skipping 13 matching lines @@
 
 import device_setup
 import devtools_monitor
 import loading_trace
 import page_track
 import request_track
 import tracing
 
 
 def MonitorUrl(connection, url, clear_cache=False,
-               categories=tracing.DEFAULT_CATEGORIES):
+               categories=tracing.DEFAULT_CATEGORIES,
+               timeout=devtools_monitor.DEFAULT_TIMEOUT):
   """Monitor a URL via a trace recorder.
 
   Args:
-    connection: A device_monitor.DevToolsConnection instance.
+    connection: A devtools_monitor.DevToolsConnection instance.
     url: url to navigate to as string.
     clear_cache: boolean indicating if cache should be cleared before loading.
     categories: List of tracing event categories to record.
+    timeout: Websocket timeout.
 
   Returns:
     loading_trace.LoadingTrace.
   """
   page = page_track.PageTrack(connection)
   request = request_track.RequestTrack(connection)
   trace = tracing.TracingTrack(connection, categories=categories)
   connection.SetUpMonitoring()
   if clear_cache:
     connection.ClearCache()
   connection.SendAndIgnoreResponse('Page.navigate', {'url': url})
-  connection.StartMonitoring()
+  connection.StartMonitoring(timeout=timeout)
   metadata = {'date': datetime.datetime.utcnow().isoformat(),
               'seconds_since_epoch': time.time()}
   return loading_trace.LoadingTrace(url, metadata, page, request, trace)
 
 def RecordAndDumpTrace(device, url, output_filename):
   with file(output_filename, 'w') as output,\
         device_setup.DeviceConnection(device) as connection:
     trace = MonitorUrl(connection, url)
     json.dump(trace.ToJsonDict(), output)
 
 
 def main():
   logging.basicConfig(level=logging.INFO)
   devil_chromium.Initialize()
 
   parser = argparse.ArgumentParser()
   parser.add_argument('--url', required=True)
   parser.add_argument('--output', required=True)
   args = parser.parse_args()
   url = args.url
   if not url.startswith('http'):
     url = 'http://' + url
   device = device_utils.DeviceUtils.HealthyDevices()[0]
   RecordAndDumpTrace(device, url, args.output)
 
 
 if __name__ == '__main__':
   main()