Limit how long tests/play.py will wait

[disorder] / tests / play.py

diff --git a/tests/play.py b/tests/play.py
index cbbb0ec10313f02ef205fcedf9234655528f1b39..43deaff7dc9d04c794fc99957e8cec0190b7b19a 100755 (executable)
--- a/tests/play.py
+++ b/tests/play.py
@@ -1,7 +1,7 @@
 #! /usr/bin/env python
 #
 # This file is part of DisOrder.
 #! /usr/bin/env python
 #
 # This file is part of DisOrder.

-# Copyright (C) 2007 Richard Kettlewell
+# Copyright (C) 2007, 2008 Richard Kettlewell

 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by


@@ -23,33 +23,86 @@ import dtest,time,disorder,re
 def test():
     """Play some tracks"""
     dtest.start_daemon()
 def test():
     """Play some tracks"""
     dtest.start_daemon()

+    dtest.create_user()

     c = disorder.client()
     c = disorder.client()

+    c.random_disable()
+    assert c.random_enabled() == False

     track = u"%s/Joe Bloggs/First Album/02:Second track.ogg" % dtest.tracks
     track = u"%s/Joe Bloggs/First Album/02:Second track.ogg" % dtest.tracks

-    print "adding track to queue"
+    print " adding track to queue"
+    c.disable()
+    assert c.enabled() == False

     c.play(track)
     c.play(track)

-    print "checking track turned up in queue"
+    print " checking track turned up in queue"

     q = c.queue()
     ts = filter(lambda t: t['track'] == track and 'submitter' in t, q)
     q = c.queue()
     ts = filter(lambda t: t['track'] == track and 'submitter' in t, q)

-    assert len(ts) == 1
+    assert len(ts) == 1, "checking track appears exactly once in queue"

     t = ts[0]
     assert t['submitter'] == u'fred', "check queue submitter"
     i = t['id']
     t = ts[0]
     assert t['submitter'] == u'fred', "check queue submitter"
     i = t['id']

-    print "waiting for track"
+    print " waiting for track"
+    c.enable()
+    assert c.enabled() == True

     p = c.playing()
     r = c.recent()
     p = c.playing()
     r = c.recent()

+    limit = 60

     while not((p is not None and p['id'] == i)
     while not((p is not None and p['id'] == i)

-              or (len(filter(lambda t: t['track'] == track and 'submitter' in t, r)) > 0)):
+              or (len(filter(lambda t: t['track'] == track
+                             and 'submitter' in t, r)) > 0)) and limit > 0:

         time.sleep(1)
         p = c.playing()
         r = c.recent()
         time.sleep(1)
         p = c.playing()
         r = c.recent()

-    print "checking track turned up in recent list"
-    q = c.recent()
-    ts = filter(lambda t: t['track'] == track and 'submitter' in t, q)
-    assert len(ts) == 1
+        limit -= 1
+    assert limit > 0, "check track did complete in a reasonable time"
+    print " checking track turned up in recent list"
+    while (p is not None and p['id'] == i):
+        time.sleep(1)
+        p = c.playing()
+    r = c.recent()
+    ts = filter(lambda t: t['track'] == track and 'submitter' in t, r)
+    assert len(ts) == 1, "check track appears exactly once in recent"

     t = ts[0]
     assert t['submitter'] == u'fred', "check recent entry submitter"
     t = ts[0]
     assert t['submitter'] == u'fred', "check recent entry submitter"

-    
-        
+
+    print " testing scratches"
+    retry = False
+    while True:
+        c.disable()
+        print " starting a track"
+        c.play(track)
+        c.enable()
+        p = c.playing()
+        if p is None:
+            print " track played too quickly, trying again..."
+            continue
+        print " scratching track"
+        i = p['id']
+        c.scratch(i)
+        print " waiting for track to finish"
+        p = c.playing()
+        limit = 60
+        while (p is not None and p['id'] == i) and limit > 0:
+            time.sleep(1)
+            p = c.playing()
+            limit -= 1
+        assert limit > 0, "check track finishes in a reasonable period"
+        print " checking scratched track turned up in recent list"
+        r = c.recent()
+        ts = filter(lambda t: t['id'] == i, r)
+        assert len(ts) == 1, "check scratched track appears exactly once in recent"
+        if ts[0]['state'] == 'ok':
+            print " track played too quickly, trying again..."
+            continue
+        assert ts[0]['state'] == 'scratched', "checking track scratched"
+        break
+    print " waiting for scratch to complete"
+    p = c.recent()
+    while p is not None:
+        time.sleep(1)
+        p = c.playing()
+    assert p is None, "checking nothing is playing"
+    assert c.enabled() == True
+    c.random_enable()
+    assert c.random_enabled() == True

 
 if __name__ == '__main__':
     dtest.run()
 
 if __name__ == '__main__':
     dtest.run()