Tools/Scripts/webkitpy/tool/commands/rebaselineserver.py

   1 # Copyright (c) 2010 Google Inc. All rights reserved.
   2 #
   3 # Redistribution and use in source and binary forms, with or without
   4 # modification, are permitted provided that the following conditions are
   5 # met:
   6 #
   7 #     * Redistributions of source code must retain the above copyright
   8 # notice, this list of conditions and the following disclaimer.
   9 #     * Redistributions in binary form must reproduce the above
  10 # copyright notice, this list of conditions and the following disclaimer
  11 # in the documentation and/or other materials provided with the
  12 # distribution.
  13 #     * Neither the name of Google Inc. nor the names of its
  14 # contributors may be used to endorse or promote products derived from
  15 # this software without specific prior written permission.
  16 #
  17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28
  29 """Starts a local HTTP server which displays layout test failures (given a test
  30 results directory), provides comparisons of expected and actual results (both
  31 images and text) and allows one-click rebaselining of tests."""
  32
  33 import os
  34 import os.path
  35
  36 from webkitpy.common import system
  37 from webkitpy.common.net.resultsjsonparser import for_each_test, JSONTestResult
  38 from webkitpy.layout_tests.layout_package import json_results_generator
  39 from webkitpy.tool.commands.abstractlocalservercommand import AbstractLocalServerCommand
  40 from webkitpy.tool.servers.rebaselineserver import get_test_baselines, RebaselineHTTPServer, STATE_NEEDS_REBASELINE
  41
  42
  43 class TestConfig(object):
  44     def __init__(self, test_port, layout_tests_directory, results_directory, platforms, filesystem, scm):
  45         self.test_port = test_port
  46         self.layout_tests_directory = layout_tests_directory
  47         self.results_directory = results_directory
  48         self.platforms = platforms
  49         self.filesystem = filesystem
  50         self.scm = scm
  51
  52
  53 class RebaselineServer(AbstractLocalServerCommand):
  54     name = "rebaseline-server"
  55     help_text = __doc__
  56     argument_names = "/path/to/results/directory"
  57
  58     server = RebaselineHTTPServer
  59
  60     def _gather_baselines(self, results_json):
  61         # Rebaseline server and it's associated JavaScript expected the tests subtree to
  62         # be key-value pairs instead of hierarchical.
  63         # FIXME: make the rebaseline server use the hierarchical tree.
  64         new_tests_subtree = {}
  65
  66         def gather_baselines_for_test(test_name, result_dict):
  67             result = JSONTestResult(test_name, result_dict)
  68             if result.did_pass_or_run_as_expected():
  69                 return
  70             result_dict['state'] = STATE_NEEDS_REBASELINE
  71             result_dict['baselines'] = get_test_baselines(test_name, self._test_config)
  72             new_tests_subtree[test_name] = result_dict
  73
  74         for_each_test(results_json['tests'], gather_baselines_for_test)
  75         results_json['tests'] = new_tests_subtree
  76
  77     def _prepare_config(self, options, args, tool):
  78         results_directory = args[0]
  79         filesystem = system.filesystem.FileSystem()
  80         scm = self._tool.scm()
  81
  82         print 'Parsing full_results.json...'
  83         results_json_path = filesystem.join(results_directory, 'full_results.json')
  84         results_json = json_results_generator.load_json(filesystem, results_json_path)
  85
  86         port = tool.port_factory.get()
  87         layout_tests_directory = port.layout_tests_dir()
  88         platforms = filesystem.listdir(filesystem.join(layout_tests_directory, 'platform'))
  89         self._test_config = TestConfig(port, layout_tests_directory, results_directory, platforms, filesystem, scm)
  90
  91         print 'Gathering current baselines...'
  92         self._gather_baselines(results_json)
  93
  94         return {
  95             'test_config': self._test_config,
  96             "results_json": results_json,
  97             "platforms_json": {
  98                 'platforms': platforms,
  99                 'defaultPlatform': port.name(),
 100             },
 101         }