Java+Selenium+Testng-web UI自动化测试框架-11实现邮件发送测试报告

前面的文章中介绍了如何生成测试报告，接下来我们就要实现将生成的测试报告通过email发送出去。

在util包下创建SendEmail.class代码如下：

package common;

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;

/**
 * 邮件发送测试报告
 */
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 files = new LinkedList();// 存放附件文件  
  
    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 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 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 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("" + 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 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.class中的方法，本框架是在@AfterSuite的时候调用的，即在每次testsuite运行完后发送 邮件。

    public static void sendEmail() {
        try {
            String userName = ""; // 发件人邮箱  
            String password = ""; // 发件人密码  
            String smtpHost = ""; // 邮件服务器    
  
            String to = ""; // 收件人，多个收件人以半角逗号分隔  
            String cc = ""; // 抄送，多个抄送以半角逗号分隔  
            String subject = ""; // 主题
            String body = GenerateReport.returnResult(); // 正文，可以用html格式的哟  
            // 附件的路径，多个附件也不怕  
            List ReportPath = Arrays.asList(GenerateReport.returnReportPath());
            List screenshotPath = GenerateReport.returnScreenshotPath();
            List attachments = new ArrayList();
            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);
        }
    }