OLD | NEW |
1 # Copyright (c) 2014 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2014 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import unittest | 5 import unittest |
6 | 6 |
7 from webkitpy.layout_tests.servers import server_base | 7 from webkitpy.layout_tests.servers import server_base |
8 from webkitpy.layout_tests.servers import cli_wrapper | 8 from webkitpy.layout_tests.servers import cli_wrapper |
9 | 9 |
10 | 10 |
11 class MockServer(object): | 11 class MockServer(object): |
| 12 |
12 def __init__(self, *args, **kwargs): | 13 def __init__(self, *args, **kwargs): |
13 self.args = args | 14 self.args = args |
14 self.kwargs = kwargs | 15 self.kwargs = kwargs |
15 self.start_called = False | 16 self.start_called = False |
16 self.stop_called = False | 17 self.stop_called = False |
17 | 18 |
18 def start(self): | 19 def start(self): |
19 self.start_called = True | 20 self.start_called = True |
20 | 21 |
21 def stop(self): | 22 def stop(self): |
22 self.stop_called = True | 23 self.stop_called = True |
23 | 24 |
24 | 25 |
25 class TestCliWrapper(unittest.TestCase): | 26 class TestCliWrapper(unittest.TestCase): |
26 | 27 |
27 def test_main(self): | 28 def test_main(self): |
28 def mock_server_constructor(*args, **kwargs): | 29 def mock_server_constructor(*args, **kwargs): |
29 self.server = MockServer(args, kwargs) | 30 self.server = MockServer(args, kwargs) |
30 return self.server | 31 return self.server |
31 | 32 |
32 cli_wrapper.main(mock_server_constructor, input_fn=lambda msg: True, arg
v=[]) | 33 cli_wrapper.main(mock_server_constructor, input_fn=lambda msg: True, arg
v=[]) |
33 self.assertTrue(self.server.start_called) | 34 self.assertTrue(self.server.start_called) |
34 self.assertTrue(self.server.stop_called) | 35 self.assertTrue(self.server.stop_called) |
OLD | NEW |