28 lines
740 B
Python
28 lines
740 B
Python
from aiosmtpd.controller import Controller
|
|
|
|
|
|
class MockSMTPServer:
|
|
def __init__(self, hostname='127.0.0.1', port=8025):
|
|
self.messages = []
|
|
self.controller = Controller(MockHandler(self), hostname, port)
|
|
|
|
def start(self):
|
|
self.controller.start()
|
|
|
|
def stop(self):
|
|
self.controller.stop()
|
|
|
|
|
|
class MockHandler:
|
|
def __init__(self, mock_server):
|
|
self.mock_server = mock_server
|
|
|
|
async def handle_DATA(self, server, session, envelope):
|
|
msg = {
|
|
'from': envelope.mail_from,
|
|
'to': envelope.rcpt_tos[0],
|
|
'content': envelope.content.decode(),
|
|
}
|
|
self.mock_server.messages.append(msg)
|
|
return '250 Message accepted for delivery'
|