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)
