blob: d234ebdc39948c84b349ea17853f97de0a423027 [file] [log] [blame]
Torne (Richard Coles)5c87bf82012-11-14 11:46:17 +00001# Copyright (C) 2012 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
30class MockServerProcess(object):
31 def __init__(self, port_obj=None, name=None, cmd=None, env=None, universal_newlines=False, lines=None, crashed=False):
32 self.timed_out = False
33 self.lines = lines or []
34 self.crashed = crashed
35 self.writes = []
36 self.cmd = cmd
37 self.env = env
38 self.started = False
39 self.stopped = False
40
41 def write(self, bytes):
42 self.writes.append(bytes)
43
44 def has_crashed(self):
45 return self.crashed
46
47 def read_stdout_line(self, deadline):
48 return self.lines.pop(0) + "\n"
49
50 def read_stdout(self, deadline, size):
51 first_line = self.lines[0]
52 if size > len(first_line):
53 self.lines.pop(0)
54 remaining_size = size - len(first_line) - 1
55 if not remaining_size:
56 return first_line + "\n"
57 return first_line + "\n" + self.read_stdout(deadline, remaining_size)
58 result = self.lines[0][:size]
59 self.lines[0] = self.lines[0][size:]
60 return result
61
62 def pop_all_buffered_stderr(self):
63 return ''
64
65 def read_either_stdout_or_stderr_line(self, deadline):
66 # FIXME: We should have tests which intermix stderr and stdout lines.
67 return self.read_stdout_line(deadline), None
68
69 def start(self):
70 self.started = True
71
72 def stop(self, kill_directly=False):
73 self.stopped = True
74 return
75
76 def kill(self):
77 return