| OLD | NEW |
| (Empty) |
| 1 # Copyright (C) 2014 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 Google name 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 from webkitpy.layout_tests.port import driver | |
| 30 import time | |
| 31 import shutil | |
| 32 | |
| 33 | |
| 34 class BrowserTestDriver(driver.Driver): | |
| 35 """Object for running print preview test(s) using browser_tests.""" | |
| 36 def __init__(self, port, worker_number, pixel_tests, no_timeout=False): | |
| 37 """Invokes the constructor of driver.Driver.""" | |
| 38 super(BrowserTestDriver, self).__init__(port, worker_number, pixel_tests
, no_timeout) | |
| 39 | |
| 40 def start(self, pixel_tests, per_test_args, deadline): | |
| 41 """Same as Driver.start() however, it has an extra step. It waits for | |
| 42 a path to a file to be used for stdin to be printed by the browser test. | |
| 43 If a path is found by the deadline test test will open the file and | |
| 44 assign it to the stdin of the process that is owned by this driver's | |
| 45 server process. | |
| 46 """ | |
| 47 # FIXME(ivandavid): Need to handle case where the layout test doesn't | |
| 48 # get a file name. | |
| 49 new_cmd_line = self.cmd_line(pixel_tests, per_test_args) | |
| 50 if not self._server_process or new_cmd_line != self._current_cmd_line: | |
| 51 self._start(pixel_tests, per_test_args) | |
| 52 self._run_post_start_tasks() | |
| 53 self._open_stdin_path(deadline) | |
| 54 | |
| 55 # Gets the path of the directory that the file for stdin communication is | |
| 56 # in. Since the browser test cannot clean it up, the layout test framework | |
| 57 # will. Everything the browser test uses is stored in the same directory as | |
| 58 # the stdin file, so deleting that directory recursively will remove all the | |
| 59 # other temp data, like the printed pdf. This function assumes the correct | |
| 60 # file path is sent. It won't delete files with only one component to avoid | |
| 61 # accidentally deleting files like /tmp. | |
| 62 def _open_stdin_path(self, deadline, test=False): | |
| 63 # FIXME(ivandavid): Come up with a way to test & see what happens when | |
| 64 # the file can't be opened. | |
| 65 path, found = self._read_stdin_path(deadline) | |
| 66 if found: | |
| 67 if test == False: | |
| 68 self._server_process._proc.stdin = open(path, 'wb', 0) | |
| 69 | |
| 70 def _read_stdin_path(self, deadline): | |
| 71 # return (stdin_path, bool) | |
| 72 block = self._read_block(deadline) | |
| 73 if block.stdin_path: | |
| 74 return (block.stdin_path, True) | |
| 75 return (None, False) | |
| 76 | |
| 77 def cmd_line(self, pixel_tests, per_test_args): | |
| 78 """Command line arguments to run the browser test.""" | |
| 79 cmd = self._command_wrapper(self._port.get_option('wrapper')) | |
| 80 cmd.append(self._port._path_to_driver()) | |
| 81 cmd.append('--gtest_filter=PrintPreviewPdfGeneratedBrowserTest.MANUAL_La
youtTestDriver') | |
| 82 cmd.append('--run-manual') | |
| 83 cmd.append('--single_process') | |
| 84 cmd.extend(per_test_args) | |
| 85 cmd.extend(self._port.get_option('additional_drt_flag', [])) | |
| 86 return cmd | |
| 87 | |
| 88 def stop(self): | |
| 89 if self._server_process: | |
| 90 self._server_process.write('QUIT') | |
| 91 super(BrowserTestDriver, self).stop(self._port.driver_stop_timeout()) | |
| OLD | NEW |