| 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() |