Remove Stream tests
authorHarmon <Harmon758@gmail.com>
Wed, 19 May 2021 15:34:58 +0000 (10:34 -0500)
committerHarmon <Harmon758@gmail.com>
Wed, 19 May 2021 15:34:58 +0000 (10:34 -0500)
tests/test_streaming.py [deleted file]

diff --git a/tests/test_streaming.py b/tests/test_streaming.py
deleted file mode 100644 (file)
index 8142450..0000000
+++ /dev/null
@@ -1,117 +0,0 @@
-import unittest
-from unittest.case import skip
-
-from mock import MagicMock, patch
-
-from .config import create_auth
-from .test_utils import mock_tweet
-from tweepy.api import API
-from tweepy.auth import OAuthHandler
-from tweepy.models import Status
-from tweepy.streaming import Stream, StreamListener
-
-
-class MockStreamListener(StreamListener):
-    def __init__(self, test_case):
-        super().__init__()
-        self.test_case = test_case
-        self.status_count = 0
-        self.status_stop_count = 0
-        self.connect_cb = None
-
-    def on_connect(self):
-        if self.connect_cb:
-            self.connect_cb()
-
-    def on_timeout(self):
-        self.test_case.fail('timeout')
-        return False
-
-    def on_error(self, code):
-        print(f"response: {code}")
-        return True
-
-    def on_status(self, status):
-        self.status_count += 1
-        self.test_case.assertIsInstance(status, Status)
-        if self.status_stop_count == self.status_count:
-            return False
-
-
-class TweepyStreamTests(unittest.TestCase):
-    def setUp(self):
-        self.auth = create_auth()
-        self.listener = MockStreamListener(self)
-        self.stream = Stream(self.auth, self.listener, timeout=3.0)
-
-    def tearDown(self):
-        self.stream.disconnect()
-
-    def on_connect(self):
-        API(self.auth).update_status(mock_tweet())
-
-    def test_sample(self):
-        self.listener.status_stop_count = 10
-        self.stream.sample()
-        self.assertEqual(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.assertEqual(self.listener.status_count,
-                         self.listener.status_stop_count)
-
-    def test_track_encoding(self):
-        s = Stream(None, None)
-        s._start = lambda is_async: None
-        s.filter(track=['Caf\xe9'])
-
-        # Should be UTF-8 encoded
-        self.assertEqual('Caf\xe9'.encode('utf8'), s.body['track'])
-
-    def test_follow_encoding(self):
-        s = Stream(None, None)
-        s._start = lambda is_async: None
-        s.filter(follow=['Caf\xe9'])
-
-        # Should be UTF-8 encoded
-        self.assertEqual('Caf\xe9'.encode('utf8'), s.body['follow'])
-
-
-class TweepyStreamBackoffTests(unittest.TestCase):
-    def setUp(self):
-        #bad auth causes twitter to return 401 errors
-        self.auth = OAuthHandler("bad-key", "bad-secret")
-        self.auth.set_access_token("bad-token", "bad-token-secret")
-        self.listener = MockStreamListener(self)
-        self.stream = Stream(self.auth, self.listener)
-
-    def tearDown(self):
-        self.stream.disconnect()
-
-    def test_exp_backoff(self):
-        self.stream = Stream(self.auth, self.listener, timeout=3.0,
-                             retry_count=1, retry_time=1.0, retry_time_cap=100.0)
-        self.stream.sample()
-        # 1 retry, should be 4x the retry_time
-        self.assertEqual(self.stream.retry_time, 4.0)
-
-    def test_exp_backoff_cap(self):
-        self.stream = Stream(self.auth, self.listener, timeout=3.0,
-                             retry_count=1, retry_time=1.0, retry_time_cap=3.0)
-        self.stream.sample()
-        # 1 retry, but 4x the retry_time exceeds the cap, so should be capped
-        self.assertEqual(self.stream.retry_time, 3.0)
-
-    mock_resp = MagicMock()
-    mock_resp.return_value.status_code = 420
-
-    @patch('requests.Session.request', mock_resp)
-    def test_420(self):
-        self.stream = Stream(self.auth, self.listener, timeout=3.0, retry_count=0,
-                             retry_time=1.0, retry_420=1.5, retry_time_cap=20.0)
-        self.stream.sample()
-        # no retries, but error 420, should be double the retry_420, not double the retry_time
-        self.assertEqual(self.stream.retry_time, 3.0)