chiark - git - mdw - disorder/blob - tests/dtest.py

   1 #-*-python-*-
   2
   3 """Utility module used by tests"""
   4
   5 import os,os.path,subprocess,sys,disorder
   6
   7 def copyfile(a,b):
   8     """copyfile(A, B)
   9 Copy A to B."""
  10     open(b,"w").write(open(a).read())
  11
  12 def maketrack(s):
  13     """maketrack(S)
  14
  15 Make track with relative path S exist"""
  16     trackpath = "%s/tracks/%s" % (testroot, s)
  17     trackdir = os.path.dirname(trackpath)
  18     if not os.path.exists(trackdir):
  19         os.makedirs(trackdir)
  20     copyfile("%s/sounds/slap.ogg" % topsrcdir, trackpath)
  21
  22 def stdtracks():
  23     maketrack("Joe Bloggs/First Album/01:First track.ogg")
  24     maketrack("Joe Bloggs/First Album/02:Second track.ogg")
  25     maketrack("Joe Bloggs/First Album/03:Third track.ogg")
  26     maketrack("Joe Bloggs/First Album/04:Fourth track.ogg")
  27     maketrack("Joe Bloggs/First Album/05:Fifth track.ogg")
  28     maketrack("Joe Bloggs/First Album/05:Fifth track.ogg")
  29     maketrack("Joe Bloggs/Second Album/01:First track.ogg")
  30     maketrack("Joe Bloggs/Second Album/02:Second track.ogg")
  31     maketrack("Joe Bloggs/Second Album/03:Third track.ogg")
  32     maketrack("Joe Bloggs/Second Album/04:Fourth track.ogg")
  33     maketrack("Joe Bloggs/Second Album/05:Fifth track.ogg")
  34     maketrack("Joe Bloggs/Second Album/05:Fifth track.ogg")
  35     maketrack("Joe Bloggs/First Album/01:First track.ogg")
  36     maketrack("Joe Bloggs/First Album/02:Second track.ogg")
  37     maketrack("Joe Bloggs/First Album/03:Third track.ogg")
  38     maketrack("Joe Bloggs/First Album/04:Fourth track.ogg")
  39     maketrack("Joe Bloggs/First Album/05:Fifth track.ogg")
  40     maketrack("Fred Smith/Boring/01:Dull.ogg")
  41     maketrack("Fred Smith/Boring/02:Tedious.ogg")
  42     maketrack("Fred Smith/Boring/03:Drum Solo.ogg")
  43     maketrack("Fred Smith/Boring/04:Yawn.ogg")
  44     maketrack("misc/blahblahblah.ogg")
  45     maketrack("Various/Greatest Hits/01:Jim Whatever - Spong.ogg")
  46     maketrack("Various/Greatest Hits/02:Joe Bloggs - Yadda.ogg")
  47
  48 def notracks():
  49     pass
  50
  51 def start(test):
  52     """start(TEST)
  53
  54 Start the daemon for test called TEST."""
  55     global daemon
  56     assert daemon == None
  57     if test == None:
  58         errs = sys.stderr
  59     else:
  60         errs = open("%s/%s.log" % (testroot, test), "w")
  61     server = None
  62     print " starting daemon"
  63     daemon = subprocess.Popen(["disorderd",
  64                                "--foreground",
  65                                "--config", "%s/config" % testroot],
  66                               stderr=errs)
  67     disorder._configfile = "%s/config" % testroot
  68     disorder._userconf = False
  69
  70 def stop():
  71     """stop()
  72
  73 Stop the daemon if it has not stopped already"""
  74     global daemon
  75     rc = daemon.poll()
  76     if rc == None:
  77         print " stopping daemon"
  78         os.kill(daemon.pid, 15)
  79         rc = daemon.wait()
  80     print " daemon has stopped"
  81     daemon = None
  82
  83 def run(test, setup=None, report=True, name=None):
  84     global tests
  85     tests += 1
  86     if setup == None:
  87         setup = stdtracks
  88     setup()
  89     start(name)
  90     try:
  91         try:
  92             test()
  93         except AssertionError, e:
  94             global failures
  95             failures += 1
  96             print e
  97     finally:
  98         stop()
  99     if report:
 100         if failures:
 101             print " FAILED"
 102             sys.exit(1)
 103         else:
 104             print " OK"
 105
 106 def remove_dir(d):
 107     """remove_dir(D)
 108
 109 Recursively delete directory D"""
 110     if os.path.lexists(d):
 111         if os.path.isdir(d):
 112             for dd in os.listdir(d):
 113                 remove_dir("%s/%s" % (d, dd))
 114             os.rmdir(d)
 115         else:
 116             os.remove(d)
 117
 118 # -----------------------------------------------------------------------------
 119 # Common setup
 120
 121 tests = 0
 122 failures = 0
 123 daemon = None
 124 testroot = "%s/testroot" % os.getcwd()
 125 topsrcdir = os.path.abspath(os.getenv("topsrcdir"))
 126 remove_dir(testroot)
 127 os.mkdir(testroot)
 128 open("%s/config" % testroot, "w").write(
 129 """player *.ogg shell 'echo "$TRACK" >> %s/played.log'
 130 home %s
 131 collection fs ASCII %s/tracks
 132 scratch %s/scratch.ogg
 133 gap 0
 134 stopword 01 02 03 04 05 06 07 08 09 10
 135 stopword 1 2 3 4 5 6 7 8 9
 136 stopword 11 12 13 14 15 16 17 18 19 20
 137 stopword 21 22 23 24 25 26 27 28 29 30
 138 stopword the a an and to too in on of we i am as im for is
 139 username fred
 140 password fredpass
 141 allow fred fredpass
 142 """ % (testroot, testroot, testroot, testroot))
 143 copyfile("%s/sounds/scratch.ogg" % topsrcdir,
 144          "%s/scratch.ogg" % testroot)