class TestQueueListener(object):
def setup(self):
self.listener = QueueListener()
send_listener(self.listener, EXAMPLE_TWEETS)
def test_iterable(self):
tweets = iter(self.listener)
next(tweets)
def test_queue_empty(self):
for tweet in self.listener:
pass
def test_length(self):
assert len(self.listener) == len(EXAMPLE_TWEETS)
def test_on_data_limit(self):
self.listener.on_data('{"limit":{"track":47}}')
assert self.listener.limit_track == 47
def test_on_data_delete(self):
mock = MagicMock()
self.listener.on_delete = mock
self.listener.on_data('{"delete":{"status":{"id": 1, "user_id": 2}}}')
expected = [call(1, 2)]
calls = mock.mock_calls
assert calls == expected
def test_on_data_unroutable(self):
assert_raises(UnroutableError, self.listener.on_data, '{"none": null}')
def test_on_error(self):
assert_raises(TwitterStreamError, self.listener.on_error, 404)
def test_maxsize(self):
self.listener = QueueListener(maxsize=1)
assert self.listener.on_data(EXAMPLE_TWEETS[0])
assert not self.listener.on_data(EXAMPLE_TWEETS[0])
def test_maxsize(self):
self.listener = QueueListener(maxsize=1)
assert self.listener.on_data(EXAMPLE_TWEETS[0])
assert not self.listener.on_data(EXAMPLE_TWEETS[0])
def setup(self):
self.listener = QueueListener()
send_listener(self.listener, EXAMPLE_TWEETS)
