mirror of
https://github.com/chatmail/relay.git
synced 2026-05-10 16:04:37 +00:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
import smtplib
|
|
import pytest
|
|
|
|
|
|
def test_remote(remote, imap_or_smtp):
|
|
lineproducer = remote.iter_output(imap_or_smtp.logcmd)
|
|
imap_or_smtp.connect()
|
|
assert imap_or_smtp.name in next(lineproducer)
|
|
|
|
|
|
def test_use_two_chatmailservers(cmfactory, maildomain2):
|
|
ac1 = cmfactory.new_online_configuring_account(cache=False)
|
|
cmfactory.switch_maildomain(maildomain2)
|
|
ac2 = cmfactory.new_online_configuring_account(cache=False)
|
|
cmfactory.bring_accounts_online()
|
|
cmfactory.get_accepted_chat(ac1, ac2)
|
|
domain1 = ac1.get_config("addr").split("@")[1]
|
|
domain2 = ac2.get_config("addr").split("@")[1]
|
|
assert domain1 != domain2
|
|
|
|
|
|
@pytest.mark.parametrize("forgeaddr", ["internal", "someone@example.org"])
|
|
def test_reject_forged_from(cmsetup, maildata, lp, forgeaddr):
|
|
user1, user3 = cmsetup.gen_users(2)
|
|
|
|
lp.sec("send encrypted message with forged from")
|
|
print("envelope_from", user1.addr)
|
|
if forgeaddr == "internal":
|
|
addr_to_forge = cmsetup.gen_users(1)[0].addr
|
|
else:
|
|
addr_to_forge = "someone@example.org"
|
|
|
|
print("message to inject:")
|
|
msg = maildata("encrypted.eml", from_addr=addr_to_forge, to_addr=user3.addr)
|
|
msg = msg.as_string()
|
|
for line in msg.split("\n")[:4]:
|
|
print(f" {line}")
|
|
|
|
lp.sec("Send forged mail and check remote postfix lmtp processing result")
|
|
with pytest.raises(smtplib.SMTPException) as e:
|
|
user1.smtp.sendmail(from_addr=user1.addr, to_addrs=[user3.addr], msg=msg)
|
|
assert "500" in str(e.value)
|
|
|
|
|
|
@pytest.mark.slow
|
|
def test_exceed_rate_limit(cmsetup, gencreds, maildata):
|
|
"""Test that the per-account send-mail limit is exceeded."""
|
|
user1, user2 = cmsetup.gen_users(2)
|
|
mail = maildata(
|
|
"encrypted.eml", from_addr=user1.addr, to_addr=user2.addr
|
|
).as_string()
|
|
for i in range(100):
|
|
print("Sending mail", str(i))
|
|
try:
|
|
user1.smtp.sendmail(user1.addr, [user2.addr], mail)
|
|
except smtplib.SMTPException as e:
|
|
if i < 80:
|
|
pytest.fail(f"rate limit was exceeded too early with msg {i}")
|
|
outcome = e.recipients[user2.addr]
|
|
assert outcome[0] == 450
|
|
assert b"4.7.1: Too much mail from" in outcome[1]
|
|
return
|
|
pytest.fail("Rate limit was not exceeded")
|