73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
from rspamd_mail_trainer.cli import SearchOptions, iter_candidates, maybe_compile, parse_user_date
|
|
|
|
|
|
def write_mail(path: Path, sender: str, subject: str, body: str = "hello") -> None:
|
|
path.write_text(
|
|
"\n".join([
|
|
f"From: {sender}",
|
|
"To: johannes@example.org",
|
|
f"Subject: {subject}",
|
|
"Date: Fri, 12 Jun 2026 13:36:00 +0000",
|
|
"Message-ID: <test@example.org>",
|
|
"",
|
|
body,
|
|
]),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
|
|
class CliTests(unittest.TestCase):
|
|
def test_maildir_sender_match(self) -> None:
|
|
with tempfile.TemporaryDirectory() as td:
|
|
root = Path(td) / "Maildir"
|
|
cur = root / "cur"
|
|
cur.mkdir(parents=True)
|
|
write_mail(cur / "1:2,S", "no-reply@example.test", "Monthly report")
|
|
opts = SearchOptions(
|
|
from_re=maybe_compile("example.test", True, True),
|
|
to_re=None,
|
|
subject_re=None,
|
|
any_re=None,
|
|
after=None,
|
|
before=None,
|
|
max_size=1024 * 1024,
|
|
include_seen_duplicates=False,
|
|
)
|
|
matches = list(iter_candidates([root], opts))
|
|
self.assertEqual(len(matches), 1)
|
|
self.assertEqual(matches[0][0].sender, "no-reply@example.test")
|
|
|
|
def test_subject_match(self) -> None:
|
|
with tempfile.TemporaryDirectory() as td:
|
|
root = Path(td) / "Maildir"
|
|
cur = root / "cur"
|
|
cur.mkdir(parents=True)
|
|
write_mail(cur / "2:2,S", "sender@example.org", "Invoice 123")
|
|
opts = SearchOptions(
|
|
from_re=None,
|
|
to_re=None,
|
|
subject_re=maybe_compile("Invoice", True, True),
|
|
any_re=None,
|
|
after=None,
|
|
before=None,
|
|
max_size=1024 * 1024,
|
|
include_seen_duplicates=False,
|
|
)
|
|
matches = list(iter_candidates([root], opts))
|
|
self.assertEqual(len(matches), 1)
|
|
|
|
def test_parse_user_date_day_start(self) -> None:
|
|
parsed = parse_user_date("2026-06-13")
|
|
self.assertIsNotNone(parsed)
|
|
assert parsed is not None
|
|
self.assertEqual(parsed.hour, 0)
|
|
self.assertEqual(parsed.minute, 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|