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 from datetime import datetime | 4 from datetime import datetime |
5 import glob | 5 import glob |
6 import optparse | 6 import optparse |
7 import os | 7 import os |
8 import re | 8 import re |
9 | 9 |
10 from telemetry import test | 10 from telemetry import test |
11 from telemetry.core.backends import png_bitmap | 11 from telemetry.core import bitmap |
12 from telemetry.page import page_test | 12 from telemetry.page import page_test |
13 | 13 |
14 test_data_dir = os.path.abspath(os.path.join( | 14 test_data_dir = os.path.abspath(os.path.join( |
15 os.path.dirname(__file__), '..', '..', 'data', 'gpu')) | 15 os.path.dirname(__file__), '..', '..', 'data', 'gpu')) |
16 | 16 |
17 default_generated_data_dir = os.path.join(test_data_dir, 'generated') | 17 default_generated_data_dir = os.path.join(test_data_dir, 'generated') |
18 default_reference_image_dir = os.path.join(test_data_dir, 'gpu_reference') | 18 default_reference_image_dir = os.path.join(test_data_dir, 'gpu_reference') |
19 | 19 |
20 test_harness_script = r""" | 20 test_harness_script = r""" |
21 var domAutomationController = {}; | 21 var domAutomationController = {}; |
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
103 if not cur_revision: | 103 if not cur_revision: |
104 cur_revision = 0 | 104 cur_revision = 0 |
105 | 105 |
106 image_path = os.path.join(img_dir, img_name) | 106 image_path = os.path.join(img_dir, img_name) |
107 | 107 |
108 PixelValidator.DeleteOldReferenceImages(image_path, cur_revision) | 108 PixelValidator.DeleteOldReferenceImages(image_path, cur_revision) |
109 | 109 |
110 image_path = image_path + '_' + str(cur_revision) + '.png' | 110 image_path = image_path + '_' + str(cur_revision) + '.png' |
111 | 111 |
112 try: | 112 try: |
113 ref_png = png_bitmap.PngBitmap.FromFile(image_path) | 113 ref_png = bitmap.Bitmap.FromPngFile(image_path) |
114 except IOError: | 114 except IOError: |
115 ref_png = None | 115 ref_png = None |
116 | 116 |
117 if ref_png: | 117 if ref_png: |
118 return ref_png | 118 return ref_png |
119 | 119 |
120 print 'Reference image not found. Writing tab contents as reference.' | 120 print 'Reference image not found. Writing tab contents as reference.' |
121 | 121 |
122 PixelValidator.WriteImage(image_path, screenshot) | 122 PixelValidator.WriteImage(image_path, screenshot) |
123 return screenshot | 123 return screenshot |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
164 group.add_option('--build-revision', | 164 group.add_option('--build-revision', |
165 help='Chrome revision being tested.', | 165 help='Chrome revision being tested.', |
166 default="unknownrev") | 166 default="unknownrev") |
167 parser.add_option_group(group) | 167 parser.add_option_group(group) |
168 | 168 |
169 def CreatePageSet(self, options): | 169 def CreatePageSet(self, options): |
170 page_set = super(Pixel, self).CreatePageSet(options) | 170 page_set = super(Pixel, self).CreatePageSet(options) |
171 for page in page_set.pages: | 171 for page in page_set.pages: |
172 page.script_to_evaluate_on_commit = test_harness_script | 172 page.script_to_evaluate_on_commit = test_harness_script |
173 return page_set | 173 return page_set |
OLD | NEW |