Commit | Line | Data |
---|---|---|
c5dbcd79 | 1 | #-*-python-*- |
aa896435 RK |
2 | # |
3 | # This file is part of DisOrder. | |
4 | # Copyright (C) 2007 Richard Kettlewell | |
5 | # | |
6 | # This program is free software; you can redistribute it and/or modify | |
7 | # it under the terms of the GNU General Public License as published by | |
8 | # the Free Software Foundation; either version 2 of the License, or | |
9 | # (at your option) any later version. | |
10 | # | |
11 | # This program is distributed in the hope that it will be useful, but | |
12 | # WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | # General Public License for more details. | |
15 | # | |
16 | # You should have received a copy of the GNU General Public License | |
17 | # along with this program; if not, write to the Free Software | |
18 | # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 | |
19 | # USA | |
20 | # | |
c5dbcd79 RK |
21 | |
22 | """Utility module used by tests""" | |
23 | ||
3bfecfac RK |
24 | import os,os.path,subprocess,sys,re |
25 | ||
26 | def fatal(s): | |
27 | """Write an error message and exit""" | |
28 | sys.stderr.write("ERROR: %s\n" % s) | |
29 | sys.exit(1) | |
30 | ||
31 | # Identify the top build directory | |
32 | cwd = os.getcwd() | |
33 | if os.path.exists("config.h"): | |
34 | top_builddir = cwd | |
35 | elif os.path.exists("alltests"): | |
36 | top_builddir = os.path.dirname(cwd) | |
37 | else: | |
38 | fatal("cannot identify build directory") | |
39 | ||
40 | # Make sure the Python build directory is on the module search path | |
41 | sys.path.insert(0, os.path.join(top_builddir, "python")) | |
42 | import disorder | |
43 | ||
44 | # Make sure the server build directory is on the executable search path | |
45 | ospath = os.environ["PATH"].split(os.pathsep) | |
46 | ospath.insert(0, os.path.join(top_builddir, "server")) | |
47 | os.environ["PATH"] = os.pathsep.join(ospath) | |
48 | ||
49 | # Parse the makefile in the current directory to identify the source directory | |
50 | top_srcdir = None | |
51 | for l in file("Makefile"): | |
52 | r = re.match("top_srcdir *= *(.*)", l) | |
53 | if r: | |
54 | top_srcdir = r.group(1) | |
55 | break | |
56 | if not top_srcdir: | |
57 | fatal("cannot identify source directory") | |
58 | ||
59 | # The tests source directory must be on the module search path already since | |
60 | # we found dtest.py | |
61 | ||
62 | # ----------------------------------------------------------------------------- | |
c5dbcd79 RK |
63 | |
64 | def copyfile(a,b): | |
65 | """copyfile(A, B) | |
66 | Copy A to B.""" | |
67 | open(b,"w").write(open(a).read()) | |
68 | ||
69 | def maketrack(s): | |
70 | """maketrack(S) | |
71 | ||
72 | Make track with relative path S exist""" | |
f31e414d | 73 | trackpath = "%s/tracks/%s" % (testroot, s) |
c5dbcd79 RK |
74 | trackdir = os.path.dirname(trackpath) |
75 | if not os.path.exists(trackdir): | |
76 | os.makedirs(trackdir) | |
3bfecfac | 77 | copyfile("%s/sounds/slap.ogg" % top_srcdir, trackpath) |
c5dbcd79 RK |
78 | |
79 | def stdtracks(): | |
80 | maketrack("Joe Bloggs/First Album/01:First track.ogg") | |
81 | maketrack("Joe Bloggs/First Album/02:Second track.ogg") | |
82 | maketrack("Joe Bloggs/First Album/03:Third track.ogg") | |
83 | maketrack("Joe Bloggs/First Album/04:Fourth track.ogg") | |
84 | maketrack("Joe Bloggs/First Album/05:Fifth track.ogg") | |
85 | maketrack("Joe Bloggs/First Album/05:Fifth track.ogg") | |
86 | maketrack("Joe Bloggs/Second Album/01:First track.ogg") | |
87 | maketrack("Joe Bloggs/Second Album/02:Second track.ogg") | |
88 | maketrack("Joe Bloggs/Second Album/03:Third track.ogg") | |
89 | maketrack("Joe Bloggs/Second Album/04:Fourth track.ogg") | |
90 | maketrack("Joe Bloggs/Second Album/05:Fifth track.ogg") | |
91 | maketrack("Joe Bloggs/Second Album/05:Fifth track.ogg") | |
92 | maketrack("Joe Bloggs/First Album/01:First track.ogg") | |
93 | maketrack("Joe Bloggs/First Album/02:Second track.ogg") | |
94 | maketrack("Joe Bloggs/First Album/03:Third track.ogg") | |
95 | maketrack("Joe Bloggs/First Album/04:Fourth track.ogg") | |
96 | maketrack("Joe Bloggs/First Album/05:Fifth track.ogg") | |
97 | maketrack("Fred Smith/Boring/01:Dull.ogg") | |
98 | maketrack("Fred Smith/Boring/02:Tedious.ogg") | |
99 | maketrack("Fred Smith/Boring/03:Drum Solo.ogg") | |
100 | maketrack("Fred Smith/Boring/04:Yawn.ogg") | |
101 | maketrack("misc/blahblahblah.ogg") | |
102 | maketrack("Various/Greatest Hits/01:Jim Whatever - Spong.ogg") | |
103 | maketrack("Various/Greatest Hits/02:Joe Bloggs - Yadda.ogg") | |
104 | ||
105 | def notracks(): | |
106 | pass | |
107 | ||
108 | def start(test): | |
109 | """start(TEST) | |
110 | ||
111 | Start the daemon for test called TEST.""" | |
112 | global daemon | |
113 | assert daemon == None | |
114 | if test == None: | |
115 | errs = sys.stderr | |
116 | else: | |
117 | errs = open("%s/%s.log" % (testroot, test), "w") | |
118 | server = None | |
119 | print " starting daemon" | |
120 | daemon = subprocess.Popen(["disorderd", | |
121 | "--foreground", | |
122 | "--config", "%s/config" % testroot], | |
123 | stderr=errs) | |
eee9d4b3 RK |
124 | disorder._configfile = "%s/config" % testroot |
125 | disorder._userconf = False | |
c5dbcd79 RK |
126 | |
127 | def stop(): | |
128 | """stop() | |
129 | ||
130 | Stop the daemon if it has not stopped already""" | |
131 | global daemon | |
132 | rc = daemon.poll() | |
133 | if rc == None: | |
eee9d4b3 | 134 | print " stopping daemon" |
c5dbcd79 RK |
135 | os.kill(daemon.pid, 15) |
136 | rc = daemon.wait() | |
137 | print " daemon has stopped" | |
138 | daemon = None | |
139 | ||
140 | def run(test, setup=None, report=True, name=None): | |
141 | global tests | |
142 | tests += 1 | |
143 | if setup == None: | |
144 | setup = stdtracks | |
145 | setup() | |
146 | start(name) | |
147 | try: | |
eee9d4b3 RK |
148 | try: |
149 | test() | |
150 | except AssertionError, e: | |
151 | global failures | |
152 | failures += 1 | |
153 | print e | |
154 | finally: | |
155 | stop() | |
c5dbcd79 RK |
156 | if report: |
157 | if failures: | |
158 | print " FAILED" | |
159 | sys.exit(1) | |
160 | else: | |
161 | print " OK" | |
162 | ||
163 | def remove_dir(d): | |
164 | """remove_dir(D) | |
165 | ||
166 | Recursively delete directory D""" | |
167 | if os.path.lexists(d): | |
168 | if os.path.isdir(d): | |
169 | for dd in os.listdir(d): | |
170 | remove_dir("%s/%s" % (d, dd)) | |
171 | os.rmdir(d) | |
172 | else: | |
173 | os.remove(d) | |
174 | ||
175 | # ----------------------------------------------------------------------------- | |
176 | # Common setup | |
177 | ||
178 | tests = 0 | |
179 | failures = 0 | |
180 | daemon = None | |
3bfecfac | 181 | testroot = "%s/tests/testroot" % top_builddir |
c5dbcd79 RK |
182 | remove_dir(testroot) |
183 | os.mkdir(testroot) | |
902b7f9d RK |
184 | open("%s/config" % testroot, "w").write( |
185 | """player *.ogg shell 'echo "$TRACK" >> %s/played.log' | |
c5dbcd79 RK |
186 | home %s |
187 | collection fs ASCII %s/tracks | |
188 | scratch %s/scratch.ogg | |
189 | gap 0 | |
190 | stopword 01 02 03 04 05 06 07 08 09 10 | |
191 | stopword 1 2 3 4 5 6 7 8 9 | |
192 | stopword 11 12 13 14 15 16 17 18 19 20 | |
193 | stopword 21 22 23 24 25 26 27 28 29 30 | |
194 | stopword the a an and to too in on of we i am as im for is | |
eee9d4b3 RK |
195 | username fred |
196 | password fredpass | |
197 | allow fred fredpass | |
be3ebd90 RK |
198 | plugins ../plugins |
199 | player *.mp3 execraw disorder-decode | |
200 | player *.ogg execraw disorder-decode | |
201 | player *.wav execraw disorder-decode | |
202 | player *.flac execraw disorder-decode | |
203 | tracklength *.mp3 disorder-tracklength | |
204 | tracklength *.ogg disorder-tracklength | |
205 | tracklength *.wav disorder-tracklength | |
206 | tracklength *.flac disorder-tracklength | |
c5dbcd79 | 207 | """ % (testroot, testroot, testroot, testroot)) |
3bfecfac | 208 | copyfile("%s/sounds/scratch.ogg" % top_srcdir, |
c5dbcd79 | 209 | "%s/scratch.ogg" % testroot) |