Fixed tests

This commit is contained in:
Adrià Casajús 2022-03-30 16:29:38 +02:00
parent 451e69a3c4
commit c9a15f4921
No known key found for this signature in database
GPG Key ID: F0033226A5AFC9B9
1 changed files with 14 additions and 10 deletions

View File

@ -11,7 +11,8 @@ from app.models import (
AuthorizedAddress, AuthorizedAddress,
IgnoredEmail, IgnoredEmail,
EmailLog, EmailLog,
Notification, VerpType, Notification,
VerpType,
) )
from email_handler import ( from email_handler import (
get_mailbox_from_mail_from, get_mailbox_from_mail_from,
@ -127,10 +128,11 @@ 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 = generate_verp_email(VerpType.bounce_forward, 99999999999999) envelope.mail_from = msg["from"]
envelope.rcpt_tos = [msg["to"]] #Ensure invalid email log
envelope.rcpt_tos = [generate_verp_email(VerpType.bounce_forward, 99999999999999)]
result = email_handler.MailHandler()._handle(envelope, msg) result = email_handler.MailHandler()._handle(envelope, msg)
assert result == status.E216 assert status.E216 == result
def test_preserve_5xx_with_valid_spf(flask_client): def test_preserve_5xx_with_valid_spf(flask_client):
@ -141,10 +143,11 @@ 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 = generate_verp_email(VerpType.bounce_forward, 99999999999999) envelope.mail_from = msg["from"]
envelope.rcpt_tos = [msg["to"]] #Ensure invalid email log
envelope.rcpt_tos = [generate_verp_email(VerpType.bounce_forward, 99999999999999)]
result = email_handler.MailHandler()._handle(envelope, msg) result = email_handler.MailHandler()._handle(envelope, msg)
assert result == status.E512 assert status.E512 == result
def test_preserve_5xx_with_no_header(flask_client): def test_preserve_5xx_with_no_header(flask_client):
@ -155,7 +158,8 @@ def test_preserve_5xx_with_no_header(flask_client):
{"alias_email": alias.email}, {"alias_email": alias.email},
) )
envelope = Envelope() envelope = Envelope()
envelope.mail_from = generate_verp_email(VerpType.bounce_forward, 99999999999999) envelope.mail_from = msg["from"]
envelope.rcpt_tos = [msg["to"]] #Ensure invalid email log
envelope.rcpt_tos = [generate_verp_email(VerpType.bounce_forward, 99999999999999)]
result = email_handler.MailHandler()._handle(envelope, msg) result = email_handler.MailHandler()._handle(envelope, msg)
assert result == status.E512 assert status.E512 == result