Commit 94378e23 authored by Jim Fulton's avatar Jim Fulton

Improved the way the test web server is handled.

Added an api for registering tearDown handlers.
parent f0d72696
...@@ -105,12 +105,15 @@ def buildoutSetUp(test, clear_home=True): ...@@ -105,12 +105,15 @@ def buildoutSetUp(test, clear_home=True):
get = get, get = get,
__original_wd__ = os.getcwd(), __original_wd__ = os.getcwd(),
__temporary_directories__ = temporary_directories, __temporary_directories__ = temporary_directories,
__tearDown__ = [],
mkdtemp = mkdtemp, mkdtemp = mkdtemp,
)) ))
def buildoutTearDown(test): def buildoutTearDown(test):
for d in test.globs['__temporary_directories__']: for d in test.globs['__temporary_directories__']:
shutil.rmtree(d) shutil.rmtree(d)
for f in test.globs['__tearDown__']:
f()
os.chdir(test.globs['__original_wd__']) os.chdir(test.globs['__original_wd__'])
if test.globs.get('_oldhome') is not None: if test.globs.get('_oldhome') is not None:
os.environ['HOME'] = test.globs['_oldhome'] os.environ['HOME'] = test.globs['_oldhome']
...@@ -290,17 +293,31 @@ def get_port(): ...@@ -290,17 +293,31 @@ def get_port():
s.close() s.close()
raise RuntimeError, "Can't find port" raise RuntimeError, "Can't find port"
def start_server(tree): def _start_server(tree, name=''):
port = get_port() port = get_port()
threading.Thread(target=_run, args=(tree, port)).start() thread = threading.Thread(target=_run, args=(tree, port), name=name)
thread.setDaemon(True)
thread.start()
wait(port, up=True) wait(port, up=True)
return port return port, thread
def start_server(tree):
return _start_server(tree)[0]
def stop_server(url): def stop_server(url, thread=None):
try: try:
urllib2.urlopen(url+'__stop__') urllib2.urlopen(url+'__stop__')
except Exception: except Exception:
pass pass
if thread is not None:
thread.join() # wait for thread to stop
def setUpServer(test, tree):
port, thread = _start_server(tree, name=test.name)
link_server = 'http://localhost:%s/' % port
test.globs['link_server'] = link_server
test.globs['__tearDown__'].append(lambda: stop_server(link_server, thread))
def wait(port, up): def wait(port, up):
addr = 'localhost', port addr = 'localhost', port
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment