1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
"""Shared fixtures for EML extraction tests."""
import os
from email.message import EmailMessage
from email.mime.application import MIMEApplication
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
import pytest
@pytest.fixture
def fixtures_dir():
"""Return path to the fixtures/ directory."""
return os.path.join(os.path.dirname(__file__), 'fixtures')
def make_plain_message(body="Test body", from_="Jonathan Smith <jsmith@example.com>",
to="Craig <craig@example.com>",
subject="Test Subject",
date="Wed, 05 Feb 2026 11:36:00 -0600"):
"""Create an EmailMessage with text/plain body."""
msg = EmailMessage()
msg['From'] = from_
msg['To'] = to
msg['Subject'] = subject
msg['Date'] = date
msg.set_content(body)
return msg
def make_html_message(html_body="<p>Test body</p>",
from_="Jonathan Smith <jsmith@example.com>",
to="Craig <craig@example.com>",
subject="Test Subject",
date="Wed, 05 Feb 2026 11:36:00 -0600"):
"""Create an EmailMessage with text/html body only."""
msg = EmailMessage()
msg['From'] = from_
msg['To'] = to
msg['Subject'] = subject
msg['Date'] = date
msg.set_content(html_body, subtype='html')
return msg
def make_message_with_attachment(body="Test body",
from_="Jonathan Smith <jsmith@example.com>",
to="Craig <craig@example.com>",
subject="Test Subject",
date="Wed, 05 Feb 2026 11:36:00 -0600",
attachment_filename="document.pdf",
attachment_content=b"fake pdf content"):
"""Create a multipart message with a text body and one attachment."""
msg = MIMEMultipart()
msg['From'] = from_
msg['To'] = to
msg['Subject'] = subject
msg['Date'] = date
msg.attach(MIMEText(body, 'plain'))
att = MIMEApplication(attachment_content, Name=attachment_filename)
att['Content-Disposition'] = f'attachment; filename="{attachment_filename}"'
msg.attach(att)
return msg
def add_received_headers(msg, headers):
"""Add Received headers to an existing message.
headers: list of header strings, added in order (first = most recent).
"""
for header in headers:
msg['Received'] = header
return msg
|
