Question

我很想知道是否有一种简单的方法来模拟Python中的IMAP服务器（一个imaplib模块），没有做很多工作。

是否有预先存在的解决方案？理想情况下，我可以连接到现有的IMAP服务器，进行转储，并让模拟服务器运行真正的邮箱/电子邮件结构。

懒惰的一些背景：我有一种讨厌的感觉，我正在写的这个小脚本会随着时间的推移而增长，并且喜欢来创建一个合适的测试环境，但鉴于它可能不随着时间的推移而增长，我不想做很多工作来让模拟服务器运行。

Answer 1

我发现在上次尝试时我很容易写一个IMAP服务器。它支持编写IMAP服务器，并且具有很大的灵活性。

Answer 2

对于任何一项测试，您真正需要多少？如果您开始按照真实服务器的复杂程度构建某些内容，以便可以在所有测试中使用它，那么您已经出错了。只需模拟任何测试所需的位。

不要费心去分享模拟实现。它们不应该是资产，而是可丢弃的零件。

Answer 3

由于我没有在python 3中找到方便我的需求（扭曲的邮件部分没有在python 3中运行），我做了一个带有asyncio的小模拟，如果你愿意，你可以改进：

我定义了一个扩展asyncio.Protocol的ImapProtocol。然后启动这样的服务器：

factory = loop.create_server(lambda: ImapProtocol(mailbox_map), 'localhost', 1143)
server = loop.run_until_complete(factory)

mailbox_map是地图的地图：email - ＆gt;邮箱地图 - ＆gt;一组消息。所以所有的邮件/邮箱都在内存中。

每次客户端连接时，都会创建一个新的ImapProtocol实例。然后，ImapProtocol为每个客户端执行并回答，实现功能/登录/获取/选择/搜索/存储：

class ImapHandler(object):
    def __init__(self, mailbox_map):
        self.mailbox_map = mailbox_map
        self.user_login = None
        # ...

    def connection_made(self, transport):
        self.transport = transport
        transport.write('* OK IMAP4rev1 MockIMAP Server ready\r\n'.encode())

    def data_received(self, data):
        command_array = data.decode().rstrip().split()
        tag = command_array[0]
        self.by_uid = False
        self.exec_command(tag, command_array[1:])

    def connection_lost(self, error):
        if error:
            log.error(error)
        else:
            log.debug('closing')
            self.transport.close()
        super().connection_lost(error)

    def exec_command(self, tag, command_array):
        command = command_array[0].lower()
        if not hasattr(self, command):
            return self.error(tag, 'Command "%s" not implemented' % command)
        getattr(self, command)(tag, *command_array[1:])

    def capability(self, tag, *args):
        # code for it...
    def login(self, tag, *args):
        # code for it...

然后在我的测试中，我在设置期间启动服务器：

self.loop = asyncio.get_event_loop()
self.server = self.loop.run_until_complete(self.loop.create_server(create_imap_protocol, 'localhost', 12345))

当我想模拟新消息时：

imap_receive(Mail(to='dest@fakemail.org', mail_from='exp@pouet.com', subject='hello'))

在拆解时停止它：

self.server.close()
asyncio.wait_for(self.server.wait_closed(), 1)

cf https://github.com/bamthomas/aioimaplib/blob/master/aioimaplib/tests/imapserver.py

编辑：我有一个错误的停止服务器，我用asyncio.Protocol重写它并修改答案以反映更改

Answer 4

我从未尝试过，但如果必须，我会从现有的SMTP服务器开始。

尽管极度懒惰，我如何在Python中模拟IMAP服务器？

4 个答案: