75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
from aiosmtpd.smtp import Envelope
|
|
|
|
import email_handler
|
|
from app.db import Session
|
|
from app.email import headers, status
|
|
from app.mail_sender import mail_sender
|
|
from app.models import Alias
|
|
from app.utils import random_string
|
|
from tests.utils import create_new_user, load_eml_file, random_email
|
|
|
|
|
|
@mail_sender.store_emails_test_decorator
|
|
def test_original_headers_from_preserved():
|
|
user = create_new_user()
|
|
alias = Alias.create_new_random(user)
|
|
Session.flush()
|
|
assert user.include_header_email_header
|
|
original_sender_address = random_email()
|
|
msg = load_eml_file(
|
|
"replacement_on_forward_phase.eml",
|
|
{
|
|
"sender_address": original_sender_address,
|
|
"recipient_address": alias.email,
|
|
"cc_address": random_email(),
|
|
},
|
|
)
|
|
envelope = Envelope()
|
|
envelope.mail_from = f"env.{original_sender_address}"
|
|
envelope.rcpt_tos = [alias.email]
|
|
result = email_handler.MailHandler()._handle(envelope, msg)
|
|
assert result == status.E200
|
|
send_requests = mail_sender.get_stored_emails()
|
|
assert len(send_requests) == 1
|
|
request = send_requests[0]
|
|
assert request.msg[headers.SL_ENVELOPE_FROM] == envelope.mail_from
|
|
assert request.msg[headers.SL_ORIGINAL_FROM] == original_sender_address
|
|
assert (
|
|
request.msg[headers.AUTHENTICATION_RESULTS]
|
|
== msg[headers.AUTHENTICATION_RESULTS]
|
|
)
|
|
|
|
|
|
@mail_sender.store_emails_test_decorator
|
|
def test_original_headers_from_with_name_preserved():
|
|
user = create_new_user()
|
|
alias = Alias.create_new_random(user)
|
|
Session.flush()
|
|
assert user.include_header_email_header
|
|
original_sender_address = random_email()
|
|
name = random_string(10)
|
|
msg = load_eml_file(
|
|
"replacement_on_forward_phase.eml",
|
|
{
|
|
"sender_address": f"{name} <{original_sender_address}>",
|
|
"recipient_address": alias.email,
|
|
"cc_address": random_email(),
|
|
},
|
|
)
|
|
envelope = Envelope()
|
|
envelope.mail_from = f"env.{original_sender_address}"
|
|
envelope.rcpt_tos = [alias.email]
|
|
result = email_handler.MailHandler()._handle(envelope, msg)
|
|
assert result == status.E200
|
|
send_requests = mail_sender.get_stored_emails()
|
|
assert len(send_requests) == 1
|
|
request = send_requests[0]
|
|
assert request.msg[headers.SL_ENVELOPE_FROM] == envelope.mail_from
|
|
assert (
|
|
request.msg[headers.SL_ORIGINAL_FROM] == f"{name} <{original_sender_address}>"
|
|
)
|
|
assert (
|
|
request.msg[headers.AUTHENTICATION_RESULTS]
|
|
== msg[headers.AUTHENTICATION_RESULTS]
|
|
)
|