| # Copyright (c) 2011, the Dart project authors. Please see the AUTHORS file |
| # for details. All rights reserved. Use of this source code is governed by a |
| # BSD-style license that can be found in the LICENSE file. |
| # |
| # For now we have to use this trampoline to turn --dart-flags command line |
| # switch into env variable DART_FLAGS. Eventually, DumpRenderTree should |
| # support --dart-flags and this hack may go away. |
| # |
| # Expected invocation: python drt-trampoline.py <path to DRT> <DRT command line> |
| |
| import optparse |
| import os |
| import signal |
| import subprocess |
| import sys |
| |
| def parse_options(argv): |
| parser = optparse.OptionParser() |
| parser.add_option('--dart-flags', |
| metavar='FLAGS', |
| dest='dart_flags') |
| parser.add_option('--out-expectation', |
| metavar='FILE', |
| dest='out_expected_file') |
| parser.add_option('--package-root', |
| metavar='DIRECTORY', |
| dest='dart_package_root') |
| parser.add_option('--no-timeout', |
| action='store_true') |
| return parser.parse_args(args=argv) |
| |
| |
| def main(argv): |
| drt_path = argv[1] |
| (options, arguments) = parse_options(argv[2:]) |
| |
| cmd = [drt_path] |
| |
| env = None |
| test_file = None |
| dart_flags = options.dart_flags |
| out_expected_file = options.out_expected_file |
| dart_package_root = options.dart_package_root |
| is_png = False |
| |
| if dart_flags: |
| if not env: |
| env = dict(os.environ.items()) |
| env['DART_FLAGS'] = dart_flags |
| |
| if dart_package_root: |
| if not env: |
| env = dict(os.environ.items()) |
| absolute_path = os.path.abspath(dart_package_root) |
| absolute_path = absolute_path.replace(os.path.sep, '/') |
| if not absolute_path.startswith('/'): |
| # Happens on Windows for C:\packages |
| absolute_path = '/%s' % absolute_path |
| env['DART_PACKAGE_ROOT'] = 'file://%s' % absolute_path |
| |
| if out_expected_file: |
| if out_expected_file.endswith('.png'): |
| cmd.append('--notree') |
| is_png = True |
| elif not out_expected_file.endswith('.txt'): |
| raise Exception( |
| 'Bad file expectation (%s) please specify either a .txt or a .png file' |
| % out_expected_file) |
| |
| if options.no_timeout: |
| cmd.append('--no-timeout') |
| |
| for arg in arguments: |
| if '.html' in arg: |
| test_file = arg |
| else: |
| cmd.append(arg) |
| |
| if is_png: |
| # pixel tests are specified by running DRT "foo.html'-p" |
| cmd.append(test_file + "'-p") |
| else: |
| cmd.append(test_file) |
| |
| stdout = subprocess.PIPE if out_expected_file else None |
| p = subprocess.Popen(cmd, env=env, stdout=stdout) |
| |
| def signal_handler(signal, frame): |
| p.terminate() |
| sys.exit(0) |
| |
| # SIGINT is Ctrl-C. |
| signal.signal(signal.SIGINT, signal_handler) |
| # SIGTERM is sent by test.dart when a process times out. |
| signal.signal(signal.SIGTERM, signal_handler) |
| output, error = p.communicate() |
| signal.signal(signal.SIGINT, signal.SIG_DFL) |
| signal.signal(signal.SIGTERM, signal.SIG_DFL) |
| |
| if p.returncode != 0: |
| raise Exception('Failed to run command. return code=%s' % p.returncode) |
| |
| if out_expected_file: |
| # Compare output to the given expectation file. |
| expectation = None |
| if is_png: |
| # DRT prints the image to STDOUT, but includes extra text that we trim: |
| # - several header lines until a line saying 'Content-Length:' |
| # - a '#EOF\n' at the end |
| last_header_line = output.find('Content-Length:') |
| start_pos = output.find('\n', last_header_line) + 1 |
| output = output[start_pos : -len('#EOF\n')] |
| if os.path.exists(out_expected_file): |
| with open(out_expected_file, 'r') as f: |
| expectation = f.read() |
| else: |
| # Instructions on how to create the expectation will be printed below |
| # (outout != expectation) |
| print 'File %s was not found' % out_expected_file |
| expectation = None |
| |
| # Report test status using the format test.dart expects to see from DRT. |
| print 'Content-Type: text/plain' |
| if expectation == output: |
| print 'PASS' |
| print 'Expectation matches' |
| else: |
| # Generate a temporary file in the same place as the .html file: |
| out_file = test_file[:test_file.rfind('.html')] + out_expected_file[-4:] |
| with open(out_file, 'w') as f: |
| f.write(output) |
| print 'FAIL' |
| print 'Expectation didn\'t match.\n' |
| if len(output) == 0: |
| print ('\033[31mERROR\033[0m: DumpRenderTree generated an empty pixel ' |
| 'output! This is commonly an error in executing DumpRenderTree, and' |
| ' not that expectations are out of date.\n') |
| print 'You can update expectations by running:\n' |
| print 'cp %s %s\n' % (out_file, out_expected_file) |
| print '#EOF' |
| |
| if __name__ == '__main__': |
| try: |
| sys.exit(main(sys.argv)) |
| except StandardError as e: |
| print 'Fail: ' + str(e) |
| sys.exit(1) |