11

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

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

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

4

4 回答 4

10

我发现上次尝试用扭曲的方式编写 IMAP 服务器非常容易。它支持编写 IMAP 服务器，你有很大的灵活性。

于 2008-12-09T03:14:39.407 回答
7

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

不要费心尝试分享一个模拟实现。它们不应该是资产，而是可丢弃的点数。

于 2008-12-09T15:41:57.213 回答
6

由于我在 python 3 中没有找到适合我需要的东西（twisted 的邮件部分没有在 python 3 中运行），所以我用 asyncio 做了一个小模拟，如果你愿意，你可以改进：

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

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

邮箱映射是映射的映射：电子邮件 -> 邮箱映射 -> 消息集。所以所有的消息/邮箱都在内存中。

每次客户端连接时，都会创建一个新的 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 重写它并修改答案以反映更改

于 2016-02-18T17:48:13.917 回答
1

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

于 2008-12-09T09:12:17.717 回答