def test_to_file_mailer_multifile(): tmp_dir = tempfile.mkdtemp() mailer = ToFileMailer(tmp_dir, multifile=True) email1, email2, email3, email4 = make_emails() assert mailer.send(email1) == 1 assert mailer.send(email2, email3) == 2 assert mailer.send(email4) == 1 assert len(os.listdir(tmp_dir)) == 3 shutil.rmtree(tmp_dir, True)
def test_to_file_mailer(): tmp_dir = tempfile.mkdtemp() mailer = ToFileMailer(tmp_dir) email1 = EmailMessage("Subject", "Content", "*****@*****.**", "*****@*****.**") assert mailer.send(email1) == 1 assert len(os.listdir(tmp_dir)) == 1 filepath = os.path.join(tmp_dir, os.listdir(tmp_dir)[0]) message = email.message_from_file(open(filepath)) assert message.get_content_type() == "text/plain" assert message.get("subject") == "Subject" assert message.get("from") == "*****@*****.**" assert message.get("to") == "*****@*****.**" shutil.rmtree(tmp_dir, True)
def test_to_file_mailer_unique_filename(): tmp_dir = tempfile.mkdtemp() mailer1 = ToFileMailer(tmp_dir) mailer2 = ToFileMailer(tmp_dir) email1 = EmailMessage("Subject", "Content", "*****@*****.**", "*****@*****.**") mailer1.send(email1) mailer2.send(email1) assert len(os.listdir(tmp_dir)) == 2 shutil.rmtree(tmp_dir, True)