mirror of
https://github.com/chatmail/relay.git
synced 2026-08-27 02:23:14 +00:00
holger krekel
fa5afadaaa
The default values are geared towards minimal-requirements server. If you have a big server, you will need to set chatmail.ini parameters.
107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
import importlib.resources
|
|
import itertools
|
|
import os
|
|
import random
|
|
from email import policy
|
|
from email.parser import BytesParser
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from chatmaild.config import read_config, write_initial_config
|
|
|
|
|
|
@pytest.fixture
|
|
def make_config(tmp_path):
|
|
inipath = tmp_path.joinpath("chatmail.ini")
|
|
|
|
def make_conf(mail_domain, settings=None):
|
|
basedir = tmp_path.joinpath(f"vmail/{mail_domain}")
|
|
basedir.mkdir(parents=True, exist_ok=True)
|
|
overrides = settings.copy() if settings else {}
|
|
overrides["mailboxes_dir"] = str(basedir)
|
|
# permissive resource limits so tests never depend on host load/memory/disk
|
|
overrides.setdefault("max_load_1m", "99999")
|
|
overrides.setdefault("min_available_memory", "0")
|
|
overrides.setdefault("min_free_disk_space", "0")
|
|
write_initial_config(inipath, mail_domain, overrides=overrides)
|
|
return read_config(inipath)
|
|
|
|
return make_conf
|
|
|
|
|
|
@pytest.fixture
|
|
def example_config(make_config):
|
|
return make_config("chat.example.org")
|
|
|
|
|
|
@pytest.fixture
|
|
def ipv4_config(make_config):
|
|
return make_config("1.3.3.7")
|
|
|
|
|
|
@pytest.fixture
|
|
def maildomain(example_config):
|
|
return example_config.mail_domain
|
|
|
|
|
|
@pytest.fixture
|
|
def testaddr(maildomain):
|
|
return f"user.name@{maildomain}"
|
|
|
|
|
|
@pytest.fixture
|
|
def gencreds(maildomain):
|
|
count = itertools.count()
|
|
next(count)
|
|
|
|
def gen(domain=None):
|
|
domain = domain if domain else maildomain
|
|
while 1:
|
|
num = next(count)
|
|
alphanumeric = "abcdefghijklmnopqrstuvwxyz1234567890"
|
|
user = "".join(random.choices(alphanumeric, k=10))
|
|
user = f"ac{num}_{user}"[:9]
|
|
password = "".join(random.choices(alphanumeric, k=12))
|
|
yield f"{user}@{domain}", f"{password}"
|
|
|
|
return lambda domain=None: next(gen(domain))
|
|
|
|
|
|
@pytest.fixture
|
|
def maildata(request):
|
|
try:
|
|
datadir = importlib.resources.files(__package__).joinpath("mail-data")
|
|
except TypeError:
|
|
# in python3.9 or lower, the above doesn't work, so we get datadir this way:
|
|
datadir = Path(os.getcwd()).joinpath("chatmaild/src/chatmaild/tests/mail-data")
|
|
|
|
assert datadir.exists(), datadir
|
|
|
|
def maildata(name, from_addr, to_addr, subject="[...]"):
|
|
# Using `.read_bytes().decode()` instead of `.read_text()` to preserve newlines.
|
|
data = datadir.joinpath(name).read_bytes().decode()
|
|
text = data.format(from_addr=from_addr, to_addr=to_addr, subject=subject)
|
|
return BytesParser(policy=policy.SMTP).parsebytes(text.encode())
|
|
|
|
return maildata
|
|
|
|
|
|
@pytest.fixture
|
|
def mockout():
|
|
class MockOut:
|
|
captured_red = []
|
|
captured_green = []
|
|
captured_plain = []
|
|
|
|
def red(self, msg):
|
|
self.captured_red.append(msg)
|
|
|
|
def green(self, msg):
|
|
self.captured_green.append(msg)
|
|
|
|
def __call__(self, msg):
|
|
self.captured_plain.append(msg)
|
|
|
|
return MockOut()
|