Tại sao không bắt đầu Máy chủ SMTP thực sự đơn giản của riêng bạn bằng cách kế thừa từ smtpd.SMTPServer
và threading.Thread
:
class TestingSMTPServer(smtpd.SMTPServer, threading.Thread):
def __init__(self, port=25):
smtpd.SMTPServer.__init__(
self,
('localhost', port),
('localhost', port),
decode_data=False
)
threading.Thread.__init__(self)
def process_message(self, peer, mailfrom, rcpttos, data, **kwargs):
self.received_peer = peer
self.received_mailfrom = mailfrom
self.received_rcpttos = rcpttos
self.received_data = data
def run(self):
asyncore.loop()
process_message được gọi bất cứ khi nào Máy chủ SMTP của bạn nhận được yêu cầu thư, bạn có thể làm bất cứ điều gì bạn muốn ở đó.
Trong mã thử nghiệm, hãy làm như sau:
smtp_server = TestingSMTPServer()
smtp_server.start()
do_thing_that_would_send_a_mail()
smtp_server.close()
self.assertIn(b'hello', smtp_server.received_data)
Chỉ cần nhớ tới close()
những asyncore.dispatcher
bằng cách gọi smtp_server.close()
để kết thúc vòng lặp asyncore (ngăn chặn các máy chủ từ nghe).