chiark / gitweb /
examples/disorder.init.in: Collect default environment settings.
[disorder] / tests / play.py
1 #! /usr/bin/env python
2 #
3 # This file is part of DisOrder.
4 # Copyright (C) 2007-2009 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 3 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,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU 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, see <http://www.gnu.org/licenses/>.
18 #
19 import dtest,time,disorder,re,sys
20
21 def test():
22     """Play some tracks"""
23     dtest.start_daemon()
24     dtest.create_user()
25     dtest.rescan()                      # ensure all files are scanned
26     c = disorder.client()
27     c.random_disable()
28     assert c.random_enabled() == False
29     track = u"%s/Joe Bloggs/First Album/02:Second track.ogg" % dtest.tracks
30     track2 = u"%s/Joe Bloggs/First Album/04:Fourth track.ogg" % dtest.tracks
31     track3 = u"%s/Joe Bloggs/First Album/05:Fifth track.ogg" % dtest.tracks
32     print " adding track to queue"
33     c.disable()
34     assert c.enabled() == False
35     c.play(track)
36     print " checking track turned up in queue"
37     q = c.queue()
38     ts = filter(lambda t: t['track'] == track and 'submitter' in t, q)
39     assert len(ts) == 1, "checking track appears exactly once in queue"
40     t = ts[0]
41     assert t['submitter'] == u'fred', "check queue submitter"
42     i = t['id']
43     print " waiting for track"
44     c.enable()
45     assert c.enabled() == True
46     p = c.playing()
47     r = c.recent()
48     limit = 60
49     while not((p is not None and p['id'] == i)
50               or (len(filter(lambda t: t['track'] == track
51                              and 'submitter' in t, r)) > 0)) and limit > 0:
52         time.sleep(1)
53         p = c.playing()
54         r = c.recent()
55         limit -= 1
56     assert limit > 0, "check track did complete in a reasonable time"
57     print " checking track turned up in recent list"
58     while (p is not None and p['id'] == i):
59         time.sleep(1)
60         p = c.playing()
61     r = c.recent()
62     ts = filter(lambda t: t['track'] == track and 'submitter' in t, r)
63     assert len(ts) == 1, "check track appears exactly once in recent"
64     t = ts[0]
65     assert t['submitter'] == u'fred', "check recent entry submitter"
66
67     print " ensuring queue is clear"
68     c.disable()
69     while c.playing() is not None:
70         time.sleep(1)
71     q = c.queue()
72     for qe in q:
73         c.remove(qe["id"])
74
75     print " testing playafter"
76     print "  adding to empty queue"
77     c.playafter(None, [track])
78     q = c.queue()
79     print '\n'.join(map(lambda n: "%d: %s" % (n, q[n]["track"]),
80                         range(0, len(q))))
81     assert len(q) == 1
82     assert q[0]['track'] == track
83     print "  insert at start of queue"
84     c.playafter(None, [track2])
85     q = c.queue()
86     print '\n'.join(map(lambda n: "%d: %s" % (n, q[n]["track"]),
87                         range(0, len(q))))
88     assert len(q) == 2
89     assert q[0]['track'] == track2
90     assert q[1]['track'] == track
91     print "  insert in middle of queue"
92     c.playafter(q[0]['id'], [track3])
93     q = c.queue()
94     print '\n'.join(map(lambda n: "%d: %s" % (n, q[n]["track"]),
95                         range(0, len(q))))
96     assert len(q) == 3
97     assert q[0]['track'] == track2
98     assert q[1]['track'] == track3
99     assert q[2]['track'] == track
100     print "  insert multiple tracks at end of queue"
101     c.playafter(q[2]['id'], [track2, track])
102     q = c.queue()
103     print '\n'.join(map(lambda n: "%d: %s" % (n, q[n]["track"]),
104                         range(0, len(q))))
105     assert len(q) == 5
106     assert q[0]['track'] == track2
107     assert q[1]['track'] == track3
108     assert q[2]['track'] == track
109     assert q[3]['track'] == track2
110     assert q[4]['track'] == track
111
112     print " clearing queue"
113     for qe in q:
114         c.remove(qe["id"])
115
116     print " testing scratches"
117     retry = False
118     scratchlimit = 5
119     while scratchlimit > 0:
120         scratchlimit -= 1
121         c.disable()
122         print " starting a track"
123         c.play(track)
124         c.enable()
125         p = c.playing()
126         if p is None:
127             print " track played too quickly, trying again..."
128             continue
129         print " scratching track"
130         i = p['id']
131         c.scratch(i)
132         print " waiting for track to finish"
133         p = c.playing()
134         limit = 60
135         while (p is not None and p['id'] == i) and limit > 0:
136             time.sleep(1)
137             p = c.playing()
138             limit -= 1
139         assert limit > 0, "check track finishes in a reasonable period"
140         print " checking scratched track turned up in recent list"
141         r = c.recent()
142         ts = filter(lambda t: t['id'] == i, r)
143         assert len(ts) == 1, "check scratched track appears exactly once in recent"
144         if ts[0]['state'] == 'ok':
145             print " track played too quickly, trying again..."
146             continue
147         assert ts[0]['state'] == 'scratched', "checking track scratched"
148         break
149     if scratchlimit == 0:
150         # TODO this is really not a great approach!
151         print " didn't complete in a reasonable time"
152         sys.exit(77)
153     print " waiting for scratch to complete"
154     p = c.recent()
155     while p is not None:
156         time.sleep(1)
157         p = c.playing()
158     assert p is None, "checking nothing is playing"
159     assert c.enabled() == True
160     c.random_enable()
161     assert c.random_enabled() == True
162
163 if __name__ == '__main__':
164     dtest.run()