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 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 #
  21 import dtest,time,disorder,re,sys
  22
  23 def test():
  24     """Play some tracks"""
  25     dtest.start_daemon()
  26     dtest.create_user()
  27     dtest.rescan()                      # ensure all files are scanned
  28     c = disorder.client()
  29     c.random_disable()
  30     assert c.random_enabled() == False
  31     track = u"%s/Joe Bloggs/First Album/02:Second 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 " testing scratches"
  68     retry = False
  69     scratchlimit = 5
  70     while scratchlimit > 0:
  71         scratchlimit -= 1
  72         c.disable()
  73         print " starting a track"
  74         c.play(track)
  75         c.enable()
  76         p = c.playing()
  77         if p is None:
  78             print " track played too quickly, trying again..."
  79             continue
  80         print " scratching track"
  81         i = p['id']
  82         c.scratch(i)
  83         print " waiting for track to finish"
  84         p = c.playing()
  85         limit = 60
  86         while (p is not None and p['id'] == i) and limit > 0:
  87             time.sleep(1)
  88             p = c.playing()
  89             limit -= 1
  90         assert limit > 0, "check track finishes in a reasonable period"
  91         print " checking scratched track turned up in recent list"
  92         r = c.recent()
  93         ts = filter(lambda t: t['id'] == i, r)
  94         assert len(ts) == 1, "check scratched track appears exactly once in recent"
  95         if ts[0]['state'] == 'ok':
  96             print " track played too quickly, trying again..."
  97             continue
  98         assert ts[0]['state'] == 'scratched', "checking track scratched"
  99         break
 100     if scratchlimit == 0:
 101         # TODO this is really not a great approach!
 102         print " didn't complete in a reasonable time"
 103         sys.exit(77)
 104     print " waiting for scratch to complete"
 105     p = c.recent()
 106     while p is not None:
 107         time.sleep(1)
 108         p = c.playing()
 109     assert p is None, "checking nothing is playing"
 110     assert c.enabled() == True
 111     c.random_enable()
 112     assert c.random_enabled() == True
 113
 114 if __name__ == '__main__':
 115     dtest.run()