86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
|
from app.extensions import db
|
||
|
|
from app.greylisting import (
|
||
|
|
greylisting_needed_forward_phase,
|
||
|
|
greylisting_needed_for_alias,
|
||
|
|
greylisting_needed_for_mailbox,
|
||
|
|
greylisting_needed_reply_phase,
|
||
|
|
)
|
||
|
|
from app.models import User, Alias, EmailLog, Contact
|
||
|
|
|
||
|
|
|
||
|
|
def test_greylisting_needed_forward_phase_for_alias(flask_client):
|
||
|
|
user = User.create(
|
||
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
||
|
|
)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
# no greylisting for a new alias
|
||
|
|
alias = Alias.create_new_random(user)
|
||
|
|
db.session.commit()
|
||
|
|
assert not greylisting_needed_for_alias(alias)
|
||
|
|
|
||
|
|
# greylisting when there's a previous activity on alias
|
||
|
|
contact = Contact.create(
|
||
|
|
user_id=user.id,
|
||
|
|
alias_id=alias.id,
|
||
|
|
website_email="contact@example.com",
|
||
|
|
reply_email="rep@sl.local",
|
||
|
|
)
|
||
|
|
db.session.commit()
|
||
|
|
EmailLog.create(user_id=user.id, contact_id=contact.id)
|
||
|
|
assert greylisting_needed_for_alias(alias)
|
||
|
|
|
||
|
|
|
||
|
|
def test_greylisting_needed_forward_phase_for_mailbox(flask_client):
|
||
|
|
user = User.create(
|
||
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
||
|
|
)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
alias = Alias.create_new_random(user)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
contact = Contact.create(
|
||
|
|
user_id=user.id,
|
||
|
|
alias_id=alias.id,
|
||
|
|
website_email="contact@example.com",
|
||
|
|
reply_email="rep@sl.local",
|
||
|
|
)
|
||
|
|
db.session.commit()
|
||
|
|
EmailLog.create(user_id=user.id, contact_id=contact.id)
|
||
|
|
|
||
|
|
# Create another alias with the same mailbox
|
||
|
|
# will be greylisted as there's a previous activity on mailbox
|
||
|
|
alias = Alias.create_new_random(user)
|
||
|
|
db.session.commit()
|
||
|
|
assert greylisting_needed_for_mailbox(alias)
|
||
|
|
|
||
|
|
|
||
|
|
def test_greylisting_needed_forward_phase(flask_client):
|
||
|
|
# no greylisting when alias not exist
|
||
|
|
assert not greylisting_needed_forward_phase("not-exist@alias.com")
|
||
|
|
|
||
|
|
|
||
|
|
def test_greylisting_needed_reply_phase(flask_client):
|
||
|
|
# no greylisting when reply_email not exist
|
||
|
|
assert not greylisting_needed_reply_phase("not-exist-reply@alias.com")
|
||
|
|
|
||
|
|
user = User.create(
|
||
|
|
email="a@b.c", password="password", name="Test User", activated=True
|
||
|
|
)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
alias = Alias.create_new_random(user)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
contact = Contact.create(
|
||
|
|
user_id=user.id,
|
||
|
|
alias_id=alias.id,
|
||
|
|
website_email="contact@example.com",
|
||
|
|
reply_email="rep@sl.local",
|
||
|
|
)
|
||
|
|
db.session.commit()
|
||
|
|
EmailLog.create(user_id=user.id, contact_id=contact.id)
|
||
|
|
|
||
|
|
assert greylisting_needed_reply_phase("rep@sl.local")
|