| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 import os | 4 import os |
| 5 import sys | 5 import sys |
| 6 | 6 |
| 7 from telemetry.page import block_page_measurement_results | 7 from telemetry.page import block_page_measurement_results |
| 8 from telemetry.page import buildbot_page_measurement_results | 8 from telemetry.page import buildbot_page_measurement_results |
| 9 from telemetry.page import csv_page_measurement_results | 9 from telemetry.page import csv_page_measurement_results |
| 10 from telemetry.page import page_measurement_results | 10 from telemetry.page import page_measurement_results |
| (...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 63 help='Redirects output to a file. Defaults to stdout.') | 63 help='Redirects output to a file. Defaults to stdout.') |
| 64 parser.add_option('--output-trace-tag', | 64 parser.add_option('--output-trace-tag', |
| 65 default='', | 65 default='', |
| 66 help='Append a tag to the key of each result trace.') | 66 help='Append a tag to the key of each result trace.') |
| 67 | 67 |
| 68 @property | 68 @property |
| 69 def output_format_choices(self): | 69 def output_format_choices(self): |
| 70 return ['buildbot', 'block', 'csv', 'none'] | 70 return ['buildbot', 'block', 'csv', 'none'] |
| 71 | 71 |
| 72 def PrepareResults(self, options): | 72 def PrepareResults(self, options): |
| 73 if not options.output_file or options.output_file == '-': | 73 if hasattr(options, 'output_file') and options.output_file: |
| 74 output_stream = open(os.path.expanduser(options.output_file), 'w') |
| 75 else: |
| 74 output_stream = sys.stdout | 76 output_stream = sys.stdout |
| 75 else: | 77 if not hasattr(options, 'output_format'): |
| 76 output_stream = open(os.path.expanduser(options.output_file), 'w') | 78 options.output_format = self.output_format_choices[0] |
| 79 if not hasattr(options, 'output_trace_tag'): |
| 80 options.output_trace_tag = '' |
| 77 | 81 |
| 78 if options.output_format == 'csv': | 82 if options.output_format == 'csv': |
| 79 return csv_page_measurement_results.CsvPageMeasurementResults( | 83 return csv_page_measurement_results.CsvPageMeasurementResults( |
| 80 output_stream, | 84 output_stream, |
| 81 self.results_are_the_same_on_every_page, | 85 self.results_are_the_same_on_every_page) |
| 82 trace_tag=options.output_trace_tag) | |
| 83 elif options.output_format == 'block': | 86 elif options.output_format == 'block': |
| 84 return block_page_measurement_results.BlockPageMeasurementResults( | 87 return block_page_measurement_results.BlockPageMeasurementResults( |
| 85 output_stream) | 88 output_stream) |
| 86 elif options.output_format == 'buildbot': | 89 elif options.output_format == 'buildbot': |
| 87 return buildbot_page_measurement_results.BuildbotPageMeasurementResults( | 90 return buildbot_page_measurement_results.BuildbotPageMeasurementResults( |
| 88 trace_tag=options.output_trace_tag) | 91 trace_tag=options.output_trace_tag) |
| 89 elif options.output_format == 'none': | 92 elif options.output_format == 'none': |
| 90 return page_measurement_results.PageMeasurementResults( | 93 return page_measurement_results.PageMeasurementResults( |
| 91 trace_tag=options.output_trace_tag) | 94 trace_tag=options.output_trace_tag) |
| 92 else: | 95 else: |
| (...skipping 27 matching lines...) Expand all Loading... |
| 120 | 123 |
| 121 Put together: | 124 Put together: |
| 122 | 125 |
| 123 def MeasurePage(self, page, tab, results): | 126 def MeasurePage(self, page, tab, results): |
| 124 res = tab.EvaluateJavaScript('2+2') | 127 res = tab.EvaluateJavaScript('2+2') |
| 125 if res != 4: | 128 if res != 4: |
| 126 raise Exception('Oh, wow.') | 129 raise Exception('Oh, wow.') |
| 127 results.Add('two_plus_two', 'count', res) | 130 results.Add('two_plus_two', 'count', res) |
| 128 """ | 131 """ |
| 129 raise NotImplementedError() | 132 raise NotImplementedError() |
| OLD | NEW |