def test_empty_msg(self):
"""Function: test_empty_msg
Description: Test empty message is returned.
Arguments:
"""
self.assertEqual(mail_2_rmq.get_text(self.msg), "")
def test_single_part_msg(self):
"""Function: test_single_part_msg
Description: Test single part message is returned.
Arguments:
"""
self.msg.walk_list = [self.part_msg]
self.part_msg.type = "single"
self.part_msg.payload = self.email_msg
self.assertEqual(mail_2_rmq.get_text(self.msg), self.email_msg)
def test_multi_part_msg(self):
"""Function: test_multi_part_msg
Description: Test multi-part message is returned.
Arguments:
"""
self.msg.walk_list = [self.part_msg, self.part_msg2]
self.part_msg.type = "multipart"
self.part_msg2.type = "single"
self.part_msg2.payload = "Email"
self.assertEqual(mail_2_rmq.get_text(self.msg), "Email")
def test_empty_payload_msg(self):
"""Function: test_empty_payload_msg
Description: Test when payload returns empty.
Arguments:
"""
self.msg.walk_list = [self.part_msg, self.part_msg2]
self.part_msg.type = "single"
self.part_msg.payload = ""
self.part_msg2.type = "single"
self.part_msg2.payload = "Email"
self.assertEqual(mail_2_rmq.get_text(self.msg), "Email")
def test_multipart_empty_payload_msg(self):
"""Function: test_multipart_empty_payload_msg
Description: Test multipart and empty payload msg.
Arguments:
"""
self.msg.walk_list = [self.part_msg, self.part_msg2, self.part_msg3]
self.part_msg.type = "multipart"
self.part_msg2.type = "single"
self.part_msg2.payload = ""
self.part_msg3.type = "single"
self.part_msg3.payload = "Email"
self.assertEqual(mail_2_rmq.get_text(self.msg), "Email")
def test_multiple_part_msg(self):
"""Function: test_multiple_part_msg
Description: Test returning multiple part messages.
Arguments:
"""
self.msg.walk_list = [self.part_msg, self.part_msg2, self.part_msg3,
self.part_msg4]
self.part_msg.type = "multipart"
self.part_msg2.type = "single"
self.part_msg2.payload = "Email"
self.part_msg3.type = "single"
self.part_msg3.payload = ""
self.part_msg4.type = "single"
self.part_msg4.payload = " Message"
self.assertEqual(mail_2_rmq.get_text(self.msg), self.email_msg)