Robot Framework通过Python SMTP进行email收发测试。
工作中需要对发送的邮件进行过滤，方法基本属于ACL控制，即查看“源/目的”邮件地址，邮件标题，邮件正文，邮件附件等进行过滤。
所以需要先模拟一下用Python能否达到邮件Client，Server的功能，还有能否在server上显示发送的 “源/目的”邮件地址，邮件标题，邮件正文，邮件附件 这些信息。
如果能取到这些信息，然后就可以对这些字符串进行查找，然后判断过滤功能是否work的目的了。
现在咱们先看下邮件服务器：
mailserver.py:
import smtpd import asyncore class CustomSMTPServer(smtpd.SMTPServer): def process_message(self, peer, mailfrom, rcpttos, data): print 'Receiving message from:', peer print 'Message addressed from:', mailfrom print 'Message addressed to :', rcpttos print 'Message length :', len(data) print 'data :', data return server = CustomSMTPServer(('172.18.0.100', 9999), None) asyncore.loop()
这个就是邮件服务器，保存完后，直接双击运行就好，它会一直运行的。
UPDATE:
将此邮件服务器做成类似于linux服务：
1.将此程序放到/home目录下：
[root@Server init]# vim /home/mailserver.py import smtpd import asyncore class CustomSMTPServer(smtpd.SMTPServer): def process_message(self, peer, mailfrom, rcpttos, data): self.peer = peer self.mailfrom = mailfrom self.rcpttos = rcpttos self.data = data mail_log = open('/home/current_mail.log','wb+') mail_log.write(str('Receiving message from:')+ str(self.peer)+'\n') mail_log.write(str('Message addressed from:')+ str(self.mailfrom)+'\n') mail_log.write(str('Message addressed to :')+ str(self.rcpttos)+'\n') mail_log.write(str('Message length :')+ str(len(self.data))+'\n') mail_log.write(str('data :')+ str(self.data)+'\n') mail_log.close() return server = CustomSMTPServer(('172.16.0.3', 9999), None) asyncore.loop()
这个程序会在当前目录下创建邮件log，用来判断客户端发送的请求是否正确。
2.在/etc/init目录下（不是/etc/init.d）创建mailserver.conf这个文件，然后填写如下配置：
[root@Server init]# vim mailserver.conf description "mailserver is in /home/mailserver.py" author "xxx@xxx.com" start on runlevel [2345] stop on runlevel [!2345] #env AN_ENVIRONMENTAL_VARIABLE=i-want-to-set respawn exec /home/mailserver.py
3. 执行“initctl reload-configuration"命令
4. 然后可以通过以下命令启动或关闭邮件服务器：
start mailserver
stop mailserver
restart mailserver
邮件客户端：
Step1: 先在pyton目录下创建一个自定义的文件夹“SelfEmailLibrary”（您的Python安装到哪里就相应的放到哪里）：
F:\Python27\Lib\site-packages\SelfEmailLibrary
Step2: 将下面的代码复制到“mailsend.py”和“__init__.py”中。
mailsend.py:
import smtplib from email.MIMEMultipart import MIMEMultipart from email.MIMEBase import MIMEBase from email.MIMEText import MIMEText from email import Encoders import os class SendEmailUtility(object): ROBOT_LIBRARY_SCOPE = 'Global' def __init__(self): print 'send email utility' def send_mail_with_attachment(self, from_user, from_password, to, subject, text, attach, mailserver, serverport): self.mailserver = str(mailserver) self.serverport = int(serverport) msg = MIMEMultipart() msg['From'] = from_user msg['To'] = to msg['Subject'] = subject msg.attach(MIMEText(text)) part = MIMEBase('application', 'octet-stream') part.set_payload(open(attach, 'rb').read()) #Encoders.encode_base64(part) part.add_header('Content-Disposition', 'attachment; filename="%s"' % os.path.basename(attach)) msg.attach(part) mailServer = smtplib.SMTP(self.mailserver, self.serverport) mailServer.ehlo() #mailServer.starttls() mailServer.ehlo() #mailServer.login(from_user, from_password) mailServer.sendmail(from_user, to, msg.as_string()) # Should be mailServer.quit(), but that crashes... mailServer.close() def send_mail_no_attachment(self, from_user, from_password, to, subject, text, mailserver, serverport): self.mailserver = str(mailserver) self.serverport = int(serverport) msg = MIMEMultipart() msg['From'] = from_user msg['To'] = to msg['Subject'] = subject msg.attach(MIMEText(text)) mailServer = smtplib.SMTP(self.mailserver, self.serverport) mailServer.ehlo() #mailServer.starttls() mailServer.ehlo() #mailServer.login(from_user, from_password) mailServer.sendmail(from_user, to, msg.as_string()) # Should be mailServer.quit(), but that crashes... mailServer.close()
__init__.py:
from mailsend import SendEmailUtility __version__ = '1.0' class SelfEmailLibrary(SendEmailUtility): ROBOT_LIBRARY_SCOPE = 'GLOBAL'
Step3：创建一个脚本测试一下这两个function是否工作：
mailclient.py:
from SelfEmailLibrary import * a = SendEmailUtility() a.send_mail_no_attachment(from_user="aaa@aaa.com", from_password='', to="bbb@bbb.com", subject="selftry", text="asdlfkajsdlfd", mailserver="172.18.0.100", serverport="9999") a.send_mail_with_attachment(from_user="ddd@ddd.com", from_password='', to="ccc@ccc.com", subject="selftry",text="bbbbbbbbbb", attach='simple.txt', mailserver="172.18.0.100", serverport="9999")
Step4: 测试这俩函数对不对：
1. 先双击开启mailserver.py。
2. 执行mailclient.py。
3. 查看mailserver的显示是否正确：
可以看到，邮件的各个属性还有附件内容（simple.txt的内容就是“testkeyword”）都可以查看到。这样我们就可以通过查看mailserver收到的DATA来判断过滤功能了。
Note：
send_mail_with_attachment和send_mail_no_attachment两个函数中的from_password参数其实没有用，可以删掉。然后如果你想用base64加密，可以取消注销掉Encoders.encode_base64(part)这一行。
其实功能已经实现了，要是想让它在RobotFramework中work就是轻而易举的事儿了。
