package choose;

import static org.junit.Assert.assertTrue;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.openqa.selenium.WebDriver;

import webdriver.CreateWebDriver;
import webdriver.QuitWebDriver;

public class ChooseEnterpriseTest {

private CreateWebDriver cwd;
private QuitWebDriver qwd;
private WebDriver driver;
private ChooseEnterprise choose;

@Before
public void setUp() throws Exception {
cwd = new CreateWebDriver();
qwd = new QuitWebDriver();
driver = cwd.GetWebDriver("ie");
choose = new ChooseEnterprise();
}

@Test
public void testChoose() {
try{
assertTrue(choose.Choose(driver));
}catch(Exception e){
e.printStackTrace();
}
}

@After
public void tearDown() throws Exception {
cwd = null;
choose = null;
qwd.quitWebDriver(driver);
driver = null;
qwd = null;
}
}

package com.cplatform.training.parameter;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;

import jxl.Cell;
import jxl.Sheet;
import jxl.Workbook;
import jxl.read.biff.BiffException;

public class ParameterByExcel {
//创建getParameterFromExcel方法，根据行和列获取excel中表格的内容
public String getParameterFromExcel(File excelFile,int rowNum,int colNum) {
//定义String对象，用于存取读取到的excel表格中的内容，便于返回
String str="";
//定义一个工作簿
Workbook rwb = null;
//定义一个Cell
Cell cell = null;

// 创建输入流
try {
InputStream stream = new FileInputStream(excelFile);
// 获取Excel文件对象
rwb = Workbook.getWorkbook(stream);
// 获取文件的指定工作表 默认的第一个
Sheet sheet = rwb.getSheet(0);
//通过行和列定位需要读取的单元格，并且赋值给str
cell = sheet.getCell(colNum,rowNum);
str = cell.getContents();

} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
System.out.println("~~~FileNotFoundException~~~");
}catch (BiffException e) {
// TODO Auto-generated catch block
System.out.println("~~~BiffException~~~");
} catch (IOException e) {
// TODO Auto-generated catch block
System.out.println("~~~IOException~~~");
}

return str;
}
}

package mail;

import java.util.* ;

import javax.mail.* ;
import javax.mail.internet.* ;
import javax.activation.* ;

/******************************
* @author qicyt1812
* @decription 邮件发送处理程序
* @version 1.0
******************************/
public class Mail {
//定义发件人、收件人、SMTP服务器、用户名、密码、主题、内容等
private String displayName;
private String[] to;
private String from;
private String smtpServer;
private String username;
private String password;
private String subject;
private String content;
private boolean ifAuth; //服务器是否要身份认证
private String filename="";
private Vector<String> file = new Vector<String>(); //用于保存发送附件的文件名的集合


/**
* 设置SMTP服务器地址
*/
public void setSmtpServer(String smtpServer){
this.smtpServer=smtpServer;
}

/**
* 设置发件人的地址
*/
public void setFrom(String from){
this.from=from;
}
/**
* 设置显示的名称
*/
public void setDisplayName(String displayName){
this.displayName=displayName;
}

/**
* 设置服务器是否需要身份认证
*/
public boolean setIfAuth(boolean ifAuth){
this.ifAuth=ifAuth;
System.out.println("~~~"+ifAuth+"~~~"+this.ifAuth);
return this.ifAuth;
}

/**
* 设置E-mail用户名
*/
public void setUserName(String username){
this.username=username;
}

/**
* 设置E-mail密码
*/
public void setPassword(String password){
this.password=password;
}

/**
* 设置接收者
*/
public void setTo(String[] to){
this.to=to;
}

/**
* 设置主题
*/
public void setSubject(String subject){
this.subject=subject;
}

/**
* 设置主体内容
*/
public void setContent(String content){
this.content=content;
}

/**
* 该方法用于收集附件名
*/
public void addAttachfile(String fname){
file.addElement(fname);
}

public Mail(){

}

/**
* 初始化SMTP服务器地址、发送者E-mail地址、用户名、密码、接收者、主题、内容
*/
public Mail(String smtpServer,String from,String displayName,String username,String password,String[] to,String subject,String content){
this.smtpServer=smtpServer;
this.from=from;
this.displayName=displayName;
this.ifAuth=true;
this.username=username;
this.password=password;
this.to=to;
this.subject=subject;
this.content=content;
}

/**
* 初始化SMTP服务器地址、发送者E-mail地址、接收者
*/
public Mail(String smtpServer,String from,String displayName,String username,String password,String[] to){
this.smtpServer=smtpServer;
this.from=from;
this.displayName=displayName;
this.ifAuth=true;
this.username=username;
this.password=password;
this.to=to;
}

/**
* 初始化SMTP服务器地址、发送者E-mail地址、接收者、主题、内容
*/
public Mail(String smtpServer,String from,String displayName,String[] to,String subject,String content){
this.smtpServer=smtpServer;
this.from=from;
this.displayName=displayName;
this.ifAuth=false;
this.to=to;
this.subject=subject;
this.content=content;
}

/**
* 初始化SMTP服务器地址、发送者E-mail地址、接收者
*/
public Mail(String smtpServer,String from,String displayName,String[] to){
this.smtpServer=smtpServer;
this.from=from;
this.displayName=displayName;
this.ifAuth=true;
this.to=to;
}
/**
* 发送邮件
*/
public HashMap<String, String> send(){
HashMap<String, String> map=new HashMap<String, String>();
map.put("state", "success");
String message="邮件发送成功！";
Session session=null;
Properties props = System.getProperties();
props.put("mail.smtp.host", smtpServer);
if(ifAuth){ //服务器需要身份认证
props.put("mail.smtp.auth","true");
SmtpAuth smtpAuth=new SmtpAuth(username,password);
session=Session.getDefaultInstance(props, smtpAuth);
}else{
props.put("mail.smtp.auth","false");
session=Session.getDefaultInstance(props, null);
}
session.setDebug(true);
Transport trans = null;
try {
Message msg = new MimeMessage(session);
try{
Address from_address = new InternetAddress(from, displayName);
msg.setFrom(from_address);
}catch(java.io.UnsupportedEncodingException e){
e.printStackTrace();
}
Address[] address = null;
if (to!=null){
address = new InternetAddress[to.length];
for (int i=0; i<to.length; i++){
String s=to[i];
address[i] = new InternetAddress(s);
}
}
msg.setRecipients(Message.RecipientType.TO,address);
msg.setSubject(subject);
Multipart mp = new MimeMultipart();
MimeBodyPart mbp = new MimeBodyPart();
mbp.setContent(content.toString(), "text/html;charset=gb2312");
mp.addBodyPart(mbp);
if(!file.isEmpty()){//有附件
Enumeration<String> efile=file.elements();
while(efile.hasMoreElements()){
mbp=new MimeBodyPart();
filename=efile.nextElement().toString(); //选择出每一个附件名
FileDataSource fds=new FileDataSource(filename); //得到数据源
mbp.setDataHandler(new DataHandler(fds)); //得到附件本身并至入BodyPart
mbp.setFileName(fds.getName()); //得到文件名同样至入BodyPart
mp.addBodyPart(mbp);
}
file.removeAllElements();
}
msg.setContent(mp); //Multipart加入到信件
msg.setSentDate(new Date()); //设置信件头的发送日期
//发送信件
msg.saveChanges();
trans = session.getTransport("smtp");
trans.connect(smtpServer, username, password);
trans.sendMessage(msg, msg.getAllRecipients());
trans.close();

}catch(AuthenticationFailedException e){
map.put("state", "failed");
message="邮件发送失败！错误原因：\n"+"身份验证错误!";
e.printStackTrace();
}catch (MessagingException e) {
message="邮件发送失败！错误原因：\n"+e.getMessage();
map.put("state", "failed");
e.printStackTrace();
Exception ex = null;
if ((ex = e.getNextException()) != null) {
System.out.println(ex.toString());
ex.printStackTrace();
}
}
System.out.println("\n提示信息:"+message);
map.put("message", message);
return map;
}

}

package mail;

public class SmtpAuth extends javax.mail.Authenticator {
private String username,password;

public SmtpAuth(String username,String password){
this.username = username;
this.password = password;
}
@Override
protected javax.mail.PasswordAuthentication getPasswordAuthentication() {
return new javax.mail.PasswordAuthentication(username,password);
}
}

package errorhandle;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

import mail.Mail;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;

public class ErrorHandle {

private String subject = "";
private String testcase = "";

//定义变量，存储时间，用于图片的存储
private Date date = null;
private SimpleDateFormat sdateFormat = null;

public ErrorHandle(){

subject = "xx平台，用户登录选择企业，遍历测试";
testcase = "本用例的目的：在不知道有多少企业的情况下，遍历链接测试，保证用户所属的每个企业都可以正常进入，可用于**同一类型链接是否到达正常页面<br>"+
"本用例的主要测试点为：<br>"+
"1.用户登录xx平台 <br>" +
"2.进入该用户所属的每个企业<br>"+
"3.退出登录 <br>";

//初始时间变量，并且设置时间格式
date = new Date();
sdateFormat=new SimpleDateFormat("yyyyMMddhhmmssSS");
//System.out.println(sdf.format(date.getTime()));
}

public boolean ErrorHandleMsg(Mail mail,WebDriver driver,String errorMsg){

String bodyText = driver.findElement(By.cssSelector("BODY")).getText();
if(bodyText.matches("^[\\s\\S]*HTTP 500 －内部服务器错误[\\s\\S]*$")){

//设置邮件标题
mail.setSubject(subject+"，测试结果：FAIL-500");

//设置邮件内容
mail.setContent(testcase+
"===========================================================<br>"+
"<b><font color=\"red\">"+errorMsg+
"<br>"+
"页面报错500，可能服务器宕机，请检查原因，当前页面返回的url链接为:<br>"
+driver.getCurrentUrl()+
"<br>当前页面的内容为:<br>"+
driver.getPageSource()+
"</font></b>");

try {
File screenShotFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
File pic = new File(System.getProperty("user.dir")+"/screenshot/choose/500-"+sdateFormat.format(date.getTime())+".png");
FileUtils.copyFile(screenShotFile, pic);
mail.addAttachfile(pic.getPath());
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

//发送邮件
mail.send();

System.out.println("！！！errorHandle方法中页面报错500，请检查服务器！！！");

// returnFlag = false;

}else if(bodyText.matches("^[\\s\\S]*404[\\s\\S]*$")){
System.out.println("errorHandle方法中出现失败，开始发送提醒邮件");
//设置邮件标题
mail.setSubject(subject+"，测试结果：FAIL-404");

//设置邮件内容
mail.setContent(testcase+
"===========================================================<br>"+
"<b><font color=\"red\">"+errorMsg+
"<br>"+
"<b><font color=\"red\">页面报错404，可能页面丢失，请检查原因，当前页面返回的url链接为:<br>"
+driver.getCurrentUrl()
+"<br>当前页面的内容为:<br>"
+driver.getPageSource()
+"</font></b>");

try {
File screenShotFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
File pic = new File(System.getProperty("user.dir")+"/screenshot/choose/404-"+sdateFormat.format(date.getTime())+".png");
FileUtils.copyFile(screenShotFile, pic);
mail.addAttachfile(pic.getPath());
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

//发送邮件
mail.send();

System.out.println("！！！errorHandle方法中页面报错404，可能页面丢失，请检查原因！！！");

}else{
// System.out.println("errorHandle方法中页面报错404，开始发送提醒邮件");
//设置邮件标题
mail.setSubject(subject+"，测试结果：FAIL");

//设置邮件内容
mail.setContent(testcase+
"===========================================================<br>"+
"<b><font color=\"red\">"+errorMsg+
"<br>"+
"<b><font color=\"red\">请检查，当前页面返回的url链接为:<br>"
+driver.getCurrentUrl()+
"<br>当前页面的title为：<br>"
+driver.getTitle()+
"<br>当前页面的内容为:<br>"
+driver.getPageSource()+
"</font></b>");

try {
File screenShotFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
File pic = new File(System.getProperty("user.dir")+"/screenshot/choose/others-"+sdateFormat.format(date.getTime())+".png");
FileUtils.copyFile(screenShotFile, pic);
mail.addAttachfile(pic.getPath());
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

//发送邮件
mail.send();

System.out.println("！！！errorHandle方法中出现500和404以外的问题，请检查！！！");
}

return false;
}
}

package choose;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;

import errorhandle.ErrorHandle;

import mail.Mail;

/****************************************************
* @author qicyt1812
* @description 遍历链接测试，可用于**同一类型链接是否到达正常页面
* @date 2013-10-14
****************************************************/

public class ChooseEnterprise
{
//布尔变量用于存储方法返回值
public boolean bool = false;

//定义邮件内容
private String content = "";

//定义mail对象，以便于邮件发送
private Mail mail = null;
File file = null;
String[] toM =null;

private String baseUrl = "";

//定义错误处理对象
private ErrorHandle error = null;

//定义变量，存储时间，用于图片的存储
private Date date = null;
private SimpleDateFormat sdateFormat = null;

public ChooseEnterprise(){
//初始测试访问地址
baseUrl = "http://www.csdn.net/";

//初始错误处理程序
error = new ErrorHandle();

//初始化邮件内容
content ="本用例的目的：在不知道有多少企业的情况下，遍历链接测试，保证用户所属的每个企业都可以正常进入，可用于**同一类型链接是否到达正常页面<br>"+
"本用例的主要测试点为：<br>"+
"1.用户登录平台 <br>" +
"2.进入该用户所属的每个企业<br>"+
"3.退出登录 <br>"+
"本次测试的企业有：<br>";

/********************************
* 配置邮件发送的相关信息
********************************/

//配置邮件发送地址，可发送多个人
toM = new String[1];
toM[0]="qicyt1812@csdn.net";

mail = new Mail("smtp.csdn.net","qicyt1812@csdn.net,"xx平台自动化测试","qicyt1812","qicyt password",toM);

//初始时间变量，并且设置时间格式
date = new Date();
sdateFormat=new SimpleDateFormat("yyyyMMddhhmmssSS");
//System.out.println(sdf.format(date.getTime()));
}

public boolean Choose(WebDriver driver){

int i = 1;
int j = 1;
String str = "";
String strLast = "";
String msgContent = "";

//因为不知道需要打开的链接一共有多少个，所以使用while循环，始终给一个真值，直到条件满足跳出循环
while(i!=0){

driver.get(baseUrl + "zwyq/cl/login.jsp");

try{
/*********第1步：进入登录页面，在登录页面输入**********/
//判断是否成功进入登录页面，如果没有则跳出循环
if(!driver.findElement(By.cssSelector("span.fontred2")).getText().matches("^[\\s\\S]*使用xxx帐号登录[\\s\\S]*$")){
msgContent = "第"+i+"轮测试，没有正常开启登录页面";
bool = error.ErrorHandleMsg(mail, driver, msgContent);
break;
}

//输入手机号
driver.findElement(By.id("usersPhone")).clear();
driver.findElement(By.id("usersPhone")).sendKeys("13770646564");

//输入密码
driver.findElement(By.id("usersPassword")).clear();
driver.findElement(By.id("usersPassword")).sendKeys("123456");

//点击登录
driver.findElement(By.cssSelector("img")).click();

/**********第2步：到达选择企业页面*********/
//判断是否到达选择企业页面
// System.out.println(i+"***"+driver.findElement(By.xpath("//h3")).getText().matches("^[\\s\\S]*选择企业[\\s\\S]*$"));
if(!driver.findElement(By.xpath("//h3")).getText().matches("^[\\s\\S]*选择企业[\\s\\S]*$")){
msgContent = "第"+i+"轮测试，没有到达选择企业页面";
bool = error.ErrorHandleMsg(mail, driver, msgContent);
System.out.println("===没有到达选择企业页面，跳出===");
break;
}

//获取属性为class=qy_bg的div元素的值
str = driver.findElement(By.xpath("(//div[@class='qy_bg'])["+i+"]")).getText();
//获取属性为class=qy_bg的div最后一个元素的值
strLast = driver.findElement(By.xpath("(//div[@class='qy_bg'])[last()]")).getText();

//点击企业名称，因为存在重复的值，所以需要做一些特殊处理，这个只针对于13770646564这个账号，一般情况下不会存在相同名称的企业，所以直接用else里面的语句即可。
if(i==2||i==7){
driver.findElement(By.xpath("(//a[contains(text(),'"+str+"')])[2]")).click();
}else{
driver.findElement(By.xpath("(//a[contains(text(),'"+str+"')])")).click();
}

/**********第3步：进入企业内刊主页***********/
//判断是否进入内刊主页
if(!driver.findElement(By.xpath("//p[2]")).getText().matches("^[\\s\\S]*"+str+"[\\s\\S]*$")){
msgContent = "第"+i+"轮测试，{"+str+"}企业，没有进入企业内刊主页";
bool = error.ErrorHandleMsg(mail, driver, msgContent);
break;
}

//如果成功进入，则截取内刊图片
try {
File screenShotFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
File pic = new File(System.getProperty("user.dir")+"/screenshot/choose/"+i+"-"+str+"-"+sdateFormat.format(date.getTime())+".png");
FileUtils.copyFile(screenShotFile, pic);
mail.addAttachfile(pic.getPath());
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

/**********第4步：点击进入个人中心的*************/
//点击个人中心的图标
driver.findElement(By.cssSelector("div.r_sz > a > img")).click();
//检查是否了进入个人中心
if(!driver.findElement(By.xpath("//div[4]/div[2]/h5")).getText().matches("^[\\s\\S]*退 出[\\s\\S]*$")){
msgContent = "第"+i+"轮测试，{"+str+"}企业，没有到达个人中心页面";
bool = error.ErrorHandleMsg(mail, driver, msgContent);
break;
}

/**********第5步：退出登录**********/
//点击退出
driver.findElement(By.linkText("退 出")).click();
//检查是否成功退出
if(!driver.findElement(By.cssSelector("p.fontred")).getText().matches("^[\\s\\S]*欢迎登录xxx[\\s\\S]*$")){
msgContent = "第"+i+"轮测试，{"+str+"}企业，退出失败，没有正常返回到登录页面";
bool = error.ErrorHandleMsg(mail, driver, msgContent);
break;
}

j = i;

// System.out.println(j);
System.out.println(j+"==="+str+"==="+strLast);

content += "<b>"+i+":"+str+"<br></b>";

//当str的值等于strLast，表示已经到达最后一个链接，此时可跳出循环
if(strLast.equalsIgnoreCase(str)){
mail.setSubject("xxx平台，用户登录选择企业，遍历测试:PASS");
mail.setContent(content);
mail.send();
break;
}

i++;

bool = true;
}catch(Exception e){
msgContent = "第"+i+"轮测试，捕获异常:<br>"+e.toString();
bool = error.ErrorHandleMsg(mail, driver, msgContent);
break;
}

}

return bool;
}
}

package webdriver;
import java.util.concurrent.TimeUnit;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.ie.InternetExplorerDriver;
import org.openqa.selenium.remote.SessionNotFoundException;

public class CreateWebDriver {
private WebDriver driver;

public WebDriver GetWebDriver(String exploreType){

if (exploreType.equals("ie")){
try{
//配置IEDriverServer.exe，使用IE浏览器，必须用IEDriverServer.exe驱动
System.setProperty("webdriver.ie.driver", "C:\\Program Files (x86)\\Internet Explorer\\IEDriverServer.exe");

//new一个InternetExplorerDriver对象
driver = new InternetExplorerDriver();

//设置超时等待时间
driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS) ;

}catch(SessionNotFoundException se){
System.out.println("~~4~~捕获SessionNotFoundException异常~~");
}

return driver;
}else if(exploreType.equals("firefox")){

//new一个FirefoxDriver对象
driver=new FirefoxDriver();

//设置超时等待时间
driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS) ;

return driver;
}else{
throw new IllegalStateException("selected explorer is not supported");
}
}

}

package webdriver;

import java.util.Set;

import org.openqa.selenium.WebDriver;

public class QuitWebDriver {
public void quitWebDriver(WebDriver driver){
Set<String> windows=driver.getWindowHandles();
for(String a:windows){
driver.switchTo().window(a).close();
}
try {
Thread.sleep(15);
driver.quit();
} catch (InterruptedException ex) {
System.out.println("~~~退出WebDriver时报InterruptedException异常~~~");
}
}
}

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>

<groupId>zwyq</groupId>
<artifactId>zwyq</artifactId>
<version>0.0.1-SNAPSHOT</version>
<packaging>jar</packaging>

<name>zwyq</name>
<url>http://maven.apache.org</url>

<properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties>

<dependencies>
<!-- 加载junit包 -->
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<version>4.10</version>
<scope>test</scope>
</dependency>

<!-- 加载selenium包 -->
<dependency>
<groupId>org.seleniumhq.selenium</groupId>
<artifactId>selenium-java</artifactId>
<version>2.31.0</version>
</dependency>

<!-- 加载javax.mail包，用于邮件发送 -->
<dependency>
<groupId>javax.mail</groupId>
<artifactId>mail</artifactId>
<version>1.5.0-b01</version>
</dependency>

<!-- 加载读取Excle所需的jar包 -->
<dependency>
<groupId>net.sourceforge.jexcelapi</groupId>
<artifactId>jxl</artifactId>
<version>2.6.12</version>
</dependency>

<!-- 加载excel读取所需的org.apache.poi包 -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-excelant</artifactId>
<version>3.9</version>
</dependency>
</dependencies>
< /project>