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)