color: Run GPU pixel tests with ColorCorrectRendering feature

content/test/gpu/gpu_tests/screenshot_sync_integration_test.py

 # Copyright 2017 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import os
 import random
 import sys
 
+from gpu_tests import color_profile_manager
 from gpu_tests import gpu_integration_test
 from gpu_tests import path_util
 from gpu_tests import screenshot_sync_expectations
 
 from telemetry.util import image_util
 from telemetry.util import rgba_color
 
 data_path = os.path.join(
     path_util.GetChromiumSrcDir(), 'content', 'test', 'data', 'gpu')
 
 class ScreenshotSyncIntegrationTest(gpu_integration_test.GpuIntegrationTest):
   """Tests that screenshots are properly synchronized with the frame on
   which they were requested.
   """
 
   @classmethod
   def Name(cls):
     """The name by which this test is invoked on the command line."""
     return 'screenshot_sync'
 
   @classmethod
   def SetUpProcess(cls):
+    color_profile_manager.ForceUntilExitSRGB()
     super(cls, ScreenshotSyncIntegrationTest).SetUpProcess()
     cls.CustomizeBrowserArgs(cls._AddDefaultArgs([]))
     cls.StartBrowser()
     cls.SetStaticServerDirs([data_path])
 
   @staticmethod
   def _AddDefaultArgs(browser_args):
     # --test-type=gpu is used to suppress the "Google API Keys are
     # missing" infobar, which causes flakiness in tests.
-    return ['--test-type=gpu'] + browser_args
+    return [
+      '--force-color-profile=srgb',
+      '--enable-features=ColorCorrectRendering',
+      '--test-type=gpu'] + browser_args
 
   @classmethod
   def _CreateExpectations(cls):
     return screenshot_sync_expectations.ScreenshotSyncExpectations()
 
   @classmethod
   def GenerateGpuTests(cls, options):
     yield('ScreenshotSync_SWRasterWithCanvas',
           'screenshot_sync_canvas.html',
           ('--disable-gpu-rasterization'))
@@ skipping 10 matching lines @@
   def _Navigate(self, test_path):
     url = self.UrlOfStaticFilePath(test_path)
     # It's crucial to use the action_runner, rather than the tab's
     # Navigate method directly. It waits for the document ready state
     # to become interactive or better, avoiding critical race
     # conditions.
     self.tab.action_runner.Navigate(url)
 
   def _CheckColorMatchAtLocation(self, expectedRGB, screenshot, x, y):
     pixel_value = image_util.GetPixelColor(screenshot, x, y)
-    if not expectedRGB.IsEqual(pixel_value):
+    # Allow for off-by-one errors due to color conversion.
+    tolerance = 1
+    if not expectedRGB.IsEqual(pixel_value, tolerance):
       error_message = ('Color mismatch at (%d, %d): expected (%d, %d, %d), ' +
                        'got (%d, %d, %d)') % (
                          x, y, expectedRGB.r, expectedRGB.g, expectedRGB.b,
                          pixel_value.r, pixel_value.g, pixel_value.b)
       self.fail(error_message)
 
   def _CheckScreenshot(self):
     canvasRGB = rgba_color.RgbaColor(random.randint(0, 255),
                                      random.randint(0, 255),
                                      random.randint(0, 255),
@@ skipping 15 matching lines @@
     browser_arg = args[0]
     self.RestartBrowserIfNecessaryWithArgs(self._AddDefaultArgs([browser_arg]))
     self._Navigate(test_path)
     repetitions = 20
     for _ in range(0, repetitions):
       self._CheckScreenshot()
 
 def load_tests(loader, tests, pattern):
   del loader, tests, pattern  # Unused.
   return gpu_integration_test.LoadAllTestsInModule(sys.modules[__name__])