def test_get_next_message_id_None(self): "Testing get_next when not specifying a message" with patch('time.time', self._time): mq = _MessageQueue(5) for x in range(15): mq.append('channel', {'message_id': x}) channel, content = mq.get_next({'channel'}) self.assertEqual(content, {'message_id': 10})
def test_get_next_timeout_expired(self): "Testing get_next when requesting an outdated message" with patch('time.time', self._time): mq = _MessageQueue(5) for x in range(15): mq.append('channel', {'message_id': x}) channel, content = mq.get_next({'channel'}, 0) self.assertEqual(content, {'message_id': 10})
def test_get_next(self): "Testing the basic functionality of get_next" with patch('time.time', self._time): mq = _MessageQueue(5) for x in range(15): mq.append('channel', {'message_id': x}) channel, content = mq.get_next({'channel'}, 11) self.assertEqual(content, {'message_id': 12})
def test_get_next_channels(self): "Testing get_next with multiple channels" with patch('time.time', self._time): mq = _MessageQueue(5) for x in range(15): mq.append('odd' if x % 2 else 'even', {'message_id': x}) channel, content = mq.get_next({'odd'}, 11) self.assertEqual(content, {'message_id': 13}) self.assertEqual(channel, 'odd')