mirror of
https://github.com/flutter/flutter.git
synced 2026-02-20 02:29:02 +08:00
35 lines
981 B
Python
35 lines
981 B
Python
# 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)
|