mirror of
https://github.com/chatmail/relay.git
synced 2026-05-10 16:04:37 +00:00
61 lines
2.2 KiB
Python
61 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, mailgen, 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 = mailgen.get_encrypted(from_addr=addr_to_forge, to_addr=user3.addr)
|
|
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, mailgen):
|
|
"""Test that the per-account send-mail limit is exceeded."""
|
|
user1, user2 = cmsetup.gen_users(2)
|
|
mail = mailgen.get_encrypted(user1.addr, user2.addr)
|
|
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")
|