Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(226)

Side by Side Diff: Tools/Scripts/webkitpy/layout_tests/servers/cli_wrapper_unittest.py

Issue 546133003: Reformat webkitpy.layout_tests w/ format-webkitpy. (Closed) Base URL: svn://svn.chromium.org/blink/trunk
Patch Set: Created 6 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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)
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698