SendEmail.java的代码
如何将生成的测试报告通过Email发送出去呢？
在util包下创建SendEmail.java代码如下：
package util;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;
import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.mail.BodyPart;
import javax.mail.Message;
import javax.mail.Multipart;
import javax.mail.Session;
import javax.mail.Transport;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeBodyPart;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import javax.mail.internet.MimeUtility;
/**
* 邮件发送测试报告.
*
* @author betty.shi
*
*/
public class SendEmail {
private static String defaultSenderName = "";// 默认的发件人用户名，defaultEntity用得到
private static String defaultSenderPass = "";// 默认的发件人密码，defaultEntity用得到
private static String defaultSmtpHost = "";// 默认的邮件服务器地址，defaultEntity用得到
private String smtpHost; // 邮件服务器地址
private String sendUserName; // 发件人的用户名
private String sendUserPass; // 发件人密码
private MimeMessage mimeMsg; // 邮件对象
private Session session;
private Properties props;
private Multipart mp;// 附件添加的组件
private List<FileDataSource> files = new LinkedList<FileDataSource>();// 存放附件文件
private void init() {
if (props == null) {
props = System.getProperties();
}
props.put("mail.smtp.host", smtpHost);
props.put("mail.smtp.auth", "true"); // 需要身份验证
session = Session.getDefaultInstance(props, null);
// 置true可以在控制台（console)上看到发送邮件的过程
session.setDebug(true);
// 用session对象来创建并初始化邮件对象
mimeMsg = new MimeMessage(session);
// 生成附件组件的实例
mp = new MimeMultipart();
}
private SendEmail(String smtpHost, String sendUserName, String sendUserPass, String to, String cc, String mailSubject, String mailBody,
List<String> attachments) {
this.smtpHost = smtpHost;
this.sendUserName = sendUserName;
this.sendUserPass = sendUserPass;
init();
setFrom(sendUserName);
setTo(to);
setCC(cc);
setBody(mailBody);
setSubject(mailSubject);
if (attachments != null) {
for (String attachment : attachments) {
addFileAffix(attachment);
}
}
}
/**
* 邮件实体
*
* @param smtpHost
* 邮件服务器地址
* @param sendUserName
* 发件邮件地址
* @param sendUserPass
* 发件邮箱密码
* @param to
* 收件人，多个邮箱地址以半角逗号分隔
* @param cc
* 抄送，多个邮箱地址以半角逗号分隔
* @param mailSubject
* 邮件主题
* @param mailBody
* 邮件正文
* @param attachmentPath
* 附件路径
* @return
*/
public static SendEmail entity(String smtpHost, String sendUserName, String sendUserPass, String to, String cc, String mailSubject, String mailBody,
List<String> attachments) {
return new SendEmail(smtpHost, sendUserName, sendUserPass, to, cc, mailSubject, mailBody, attachments);
}
/**
* 默认邮件实体，用了默认的发送帐号和邮件服务器
*
* @param to
* 收件人，多个邮箱地址以半角逗号分隔
* @param cc
* 抄送，多个邮箱地址以半角逗号分隔
* @param subject
* 邮件主题
* @param body
* 邮件正文
* @param attachment
* 附件全路径
* @return
*/
public static SendEmail defaultEntity(String to, String cc, String subject, String body, List<String> attachments) {
return new SendEmail(defaultSmtpHost, defaultSenderName, defaultSenderPass, to, cc, subject, body, attachments);
}
/**
* 设置邮件主题
*
* @param mailSubject
* @return
*/
private boolean setSubject(String mailSubject) {
try {
mimeMsg.setSubject(mailSubject);
} catch (Exception e) {
return false;
}
return true;
}
/**
* 设置邮件内容,并设置其为文本格式或HTML文件格式，编码方式为UTF-8
*
* @param mailBody
* @return
*/
private boolean setBody(String mailBody) {
try {
BodyPart bp = new MimeBodyPart();
bp.setContent("<meta http-equiv=Content-Type content=text/html; charset=UTF-8>" + mailBody, "text/html;charset=UTF-8");
// 在组件上添加邮件文本
mp.addBodyPart(bp);
} catch (Exception e) {
System.err.println("设置邮件正文时发生错误！" + e);
return false;
}
return true;
}
/**
* 添加一个附件
*
* @param filename
* 邮件附件的地址，只能是本机地址而不能是网络地址，否则抛出异常
* @return
*/
public boolean addFileAffix(String filename) {
try {
if (filename != null && filename.length() > 0) {
BodyPart bp = new MimeBodyPart();
FileDataSource fileds = new FileDataSource(filename);
bp.setDataHandler(new DataHandler(fileds));
bp.setFileName(MimeUtility.encodeText(fileds.getName(), "utf-8", null)); // 解决附件名称乱码
mp.addBodyPart(bp);// 添加附件
files.add(fileds);
}
} catch (Exception e) {
System.err.println("增加邮件附件：" + filename + "发生错误！" + e);
return false;
}
return true;
}
/**
* 删除所有附件
*
* @return
*/
public boolean delFileAffix() {
try {
FileDataSource fileds = null;
for (Iterator<FileDataSource> it = files.iterator(); it.hasNext();) {
fileds = it.next();
if (fileds != null && fileds.getFile() != null) {
fileds.getFile().delete();
}
}
} catch (Exception e) {
return false;
}
return true;
}
/**
* 设置发件人地址
*
* @param from
* 发件人地址
* @return
*/
private boolean setFrom(String from) {
try {
mimeMsg.setFrom(new InternetAddress(from));
} catch (Exception e) {
return false;
}
return true;
}
/**
* 设置收件人地址
*
* @param to收件人的地址
* @return
*/
private boolean setTo(String to) {
if (to == null)
return false;
try {
mimeMsg.setRecipients(Message.RecipientType.TO, InternetAddress.parse(to));
} catch (Exception e) {
return false;
}
return true;
}
/**
* 设置抄送
*
* @param cc
* @return
*/
private boolean setCC(String cc) {
if (cc == null) {
return false;
}
try {
mimeMsg.setRecipients(Message.RecipientType.CC, InternetAddress.parse(cc));
} catch (Exception e) {
return false;
}
return true;
}
/**
* 发送邮件
*
* @return
*/
public boolean send() throws Exception {
mimeMsg.setContent(mp);
mimeMsg.saveChanges();
System.out.println("正在发送邮件....");
Transport transport = session.getTransport("smtp");
// 连接邮件服务器并进行身份验证
transport.connect(smtpHost, sendUserName, sendUserPass);
// 发送邮件
transport.sendMessage(mimeMsg, mimeMsg.getRecipients(Message.RecipientType.TO));
System.out.println("发送邮件成功！");
transport.close();
return true;
}
}
SendEmail的方法调用
可以在任何想要发送邮件的地方调用SendEmail.java中的方法，本框架是在@AfterSuite的时候调用的，即在每次testsuite运行完后发送邮件。
@AfterSuite
public static void sendEmail() {
try {
String userName = "[email protected]"; // 发件人邮箱
String password = "password"; // 发件人密码
String smtpHost = "smtp.126.com"; // 邮件服务器
String to = "[email protected]"; // 收件人，多个收件人以半角逗号分隔
String cc = ""; // 抄送，多个抄送以半角逗号分隔
String subject = "Web UI Test Report"; // 主题
String body = ReportListener.returnResult(); // 正文，可以用html格式的哟
// 附件的路径，多个附件也不怕
List<String> ReportPath = Arrays.asList(ReportListener.returnReportPath());
List<String> screenshotPath = ReportListener.returnScreenshotPath();
List<String> attachments = new ArrayList<String>();
attachments.addAll(ReportPath);
attachments.addAll(screenshotPath);
SendEmail email = SendEmail.entity(smtpHost, userName, password, to, cc, subject, body, attachments);
email.send(); // 发送！
}
catch(Exception e) {
Log4jUtil.error(e);
}
}
其中，如下代码需要修改：
String userName = "[email protected]"; // 发件人邮箱
String password = "password"; // 发件人密码
String smtpHost = "smtp.126.com"; // 邮件服务器
String to = "[email protected]"; // 收件人，多个收件人以半角逗号分隔
String cc = ""; // 抄送，多个抄送以半角逗号分隔
String subject = "Web UI Test Report"; // 主题
