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

   1 #! /usr/bin/env python
   2 #
   3 # This file is part of DisOrder.
   4 # Copyright (C) 2007, 2008 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()