Begin experimental use of exceptions.

Provides IqTimeout and IqError which are raised when an Iq response
does not arrive in time, or it arrives with type='error'.

1 files changed, 4 insertions, 11 deletions

diff --git a/tests/test_stream_roster.py b/tests/test_stream_roster.py
index e1aa1766..95163744 100644
--- a/tests/test_stream_roster.py
+++ b/tests/test_stream_roster.py
@@ -111,19 +111,12 @@ class TestStreamRoster(SleekTest):
     def testRosterTimeout(self):
         """Test handling a timed out roster request."""
         self.stream_start()
-        events = []
-
-        def roster_timeout(event):
-            events.append('roster_timeout')
-
-        self.xmpp.add_event_handler('roster_timeout', roster_timeout)
-        self.xmpp.get_roster(timeout=0)
 
-        # Give the event queue time to process.
-        time.sleep(.1)
+        def do_test():
+            self.xmpp.get_roster(timeout=0)
+            time.sleep(.1)
 
-        self.failUnless(events == ['roster_timeout'],
-                 "Roster timeout event not triggered: %s." % events)
+        self.assertRaises(IqTimeout, do_test)
 
     def testRosterCallback(self):
         """Test handling a roster request callback."""