# 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)