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'