| Index: Tools/Scripts/webkitpy/layout_tests/servers/cli_wrapper_unittest.py
|
| diff --git a/Tools/Scripts/webkitpy/layout_tests/servers/cli_wrapper_unittest.py b/Tools/Scripts/webkitpy/layout_tests/servers/cli_wrapper_unittest.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..54647394d6b0689526e37f560fd62d3b5080668e
|
| --- /dev/null
|
| +++ b/Tools/Scripts/webkitpy/layout_tests/servers/cli_wrapper_unittest.py
|
| @@ -0,0 +1,34 @@
|
| +# Copyright (c) 2014 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import unittest
|
| +
|
| +from webkitpy.layout_tests.servers import server_base
|
| +from webkitpy.layout_tests.servers import cli_wrapper
|
| +
|
| +
|
| +class MockServer(object):
|
| + def __init__(self, *args, **kwargs):
|
| + self.args = args
|
| + self.kwargs = kwargs
|
| + self.start_called = False
|
| + self.stop_called = False
|
| +
|
| + def start(self):
|
| + self.start_called = True
|
| +
|
| + def stop(self):
|
| + self.stop_called = True
|
| +
|
| +
|
| +class TestCliWrapper(unittest.TestCase):
|
| +
|
| + def test_main(self):
|
| + def mock_server_constructor(*args, **kwargs):
|
| + self.server = MockServer(args, kwargs)
|
| + return self.server
|
| +
|
| + cli_wrapper.main(mock_server_constructor, input_fn=lambda msg: True, argv=[])
|
| + self.assertTrue(self.server.start_called)
|
| + self.assertTrue(self.server.stop_called)
|
|
|