blob: 59e662abbcca4710b4b8df50738973766c6a9cb6 [file] [log] [blame]
# Copyright 2020 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.
"""Implementation of skia_gold_session.py without output managers.
Diff output is instead stored in a directory and pointed to with file:// URLs.
"""
import os
import subprocess
import tempfile
from skia_gold_common import skia_gold_session
class OutputManagerlessSkiaGoldSession(skia_gold_session.SkiaGoldSession):
def RunComparison( # pylint: disable=too-many-arguments
self,
name,
png_file,
output_manager=True,
inexact_matching_args=None,
use_luci=True,
optional_keys=None):
# Passing True for the output manager is a bit of a hack, as we don't
# actually need an output manager and just need to get past the truthy
# check.
return super(OutputManagerlessSkiaGoldSession, self).RunComparison(
name=name,
png_file=png_file,
output_manager=output_manager,
inexact_matching_args=inexact_matching_args,
use_luci=use_luci,
optional_keys=optional_keys)
def _CreateDiffOutputDir(self):
# We intentionally don't clean this up and don't put it in self._working_dir
# since we need it to stick around after the test completes so the user
# can look at its contents.
return tempfile.mkdtemp()
def _StoreDiffLinks(self, image_name, _, output_dir):
results = self._comparison_results.setdefault(image_name,
self.ComparisonResults())
# The directory should contain "input-<hash>.png", "closest-<hash>.png",
# and "diff.png".
for f in os.listdir(output_dir):
file_url = 'file://%s' % os.path.join(output_dir, f)
if f.startswith('input-'):
results.local_diff_given_image = file_url
elif f.startswith('closest-'):
results.local_diff_closest_image = file_url
elif f == 'diff.png':
results.local_diff_diff_image = file_url
@staticmethod
def _RunCmdForRcAndOutput(cmd):
try:
output = subprocess.check_output(cmd, stderr=subprocess.STDOUT)
return 0, output
except subprocess.CalledProcessError as e:
return e.returncode, e.output