More rebase fixes

This commit is contained in:
Adrià Casajús 2022-03-30 16:09:17 +02:00
parent dce9e633bf
commit 451e69a3c4
No known key found for this signature in database
GPG Key ID: F0033226A5AFC9B9
3 changed files with 8 additions and 7 deletions

View File

@ -1474,7 +1474,7 @@ def get_spamd_result(msg: Message) -> Optional[SpamdResult]:
def generate_verp_email( def generate_verp_email(
verp_type: VerpType, object_id: int, sender_domain: Optional[str] verp_type: VerpType, object_id: int, sender_domain: Optional[str] = None
) -> str: ) -> str:
# Encoded as a list to minimize size of email address # Encoded as a list to minimize size of email address
data = [verp_type.bounce_forward.value, object_id, int(time.time())] data = [verp_type.bounce_forward.value, object_id, int(time.time())]

View File

@ -157,7 +157,8 @@ from app.models import (
DomainDeletedAlias, DomainDeletedAlias,
Notification, Notification,
DmarcCheckResult, DmarcCheckResult,
SPFCheckResult, VerpType, SPFCheckResult,
VerpType,
) )
from app.pgp_utils import PGPException, sign_data_with_pgpy, sign_data from app.pgp_utils import PGPException, sign_data_with_pgpy, sign_data
from app.utils import sanitize_email from app.utils import sanitize_email

View File

@ -3,15 +3,15 @@ from email.message import EmailMessage
from aiosmtpd.smtp import Envelope from aiosmtpd.smtp import Envelope
import email_handler import email_handler
from app.config import BOUNCE_EMAIL
from app.email import headers, status from app.email import headers, status
from app.email_utils import generate_verp_email
from app.models import ( from app.models import (
User, User,
Alias, Alias,
AuthorizedAddress, AuthorizedAddress,
IgnoredEmail, IgnoredEmail,
EmailLog, EmailLog,
Notification, Notification, VerpType,
) )
from email_handler import ( from email_handler import (
get_mailbox_from_mail_from, get_mailbox_from_mail_from,
@ -127,7 +127,7 @@ def test_prevent_5xx_from_spf(flask_client):
{"alias_email": alias.email, "spf_result": "R_SPF_FAIL"}, {"alias_email": alias.email, "spf_result": "R_SPF_FAIL"},
) )
envelope = Envelope() envelope = Envelope()
envelope.mail_from = BOUNCE_EMAIL.format(999999999999999999) envelope.mail_from = generate_verp_email(VerpType.bounce_forward, 99999999999999)
envelope.rcpt_tos = [msg["to"]] envelope.rcpt_tos = [msg["to"]]
result = email_handler.MailHandler()._handle(envelope, msg) result = email_handler.MailHandler()._handle(envelope, msg)
assert result == status.E216 assert result == status.E216
@ -141,7 +141,7 @@ def test_preserve_5xx_with_valid_spf(flask_client):
{"alias_email": alias.email, "spf_result": "R_SPF_ALLOW"}, {"alias_email": alias.email, "spf_result": "R_SPF_ALLOW"},
) )
envelope = Envelope() envelope = Envelope()
envelope.mail_from = BOUNCE_EMAIL.format(999999999999999999) envelope.mail_from = generate_verp_email(VerpType.bounce_forward, 99999999999999)
envelope.rcpt_tos = [msg["to"]] envelope.rcpt_tos = [msg["to"]]
result = email_handler.MailHandler()._handle(envelope, msg) result = email_handler.MailHandler()._handle(envelope, msg)
assert result == status.E512 assert result == status.E512
@ -155,7 +155,7 @@ def test_preserve_5xx_with_no_header(flask_client):
{"alias_email": alias.email}, {"alias_email": alias.email},
) )
envelope = Envelope() envelope = Envelope()
envelope.mail_from = BOUNCE_EMAIL.format(999999999999999999) envelope.mail_from = generate_verp_email(VerpType.bounce_forward, 99999999999999)
envelope.rcpt_tos = [msg["to"]] envelope.rcpt_tos = [msg["to"]]
result = email_handler.MailHandler()._handle(envelope, msg) result = email_handler.MailHandler()._handle(envelope, msg)
assert result == status.E512 assert result == status.E512