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

我很想知道是否有一种简单的方法可以imaplib在Python中模拟IMAP服务器(模块),而无需做很多工作.

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

一些背景到懒惰:我有一个讨厌的感觉,这我在写小脚本会随时间增长,并会想创造一个合适的测试环境,但考虑到它可能不会随时间增长,我不想做让模拟服务器运行的工作量很大.

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

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

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

因为我没有在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是地图的地图:电子邮件 - >邮箱地图 - >邮件集.所以所有的消息/邮箱都在内存中.

每次客户端连接时,都会创建一个新的ImapProtocol实例.然后,ImapProtocol为每个客户端执行并回答,实现功能/ login/fetch/select/search/store:

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重写它并修改答案以反映更改