def test_from_addr(self):
"""Function: test_from_addr
Description: Test with from address is passed.
Arguments:
"""
email = gen_class.Mail(self.toaddr, subj=self.subj4, frm=self.frm2)
self.assertEqual(email.frm, self.frm2)
def test_subj_list(self):
"""Function: test_subj_list
Description: Test subject as a list.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
email.create_subject(subj=self.subj4)
self.assertEqual(email.subj, self.subj4a)
def test_default(self):
"""Function: test_default
Description: Test __init__ method with default arguments.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
self.assertEqual((email.to, email.subj, email.host_name, email.host),
(self.toaddr, None, None, None))
def test_subj_list_delimiter(self):
"""Function: test_subj_list_delimiter
Description: Test with delimiter passed.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
email.create_subject(subj=self.subj4, delimiter="_")
self.assertEqual(email.subj, self.subj4b)
def test_to_list(self):
"""Function: test_to_list
Description: Test to line with a list.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
self.assertEqual((email.to, email.subj, email.host_name, email.host),
(self.toaddr, None, None, None))
def test_no_subject(self):
"""Function: test_no_subject
Description: Test creating subject.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
email.create_subject(subj=self.subj)
self.assertEqual(email.subj, self.subj)
def test_send_mailx(self, mock_subp):
"""Function: test_send_mailx
Description: Test send_mailx function.
Arguments:
"""
mail = gen_class.Mail(self.toaddr, subj=self.subj, frm=self.frm)
mock_subp.side_effect = [self.subp, self.subp2]
self.assertFalse(mail.send_mailx())
def test_subject_none(self):
"""Function: test_subject_none
Description: Test with no subject passed.
Arguments:
"""
email = gen_class.Mail(self.toaddr, subj=self.subj)
email.create_subject(subj=None)
self.assertEqual(email.subj, self.subj)
def test_initial_add(self):
"""Function: test_initial_add
Description: Test with adding data to empty message.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
email.add_2_msg(self.msg)
self.assertEqual((email.to, email.msg), (self.toaddr, self.msg))
def test_subj_overwrite(self):
"""Function: test_subj_overwrite
Description: Test overwriting existing subject.
Arguments:
"""
email = gen_class.Mail(self.toaddr, subj=self.subj)
email.create_subject(subj=self.subj2)
self.assertEqual(email.subj, self.subj2)
def test_subj_string(self):
"""Function: test_subj_string
Description: Test subject as a string.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
email.create_subject(subj=self.subj3)
self.assertEqual(email.subj, self.subj3)
def test_empty_test(self):
"""Function: test_empty_test
Description: Test with empty text line.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
email.add_2_msg(self.msg)
email.add_2_msg()
self.assertEqual((email.to, email.msg), (self.toaddr, self.msg))
def test_with_data(self):
"""Function: test_with_data
Description: Test other attributes with data.
Arguments:
"""
email = gen_class.Mail(self.toaddr, subj=self.subj2, frm=self.frm2)
self.assertEqual(
(email.to, email.subj, email.frm, email.host_name, email.host),
(email.to, self.subj2, self.frm2, None, None))
def test_send_mail(self, mock_smtp, mock_body):
"""Function: test_send_mail
Description: Test test_send_mail function.
Arguments:
"""
mock_smtp.return_value = Smtplib()
mock_body.return_value = True
email = gen_class.Mail(self.toaddr)
self.assertFalse(email.send_mail())
def test_to_string(self):
"""Function: test_to_string
Description: Test to line with a string.
Arguments:
"""
email = gen_class.Mail(self.toaddr)
self.assertEqual(
(self.mailaddr, email.subj, email.host_name, email.host),
(self.mailaddr, None, None, None))
def test_subj_list(self):
"""Function: test_subj_string
Description: Test subject line as a string.
Arguments:
"""
email = gen_class.Mail(self.toaddr, subj=self.subj4, frm=self.frm2)
self.assertEqual(
(email.to, email.subj, email.frm, email.host_name, email.host),
(email.to, self.subj4a, self.frm2, None, None))
def test_create_body(self):
"""Function: test_create_body
Description: Test creating body.
Arguments:
"""
msg = "Test email line"
email = gen_class.Mail(self.toaddr)
email.add_2_msg(msg)
email.create_body()
self.assertEqual(email.create_body(), (self.subj_mask % (msg, msg)))
def test_to_list(self, mock_subp):
"""Function: test_to_list
Description: Test to address as a list.
Arguments:
"""
mail = gen_class.Mail(self.toaddr2, subj=self.subj, frm=self.frm)
mock_subp.side_effect = [self.subp, self.subp2]
mail.send_mailx()
self.assertEqual((mail.subj, mail.to), (self.subj, self.toaddr))
def test_add_exist(self):
"""Function: test_add_exist
Description: Test with adding data to existing message.
Arguments:
"""
msg2 = "Test email line2"
email = gen_class.Mail(self.toaddr)
email.add_2_msg(self.msg)
email.add_2_msg(msg2)
self.assertEqual((email.to, email.msg), (self.toaddr, self.msg + msg2))
def test_subj_spaces(self, mock_subp):
"""Function: test_subj_spaces
Description: Test subject line with white spaces.
Arguments:
"""
mail = gen_class.Mail(self.toaddr, subj=self.subj2, frm=self.frm)
mock_subp.side_effect = [self.subp, self.subp2]
mail.send_mailx()
self.assertEqual((mail.subj, mail.to), (self.subj2a, self.toaddr))
def test_send_mailx2(self, mock_subp):
"""Function: test_send_mailx2
Description: Test send_mailx function.
Arguments:
"""
mail = gen_class.Mail(self.toaddr, subj=self.subj, frm=self.frm)
mock_subp.side_effect = [self.subp, self.subp2]
mail.send_mailx()
self.assertEqual((mail.subj, mail.to), (self.subj, self.toaddr))
def test_subj_str(self, mock_subp):
"""Function: test_subj_str
Description: Test subject line as a string.
Arguments:
"""
mail = gen_class.Mail(self.toaddr, subj=self.subj, frm=self.frm)
mock_subp.side_effect = [self.subp, self.subp2]
mail.send_mail(use_mailx=True)
self.assertEqual((mail.subj, mail.to), (self.subj, self.toaddr))
def test_print_email(self):
"""Function: test_print_email
Description: Test printing email.
Arguments:
"""
email = gen_class.Mail(self.toaddr, subj=self.subj, frm=self.frm)
email.create_subject(subj=self.subj)
email.add_2_msg(txt_ln=self.body)
self.assertEqual(
email.print_email(), "To: %s\nFrom: %s\n%s" %
(self.toaddr, self.frm, email.create_body()))
def test_long_subject(self):
"""Function: test_long_subject
Description: Test with subject line to maximum.
Arguments:
"""
msg = "Test email line with a line greater than thirty characters"
email = gen_class.Mail(self.toaddr)
email.add_2_msg(msg)
email.create_body()
self.assertEqual(email.create_body(),
(self.subj_mask % (msg[:30], msg)))
def test_with_subject(self):
"""Function: test_with_subject
Description: Test creating body with subject line in place.
Arguments:
"""
msg = "Test email line"
subj = "Test subject"
email = gen_class.Mail(self.toaddr, subj=subj)
email.add_2_msg(msg)
email.create_body()
self.assertEqual(email.create_body(), (self.subj_mask % (subj, msg)))
def setUp(self):
"""Function: setUp
Description: Initialization for unit testing.
Arguments:
"""
self.data = {"key1": "value1", "key2": {"key3": "value2"}}
self.data2 = {"key1": "value1", "key2": "value2"}
self.ofile = "test/integration/gen_libs/tmp/test_print_dict.txt"
self.basefile = "test/integration/gen_libs/basefiles/print_dict.txt"
self.mail = gen_class.Mail("to_address", "subject_line")
self.msg = '{\n "key2": "value2", \n "key1": "value1"\n}'
self.msg2 = '{"key2": "value2", "key1": "value1"}'
def test_non_string(self):
"""Function: test_non_string
Description: Test with a list argument.
Arguments:
"""
msg2 = ["Test ", "email ", "line2"]
email = gen_class.Mail(self.toaddr)
email.add_2_msg(self.msg)
email.add_2_msg(msg2)
self.assertEqual(
(email.to, email.msg),
(self.toaddr, self.msg + '["Test ", "email ", "line2"]'))
def setUp(self):
"""Function: setUp
Description: Initialization for unit testing.
Arguments:
"""
self.data = {}
self.data2 = {"key1": "value1"}
self.ofile = "Out_File"
self.json_fmt = True
self.no_std = True
self.mail = gen_class.Mail("to_address", "subject_line")
self.msg = '{\n "key1": "value1"\n}'
self.msg2 = '{"key1": "value1"}'
def test_non_string2(self):
"""Function: test_non_string2
Description: Test with a dictionary argument.
Arguments:
"""
msg2 = {1: "Test ", 2: "email ", 3: "line2"}
email = gen_class.Mail(self.toaddr)
email.add_2_msg(self.msg)
email.add_2_msg(msg2)
self.assertEqual(
(email.to, email.msg),
(self.toaddr,
self.msg + '{"1": "Test ", "2": "email ", "3": "line2"}'))
