X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/disorder/blobdiff_plain/cdabf44d4bf72678b402c0fd7dac394eb36513da..05fd0aa117566503026dbedfe4d9b0485da2d5ed:/tests/dtest.py diff --git a/tests/dtest.py b/tests/dtest.py index a5a8be9..ffb63cb 100644 --- a/tests/dtest.py +++ b/tests/dtest.py @@ -1,7 +1,7 @@ #-*-python-*- # # This file is part of DisOrder. -# Copyright (C) 2007-2009 Richard Kettlewell +# Copyright (C) 2007-2012 Richard Kettlewell # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by @@ -20,12 +20,20 @@ """Utility module used by tests""" import os,os.path,subprocess,sys,re,time,unicodedata,random,socket,traceback +import atexit,base64,errno + +homelink = None def fatal(s): """Write an error message and exit""" sys.stderr.write("ERROR: %s\n" % s) sys.exit(1) +@atexit.register +def cleanup(): + if homelink is not None: + os.unlink(homelink) + # Identify the top build directory cwd = os.getcwd() if os.path.exists("config.h"): @@ -167,10 +175,9 @@ def bindable(p): def default_config(encoding="UTF-8"): """Write the default config""" open("%s/config" % testroot, "w").write( - """home %s/home + """home %s collection fs %s %s/tracks scratch %s/scratch.ogg -gap 0 queue_pad 5 stopword 01 02 03 04 05 06 07 08 09 10 stopword 1 2 3 4 5 6 7 8 9 @@ -194,12 +201,33 @@ api rtp broadcast 127.0.0.1 %d broadcast_from 127.0.0.1 %d mail_sender no.such.user.sorry@greenend.org.uk -""" % (testroot, encoding, testroot, testroot, top_builddir, top_builddir, +""" % (homelink, encoding, testroot, testroot, top_builddir, top_builddir, port, port + 1)) def common_setup(): + global homelink remove_dir(testroot) - os.mkdir(testroot) + os.makedirs(testroot) + os.makedirs("%s/home" % testroot) + # Establish a symlink to the home directory, to keep the socket pathnames + # short enough. + tmpdir = "/tmp" + for v in ["TMPDIR", "TMP"]: + try: tmpdir = os.environ[v] + except KeyError: pass + else: break + for i in xrange(1024): + r = base64.b64encode(os.urandom(9)).replace("/", "_") + f = "%s/disorder-home.%s" % (tmpdir, r) + try: + os.symlink("%s/home" % testroot, f) + except OSError, e: + if e.errno != errno.EEXIST: raise + else: + homelink = f + break + else: + fatal("failed to make home link") # Choose a port global port port = random.randint(49152, 65530) @@ -229,7 +257,7 @@ Start the daemon.""" time.sleep(1) print " starting daemon" # remove the socket if it exists - socket = "%s/home/socket" % testroot + socket = "%s/socket" % homelink if os.path.exists(socket): os.remove(socket) daemon = subprocess.Popen(["disorderd", @@ -256,8 +284,8 @@ Start the daemon.""" if waited > 0: print " took about %ss for socket to appear" % waited # Wait for root user to be created - command(["disorder", - "--config", disorder._configfile, "--no-per-user-config", + command(["disorderd", + "--config", disorder._configfile, "--wait-for-root"]) def create_user(username="fred", password="fredpass"): @@ -411,5 +439,6 @@ def command(args): tests = 0 failures = 0 daemon = None -testroot = "%s/tests/testroot" % top_builddir +testroot = "%s/tests/testroot/%s" % \ + (top_builddir, os.path.basename(sys.argv[0])) tracks = "%s/tracks" % testroot