acdf73a04e
There are some lines that will never be hit in tests, excluding those from being covered!
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from twython import TwythonStreamer, TwythonStreamError
|
|
|
|
from .config import (
|
|
app_key, app_secret, oauth_token, oauth_token_secret
|
|
)
|
|
|
|
import unittest
|
|
|
|
|
|
class TwythonStreamTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
class MyStreamer(TwythonStreamer):
|
|
def on_success(self, data):
|
|
self.disconnect()
|
|
|
|
def on_error(self, status_code, data):
|
|
raise TwythonStreamError(data)
|
|
|
|
self.api = MyStreamer(app_key, app_secret,
|
|
oauth_token, oauth_token_secret)
|
|
|
|
client_args = {
|
|
'headers': {
|
|
'User-Agent': '__twython__ Stream Test'
|
|
}
|
|
}
|
|
# Initialize with header for coverage checking for User-Agent
|
|
self.api_with_header = MyStreamer(app_key, app_secret,
|
|
oauth_token, oauth_token_secret,
|
|
client_args=client_args)
|
|
|
|
def test_stream_status_filter(self):
|
|
self.api.statuses.filter(track='twitter')
|
|
|
|
def test_stream_status_sample(self):
|
|
self.api.statuses.sample()
|
|
|
|
def test_stream_status_firehose(self):
|
|
self.assertRaises(TwythonStreamError, self.api.statuses.firehose,
|
|
track='twitter')
|
|
|
|
def test_stream_site(self):
|
|
self.assertRaises(TwythonStreamError, self.api.site,
|
|
follow='twitter')
|
|
|
|
def test_stream_user(self):
|
|
self.api.user(track='twitter')
|