Skip to content

Commit b2b23d7

Browse files
committed
In SMTP, append envelope details to each message. Fixes #20.
1 parent 908f070 commit b2b23d7

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

pytest_localserver/smtp.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,17 @@ def process_message(self, peer, mailfrom, rcpttos, data, **kwargs):
4848
Adds message to outbox.
4949
"""
5050
try:
51-
self.outbox += [email.message_from_bytes(data)]
51+
message = email.message_from_bytes(data)
5252
except AttributeError:
53-
self.outbox += [email.message_from_string(data)]
53+
message = email.message_from_string(data)
54+
# on the message, also set the envelope details
55+
message.details = dict(
56+
peer=peer,
57+
mailfrom=mailfrom,
58+
rcpttos=rcpttos,
59+
**kwargs
60+
)
61+
self.outbox.append(message)
5462

5563
def run(self):
5664
"""

0 commit comments

Comments
 (0)