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')
