From: Joshua Roesslein <jroesslein@gmail.com>
Date: Sun, 19 May 2013 08:11:50 +0000 (-0700)
Subject: Add test for Stream.filter().
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=02a43b7f5f30569934dd9785d0d1423cf7ccaf95;p=tweepy.git

Add test for Stream.filter().
---

diff --git a/tests/test_streaming.py b/tests/test_streaming.py
index 0f32e95..d3fdefc 100644
--- a/tests/test_streaming.py
+++ b/tests/test_streaming.py
@@ -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)
+