page_runner.py revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd
12a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Copyright (c) 2012 The Chromium Authors. All rights reserved. 22a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# Use of this source code is governed by a BSD-style license that can be 32a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)# found in the LICENSE file. 42a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import codecs 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import logging 62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import os 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import time 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import traceback 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import urlparse 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)import random 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from telemetry.core import util 132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from telemetry.core import wpr_modes 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from telemetry.core import exceptions 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from telemetry.page import page_filter as page_filter_module 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)from telemetry.page import page_test 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class PageState(object): 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self): 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.did_login = False 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class _RunState(object): 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self): 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.first_browser = True 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.browser = None 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.tab = None 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.is_tracing = False 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Close(self): 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.is_tracing = False 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if self.tab: 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.tab.Disconnect() 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.tab = None 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if self.browser: 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.browser.Close() 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.browser = None 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)def _ShuffleAndFilterPageSet(page_set, options): 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.pageset_shuffle_order_file and not options.pageset_shuffle: 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise Exception('--pageset-shuffle-order-file requires --pageset-shuffle.') 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.pageset_shuffle_order_file: 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return page_set.ReorderPageSet(options.pageset_shuffle_order_file) 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page_filter = page_filter_module.PageFilter(options) 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pages = [page for page in page_set.pages[:] 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not page.disabled and page_filter.IsSelected(page)] 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.pageset_shuffle: 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) random.Random().shuffle(pages) 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return [page 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for _ in xrange(int(options.pageset_repeat)) 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for page in pages 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for _ in xrange(int(options.page_repeat))] 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class PageRunner(object): 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """Runs a given test against a given test.""" 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __init__(self, page_set): 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.page_set = page_set 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __enter__(self): 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return self 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def __exit__(self, *args): 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.Close() 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Run(self, options, possible_browser, test, results): 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Reorder page set based on options. 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pages = _ShuffleAndFilterPageSet(self.page_set, options) 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Check if we can run against WPR. 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pages_without_archives = [] 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for page in pages: 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) parsed_url = urlparse.urlparse(page.url) 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if parsed_url.scheme == 'file': 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not page.archive_path: 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.allow_live_sites: 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning(""" 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) No page set archive provided for the page %s. Benchmarking against live sites! 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Results won't be repeatable or comparable. 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)""", page.url) 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning(""" 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) No page set archive provided for the page %s. Not running the page. To run 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) against live sites, pass the flag --allow-live-sites. 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)""", page.url) 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, 'Page set archive not defined', '') 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pages_without_archives.append(page) 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) elif options.wpr_mode != wpr_modes.WPR_RECORD: 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # The page has an archive, and we're not recording. 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not os.path.isfile(page.archive_path): 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.allow_live_sites: 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning(""" 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) The page set archive %s for page %s does not exist, benchmarking against live 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sites! Results won't be repeatable or comparable. 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) To fix this, either add svn-internal to your .gclient using 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) http://goto/read-src-internal, or create a new archive using record_wpr. 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """, os.path.relpath(page.archive_path), page.url) 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning(""" 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) The page set archive %s for page %s does not exist. Not running the page. 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) To fix this, either add svn-internal to your .gclient using 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) http://goto/read-src-internal, or create a new archive using record_wpr. 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) To run against live sites, pass the flag --allow-live-sites. 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) """, os.path.relpath(page.archive_path), page.url) 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, 'Page set archive doesn\'t exist', '') 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pages_without_archives.append(page) 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pages = [page for page in pages if page not in pages_without_archives] 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Verify credentials path. 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) credentials_path = None 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if self.page_set.credentials_path: 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) credentials_path = os.path.join(os.path.dirname(self.page_set.file_path), 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self.page_set.credentials_path) 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not os.path.exists(credentials_path): 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) credentials_path = None 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Set up user agent. 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if self.page_set.user_agent_type: 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.browser_user_agent_type = self.page_set.user_agent_type 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for page in pages: 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test.CustomizeBrowserOptionsForPage(page, possible_browser.options) 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Check tracing directory. 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.trace_dir: 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not os.path.exists(options.trace_dir): 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) os.mkdir(options.trace_dir) 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not os.path.isdir(options.trace_dir): 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise Exception('--trace-dir isn\'t a directory: %s' % 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.trace_dir) 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) elif os.listdir(options.trace_dir): 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise Exception('Trace directory isn\'t empty: %s' % options.trace_dir) 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state = _RunState() 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) last_archive_path = None 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for page in pages: 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.wpr_mode != wpr_modes.WPR_RECORD: 1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if page.archive_path and os.path.isfile(page.archive_path): 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) possible_browser.options.wpr_mode = wpr_modes.WPR_REPLAY 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) possible_browser.options.wpr_mode = wpr_modes.WPR_OFF 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if last_archive_path != page.archive_path: 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.Close() 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state = _RunState() 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) last_archive_path = page.archive_path 1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tries = 3 1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while tries: 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not state.browser: 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._SetupBrowser(state, test, possible_browser, 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) credentials_path, page.archive_path) 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not state.tab: 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if len(state.browser.tabs) == 0: 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.tabs.New() 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.tab = state.browser.tabs[0] 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if state.browser.supports_tab_control: 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while len(state.browser.tabs) > 1: 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.tabs[-1].Close() 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.trace_dir: 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._SetupTracingTab(state) 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._WaitForThermalThrottlingIfNeeded(state.browser.platform) 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._RunPage(options, page, state.tab, test, results) 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._CheckThermalThrottling(state.browser.platform) 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except exceptions.TabCrashException: 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stdout = '' 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not options.show_stdout: 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stdout = state.browser.GetStandardOutput() 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) stdout = (('\nStandard Output:\n') + 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ('*' * 80) + 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) '\n\t' + stdout.replace('\n', '\n\t') + '\n' + 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ('*' * 80)) 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning('Tab crashed: %s%s', page.url, stdout) 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.Close() 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.trace_dir: 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._EndTracing(state, options, page) 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if test.needs_browser_restart_after_each_run: 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.Close() 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except exceptions.BrowserGoneException: 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning('Lost connection to browser. Retrying.') 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.Close() 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tries -= 1 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not tries: 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error('Lost connection to browser 3 times. Failing.') 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) finally: 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.Close() 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _RunPage(self, options, page, tab, test, results): 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not test.CanRunForPage(page): 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning('Skiping test: it cannot run for %s', page.url) 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddSkippedPage(page, 'Test cannot run', '') 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info('Running %s' % page.url) 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page_state = PageState() 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) did_prepare = self._PreparePage(page, tab, page_state, test, results) 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except util.TimeoutException, ex: 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning('Timed out waiting for reply on %s. This is unusual.', 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page.url) 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, ex, traceback.format_exc()) 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except exceptions.TabCrashException, ex: 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, ex, traceback.format_exc()) 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except exceptions.BrowserGoneException: 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except Exception, ex: 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error('Unexpected failure while running %s: %s', 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page.url, traceback.format_exc()) 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._CleanUpPage(page, tab, page_state) 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not did_prepare: 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._CleanUpPage(page, tab, page_state) 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test.Run(options, page, tab, results) 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except page_test.Failure, ex: 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info('%s: %s', ex, page.url) 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, ex, traceback.format_exc()) 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except util.TimeoutException, ex: 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning('Timed out while running %s', page.url) 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, ex, traceback.format_exc()) 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except exceptions.TabCrashException, ex: 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, ex, traceback.format_exc()) 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except exceptions.BrowserGoneException: 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except Exception, ex: 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error('Unexpected failure while running %s: %s', 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page.url, traceback.format_exc()) 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) raise 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) finally: 2562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) self._CleanUpPage(page, tab, page_state) 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddSuccess(page) 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def Close(self): 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pass 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _SetupBrowser(self, state, test, possible_browser, credentials_path, 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) archive_path): 2652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) assert not state.tab 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser = possible_browser.Create() 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.credentials.credentials_path = credentials_path 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test.SetUpBrowser(state.browser) 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if state.first_browser: 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.credentials.WarnIfMissingCredentials(self.page_set) 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.first_browser = False 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.SetReplayArchivePath(archive_path) 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _SetupTracingTab(self, state): 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if state.browser.supports_tracing: 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.is_tracing = True 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.StartTracing() 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _EndTracing(self, state, options, page): 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if state.is_tracing: 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) assert state.browser 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.is_tracing = False 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) state.browser.StopTracing() 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_result = state.browser.GetTraceResultAndReset() 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info('Processing trace...') 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_file_base = os.path.join( 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) options.trace_dir, page.url_as_file_safe_name) 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if options.page_repeat != 1 or options.pageset_repeat != 1: 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_file_index = 0 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while True: 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_file = '%s_%03d.json' % (trace_file_base, trace_file_index) 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not os.path.exists(trace_file): 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_file_index = trace_file_index + 1 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_file = '%s.json' % trace_file_base 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) with codecs.open(trace_file, 'w', 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) encoding='utf-8') as trace_file: 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) trace_result.Serialize(trace_file) 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info('Trace saved.') 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _PreparePage(self, page, tab, page_state, test, results): 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) parsed_url = urlparse.urlparse(page.url) 3092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if parsed_url[0] == 'file': 3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) serving_dirs, filename = page.serving_dirs_and_file 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab.browser.SetHTTPServerDirectories(serving_dirs) 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) target_side_url = tab.browser.http_server.UrlOf(filename) 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else: 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) target_side_url = page.url 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if page.credentials: 3172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page_state.did_login = tab.browser.credentials.LoginNeeded( 3182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab, page.credentials) 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not page_state.did_login: 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) msg = 'Could not login to %s on %s' % (page.credentials, 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) target_side_url) 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.info(msg) 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) results.AddFailure(page, msg, "") 3242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return False 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test.WillNavigateToPage(page, tab) 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab.Navigate(target_side_url) 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test.DidNavigateToPage(page, tab) 3292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) # Wait for unpredictable redirects. 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if page.wait_time_after_navigate: 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) time.sleep(page.wait_time_after_navigate) 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page.WaitToLoad(tab, 60) 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab.WaitForDocumentReadyStateToBeInteractiveOrBetter() 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return True 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _CleanUpPage(self, page, tab, page_state): # pylint: disable=R0201 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if page.credentials and page_state.did_login: 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab.browser.credentials.LoginNoLongerNeeded(tab, page.credentials) 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) try: 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab.EvaluateJavaScript("""window.chrome && chrome.benchmarking && 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) chrome.benchmarking.closeConnections()""") 3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) except Exception: 3452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pass 3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _WaitForThermalThrottlingIfNeeded(self, platform): 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not platform.CanMonitorThermalThrottling(): 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) thermal_throttling_retry = 0 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (platform.IsThermallyThrottled() and 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) thermal_throttling_retry < 3): 3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.warning('Thermally throttled, waiting (%d)...', 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) thermal_throttling_retry) 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) thermal_throttling_retry += 1 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) time.sleep(thermal_throttling_retry * 2) 3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if platform.IsThermallyThrottled(): 3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error('Device is thermally throttled before running ' 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'performance tests, results will vary.') 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def _CheckThermalThrottling(self, platform): 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if not platform.CanMonitorThermalThrottling(): 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if platform.HasBeenThermallyThrottled(): 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) logging.error('Device has been thermally throttled during ' 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 'performance tests, results will vary.') 3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) @staticmethod 3702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) def AddCommandLineOptions(parser): 3712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) page_filter_module.PageFilter.AddCommandLineOptions(parser) 372