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 |