def test_message_order(self): """test that we get messages in the order we expect""" queue = PriorityQueue() messages = [ message_format( ident=None, control={"priority" : 0, "expected-order" : 0}, body=None ), message_format( ident=None, control={"priority" : 5, "expected-order" : 2}, body=None ), message_format( ident=None, control={"priority" : 3, "expected-order" : 1}, body=None ), ] self.assertEqual(len(queue), 0) for message in messages: queue.append((message.control, message.body, )) self.assertEqual(len(queue), len(messages)) for order in range(len(messages)): retrieved_message, _retrieved_data = queue.popleft() self.assertEqual( order, retrieved_message["expected-order"] ) self.assertEqual(len(queue), 0) self.assertRaises(IndexError, queue.popleft)
def test_single_entry(self): """test adding and removing a single entry""" queue = PriorityQueue() message = message_format( ident=None, control={"priority" : 0}, body=None ) self.assertEqual(len(queue), 0) queue.append((message.control, message.body, )) self.assertEqual(len(queue), 1) retrieved_message, _retrieved_body = queue.popleft() self.assertEqual(retrieved_message, message.control) self.assertEqual(len(queue), 0) self.assertRaises(IndexError, queue.popleft)
def test_single_entry(self): """test adding and removing a single entry""" queue = PriorityQueue() message = message_format(ident=None, control={"priority": 0}, body=None) self.assertEqual(len(queue), 0) queue.append(( message.control, message.body, )) self.assertEqual(len(queue), 1) retrieved_message, _retrieved_body = queue.popleft() self.assertEqual(retrieved_message, message.control) self.assertEqual(len(queue), 0) self.assertRaises(IndexError, queue.popleft)
def test_message_order(self): """test that we get messages in the order we expect""" queue = PriorityQueue() messages = [ message_format(ident=None, control={ "priority": 0, "expected-order": 0 }, body=None), message_format(ident=None, control={ "priority": 5, "expected-order": 2 }, body=None), message_format(ident=None, control={ "priority": 3, "expected-order": 1 }, body=None), ] self.assertEqual(len(queue), 0) for message in messages: queue.append(( message.control, message.body, )) self.assertEqual(len(queue), len(messages)) for order in range(len(messages)): retrieved_message, _retrieved_data = queue.popleft() self.assertEqual(order, retrieved_message["expected-order"]) self.assertEqual(len(queue), 0) self.assertRaises(IndexError, queue.popleft)