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
   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/%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
  68 def stop():
  69     """stop()
  70
  71 Stop the daemon if it has not stopped already"""
  72     global daemon
  73     rc = daemon.poll()
  74     if rc == None:
  75         os.kill(daemon.pid, 15)
  76         rc = daemon.wait()
  77     print " daemon has stopped"
  78     daemon = None
  79
  80 def run(test, setup=None, report=True, name=None):
  81     global tests
  82     tests += 1
  83     if setup == None:
  84         setup = stdtracks
  85     setup()
  86     start(name)
  87     try:
  88         test()
  89     except AssertionError, e:
  90         global failures
  91         failures += 1
  92         print e
  93
  94     stop()
  95     if report:
  96         if failures:
  97             print " FAILED"
  98             sys.exit(1)
  99         else:
 100             print " OK"
 101
 102 def remove_dir(d):
 103     """remove_dir(D)
 104
 105 Recursively delete directory D"""
 106     if os.path.lexists(d):
 107         if os.path.isdir(d):
 108             for dd in os.listdir(d):
 109                 remove_dir("%s/%s" % (d, dd))
 110             os.rmdir(d)
 111         else:
 112             os.remove(d)
 113
 114 # -----------------------------------------------------------------------------
 115 # Common setup
 116
 117 tests = 0
 118 failures = 0
 119 daemon = None
 120 testroot = "%s/testroot" % os.getcwd()
 121 topsrcdir = os.path.abspath(os.getenv("topsrcdir"))
 122 remove_dir(testroot)
 123 os.mkdir(testroot)
 124 open("%s/config" % testroot, "w").write(
 125 """player *.ogg shell 'echo "$TRACK" >> %s/played.log'
 126 home %s
 127 collection fs ASCII %s/tracks
 128 scratch %s/scratch.ogg
 129 gap 0
 130 stopword 01 02 03 04 05 06 07 08 09 10
 131 stopword 1 2 3 4 5 6 7 8 9
 132 stopword 11 12 13 14 15 16 17 18 19 20
 133 stopword 21 22 23 24 25 26 27 28 29 30
 134 stopword the a an and to too in on of we i am as im for is
 135 """ % (testroot, testroot, testroot, testroot))
 136 copyfile("%s/sounds/scratch.ogg" % topsrcdir,
 137          "%s/scratch.ogg" % testroot)