native_client_sdk/src/build_tools/tests/test_server.py

   1 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import multiprocessing
   6 import os
   7 import SimpleHTTPServer
   8
   9
  10 class LocalHTTPServer(object):
  11   """Class to start a local HTTP server as a child process."""
  12
  13   def __init__(self, serve_dir):
  14     parent_conn, child_conn = multiprocessing.Pipe()
  15     self.process = multiprocessing.Process(target=_HTTPServerProcess,
  16                                            args=(child_conn, serve_dir))
  17     self.process.start()
  18     if parent_conn.poll(10):  # wait 10 seconds
  19       self.port = parent_conn.recv()
  20     else:
  21       raise Exception('Unable to launch HTTP server.')
  22
  23     self.conn = parent_conn
  24
  25   def Shutdown(self):
  26     """Send a message to the child HTTP server process and wait for it to
  27         finish."""
  28     self.conn.send(False)
  29     self.process.join()
  30
  31   def GetURL(self, rel_url):
  32     """Get the full url for a file on the local HTTP server.
  33
  34     Args:
  35       rel_url: A URL fragment to convert to a full URL. For example,
  36           GetURL('foobar.baz') -> 'http://localhost:1234/foobar.baz'
  37     """
  38     return 'http://localhost:%d/%s' % (self.port, rel_url)
  39
  40
  41 class QuietHTTPRequestHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
  42   def log_message(self, msg_format, *args):
  43     pass
  44
  45
  46 def _HTTPServerProcess(conn, serve_dir):
  47   """Run a local httpserver with a randomly-chosen port.
  48
  49   This function assumes it is run as a child process using multiprocessing.
  50
  51   Args:
  52     conn: A connection to the parent process. The child process sends
  53         the local port, and waits for a message from the parent to
  54         stop serving.
  55     serve_dir: The directory to serve. All files are accessible through
  56        http://localhost:<port>/path/to/filename.
  57   """
  58   import BaseHTTPServer
  59
  60   os.chdir(serve_dir)
  61   httpd = BaseHTTPServer.HTTPServer(('', 0), QuietHTTPRequestHandler)
  62   conn.send(httpd.server_address[1])  # the chosen port number
  63   httpd.timeout = 0.5  # seconds
  64   running = True
  65   while running:
  66     httpd.handle_request()
  67     if conn.poll():
  68       running = conn.recv()
  69   conn.close()