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)
