Clean up run-blink-httpd, run-blink-websocketserver, add tests.

Tools/Scripts/webkitpy/layout_tests/servers/cli_wrapper.py

-#!/usr/bin/env python
-# Copyright (C) 2012 Google Inc. All rights reserved.
+# Copyright (C) 2010 Google Inc. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are
 # met:
 #
 #     * Redistributions of source code must retain the above copyright
 # notice, this list of conditions and the following disclaimer.
 #     * Redistributions in binary form must reproduce the above
 # copyright notice, this list of conditions and the following disclaimer
 # in the documentation and/or other materials provided with the
 # distribution.
 #     * Neither the name of Google Inc. nor the names of its
 # contributors may be used to endorse or promote products derived from
 # this software without specific prior written permission.
 #
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-"""A utility script for starting and stopping the web socket server with the
-   same configuration as used in the layout tests."""
+"""A utility script for starting and stopping servers as they are used in the layout tests."""
 
 import logging
 import optparse
-import tempfile
 
 from webkitpy.common.host import Host
-from webkitpy.layout_tests.servers import websocket_server
+
+_log = logging.getLogger(__name__)
 
 
-def main():
+def main(server_constructor, input_fn=None, argv=None, **kwargs):
+    input_fn = input_fn or raw_input
+
     option_parser = optparse.OptionParser()
-    option_parser.add_option('--server', type='choice',
-                             choices=['start', 'stop'], default='start',
-                             help='Server action (start|stop).')
     option_parser.add_option('--output-dir', dest='output_dir',
                              default=None, help='output directory.')
-    option_parser.add_option('-v', '--verbose', action='store_true',
-                             default=False,
-                             help='Include debug-level logging.')
-    options, args = option_parser.parse_args()
+    option_parser.add_option('-v', '--verbose', action='store_true')
+    options, args = option_parser.parse_args(argv)
 
-    if not options.output_dir:
-        options.output_dir = tempfile.gettempdir()
+    logging.basicConfig()
+    logger = logging.getLogger()
+    logger.setLevel(logging.DEBUG if options.verbose else logging.INFO)
 
     host = Host()
     port_obj = host.port_factory.get()
-    pywebsocket = websocket_server.PyWebSocket(port_obj, options.output_dir)
+    if not options.output_dir:
+        options.output_dir = port_obj.default_results_directory()
 
-    log_level = logging.WARN
-    if options.verbose:
-        log_level = logging.DEBUG
-    logging.basicConfig(level=log_level)
+    server = server_constructor(port_obj, options.output_dir, **kwargs)
+    server.start()
+    try:
+        _ = input_fn('Hit any key to stop the server and exit.')
+    except (KeyboardInterrupt, EOFError) as e:
+        pass
 
-    if 'start' == options.server:
-        pywebsocket.start()
-    else:
-        pywebsocket.stop()
-
-if '__main__' == __name__:
-    main()
+    server.stop()