Home | History | Annotate | Download | only in port 
      1 # Copyright (C) 2011 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 import unittest
     30 
     31 from webkitpy.layout_tests.port import server_process
     32 
     33 
     34 class MockFile(object):
     35     def __init__(self, server_process):
     36         self._server_process = server_process
     37 
     38     def fileno(self):
     39         return 1
     40 
     41     def write(self, line):
     42         self._server_process.broken_pipes.append(self)
     43         raise IOError
     44 
     45     def close(self):
     46         pass
     47 
     48 
     49 class MockProc(object):
     50     def __init__(self, server_process):
     51         self.stdin = MockFile(server_process)
     52         self.stdout = MockFile(server_process)
     53         self.stderr = MockFile(server_process)
     54         self.pid = 1
     55 
     56     def poll(self):
     57         return 1
     58 
     59 
     60 class FakeServerProcess(server_process.ServerProcess):
     61     def _start(self):
     62         self._proc = MockProc(self)
     63         self.stdin = self._proc.stdin
     64         self.broken_pipes = []
     65 
     66 
     67 class TestServerProcess(unittest.TestCase):
     68     def test_broken_pipe(self):
     69         server_process = FakeServerProcess(port_obj=None, name="test", cmd=["test"])
     70         server_process.write("should break")
     71         self.assertTrue(server_process.crashed)
     72         self.assertEquals(server_process._proc, None)
     73         self.assertEquals(server_process.broken_pipes, [server_process.stdin])
     74 
     75 
     76 if __name__ == '__main__':
     77     unittest.main()
     78