Add test for Stream.filter().
authorJoshua Roesslein <jroesslein@gmail.com>
Sun, 19 May 2013 08:11:50 +0000 (01:11 -0700)
committerJoshua Roesslein <jroesslein@gmail.com>
Sun, 19 May 2013 08:11:50 +0000 (01:11 -0700)
tests/test_streaming.py

index 0f32e9598533b4171f9f9a143544ee44a43aa7b7..d3fdefc6c961dbaaf4d84dbcb55e277669fd748c 100644 (file)
@@ -20,6 +20,10 @@ class MockStreamListener(StreamListener):
         if self.connect_cb:
             self.connect_cb()
 
+    def on_timeout(self):
+        self.test_case.fail('timeout')
+        return False
+
     def on_status(self, status):
         self.status_count += 1
         self.test_case.assertIsInstance(status, Status)
@@ -30,7 +34,7 @@ class TweepyStreamTests(unittest.TestCase):
     def setUp(self):
         self.auth = create_auth()
         self.listener = MockStreamListener(self)
-        self.stream = Stream(self.auth, self.listener)
+        self.stream = Stream(self.auth, self.listener, timeout=3.0)
 
     def tearDown(self):
         self.stream.disconnect()
@@ -51,3 +55,10 @@ class TweepyStreamTests(unittest.TestCase):
         self.assertEquals(self.listener.status_count,
                           self.listener.status_stop_count)
 
+    def test_filter_track(self):
+        self.listener.status_stop_count = 5
+        phrases = ['twitter']
+        self.stream.filter(track=phrases)
+        self.assertEquals(self.listener.status_count,
+                          self.listener.status_stop_count)
+