| # 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 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 CliWrapperTest(unittest.TestCase): |
| |
| def setUp(self): |
| self.server = None |
| |
| 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) |