def test_missing_severity(self):
"""Assert the default severity is INFO if it's not in the headers."""
msg = message.Message(severity=message.ERROR)
del msg._headers["fedora_messaging_severity"]
recv_msg = message.get_message("", msg._properties, b"{}")
self.assertEqual(recv_msg.severity, message.INFO)
def test_missing_headers(self):
"""Assert missing headers results in a default message."""
msg = message.Message()
msg._headers = None
received_msg = message.get_message(msg._encoded_routing_key,
msg._properties, msg._encoded_body)
self.assertIsInstance(received_msg, message.Message)