| 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 |