# Copyright (C) 2011 Google Inc. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions are # met: # # * Redistributions of source code must retain the above copyright # notice, this list of conditions and the following disclaimer. # * Redistributions in binary form must reproduce the above # copyright notice, this list of conditions and the following disclaimer # in the documentation and/or other materials provided with the # distribution. # * Neither the name of Google Inc. nor the names of its # contributors may be used to endorse or promote products derived from # this software without specific prior written permission. # # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. """Handle messages from the Manager and executes actual tests.""" import logging import sys import threading import time from webkitpy.layout_tests.controllers import manager_worker_broker from webkitpy.layout_tests.controllers import single_test_runner from webkitpy.layout_tests.models import test_expectations from webkitpy.layout_tests.models import test_results _log = logging.getLogger(__name__) class Worker(manager_worker_broker.AbstractWorker): def __init__(self, worker_connection, worker_number, options): manager_worker_broker.AbstractWorker.__init__(self, worker_connection, worker_number, options) self._done = False self._canceled = False self._port = None self._batch_size = None self._batch_count = None self._filesystem = None self._driver = None self._tests_run_file = None self._tests_run_filename = None def __del__(self): self.cleanup() def safe_init(self, port): """This method should only be called when it is is safe for the mixin to create state that can't be Pickled. This routine exists so that the mixin can be created and then marshaled across into a child process.""" self._port = port self._filesystem = port.filesystem self._batch_count = 0 self._batch_size = self._options.batch_size or 0 tests_run_filename = self._filesystem.join(port.results_directory(), "tests_run%d.txt" % self._worker_number) self._tests_run_file = self._filesystem.open_text_file_for_writing(tests_run_filename) def cancel(self): """Attempt to abort processing (best effort).""" self._canceled = True def is_done(self): return self._done or self._canceled def name(self): return self._name def run(self, port): self.safe_init(port) exception_msg = "" _log.debug("%s starting" % self._name) try: self._worker_connection.run_message_loop() if not self.is_done(): raise AssertionError("%s: ran out of messages in worker queue." % self._name) except KeyboardInterrupt: exception_msg = ", interrupted" self._worker_connection.raise_exception(sys.exc_info()) except: exception_msg = ", exception raised" self._worker_connection.raise_exception(sys.exc_info()) finally: _log.debug("%s done with message loop%s" % (self._name, exception_msg)) self._worker_connection.post_message('done') self.cleanup() _log.debug("%s exiting" % self._name) def handle_test_list(self, src, list_name, test_list): start_time = time.time() num_tests = 0 for test_input in test_list: self._run_test(test_input) num_tests += 1 self._worker_connection.yield_to_broker() elapsed_time = time.time() - start_time self._worker_connection.post_message('finished_list', list_name, num_tests, elapsed_time) def handle_stop(self, src): self._done = True def _run_test(self, test_input): test_timeout_sec = self.timeout(test_input) start = time.time() self._worker_connection.post_message('started_test', test_input, test_timeout_sec) result = self.run_test_with_timeout(test_input, test_timeout_sec) elapsed_time = time.time() - start self._worker_connection.post_message('finished_test', result, elapsed_time) self.clean_up_after_test(test_input, result) def cleanup(self): _log.debug("%s cleaning up" % self._name) self.kill_driver() if self._tests_run_file: self._tests_run_file.close() self._tests_run_file = None def timeout(self, test_input): """Compute the appropriate timeout value for a test.""" # The DumpRenderTree watchdog uses 2.5x the timeout; we want to be # larger than that. We also add a little more padding if we're # running tests in a separate thread. # # Note that we need to convert the test timeout from a # string value in milliseconds to a float for Python. driver_timeout_sec = 3.0 * float(test_input.timeout) / 1000.0 if not self._options.run_singly: return driver_timeout_sec thread_padding_sec = 1.0 thread_timeout_sec = driver_timeout_sec + thread_padding_sec return thread_timeout_sec def kill_driver(self): if self._driver: _log.debug("%s killing driver" % self._name) self._driver.stop() self._driver = None def run_test_with_timeout(self, test_input, timeout): if self._options.run_singly: return self._run_test_in_another_thread(test_input, timeout) return self._run_test_in_this_thread(test_input) def clean_up_after_test(self, test_input, result): self._batch_count += 1 test_name = test_input.test_name self._tests_run_file.write(test_name + "\n") if result.failures: # Check and kill DumpRenderTree if we need to. if any([f.driver_needs_restart() for f in result.failures]): self.kill_driver() # Reset the batch count since the shell just bounced. self._batch_count = 0 # Print the error message(s). _log.debug("%s %s failed:" % (self._name, test_name)) for f in result.failures: _log.debug("%s %s" % (self._name, f.message())) elif result.type == test_expectations.SKIP: _log.debug("%s %s skipped" % (self._name, test_name)) else: _log.debug("%s %s passed" % (self._name, test_name)) if self._batch_size > 0 and self._batch_count >= self._batch_size: self.kill_driver() self._batch_count = 0 def _run_test_in_another_thread(self, test_input, thread_timeout_sec): """Run a test in a separate thread, enforcing a hard time limit. Since we can only detect the termination of a thread, not any internal state or progress, we can only run per-test timeouts when running test files singly. Args: test_input: Object containing the test filename and timeout thread_timeout_sec: time to wait before killing the driver process. Returns: A TestResult """ worker = self driver = self._port.create_driver(self._worker_number) driver.start() class SingleTestThread(threading.Thread): def __init__(self): threading.Thread.__init__(self) self.result = None def run(self): self.result = worker.run_single_test(driver, test_input) thread = SingleTestThread() thread.start() thread.join(thread_timeout_sec) result = thread.result if thread.isAlive(): # If join() returned with the thread still running, the # DumpRenderTree is completely hung and there's nothing # more we can do with it. We have to kill all the # DumpRenderTrees to free it up. If we're running more than # one DumpRenderTree thread, we'll end up killing the other # DumpRenderTrees too, introducing spurious crashes. We accept # that tradeoff in order to avoid losing the rest of this # thread's results. _log.error('Test thread hung: killing all DumpRenderTrees') driver.stop() if not result: result = test_results.TestResult(test_input.test_name, failures=[], test_run_time=0) return result def _run_test_in_this_thread(self, test_input): """Run a single test file using a shared DumpRenderTree process. Args: test_input: Object containing the test filename, uri and timeout Returns: a TestResult object. """ if not self._driver or self._driver.poll() is not None: self._driver = self._port.create_driver(self._worker_number) self._driver.start() return self.run_single_test(self._driver, test_input) def run_single_test(self, driver, test_input): return single_test_runner.run_single_test(self._port, self._options, test_input, driver, self._name)