自动化测试-东航B2C网站测试框架搭建

package org.travelsky.autotest.selenium.model;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Date;
import java.util.List;

import org.openqa.selenium.Alert;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.htmlunit.HtmlUnitDriver;
import org.openqa.selenium.ie.InternetExplorerDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.travelsky.autotest.selenium.config.BaseConst;
import org.travelsky.autotest.selenium.config.StartUpType;
import org.travelsky.autotest.selenium.exception.MyExecuteMethodOutOfTimesException;
import org.travelsky.autotest.selenium.exception.MyInvokeException;
import org.travelsky.autotest.selenium.exception.MySeleniumException;
import org.travelsky.autotest.util.DateOpt;
import org.travelsky.autotest.util.RegexUtil;

public class BaseSeleniumManager implements ExpectedCondition<WebElement>, BaseSelenium {
    protected Logger log = LoggerFactory.getLogger(getClass());

    private String xpath;
    public int failCout = BaseConst.FAIL_COUNT;
    private boolean acceptNextAlert = true;

    /**
     * 启动浏览器
     */
    public WebDriver startUpBrowse(StartUpType type) {
        WebDriver driver;
        switch (type) {
        case IE:
            System.setProperty("webdriver.ie.driver", "E:\\sts-bundle\\selenium-2.42.2\\IEDriverServer.exe");
            driver = new InternetExplorerDriver();
            break;
        case FireFox:
            System.setProperty("webdriver.firefox.bin", "C:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");
            driver = new FirefoxDriver();
            break;
        case Chrome:
            System.setProperty("webdriver.chrome.bin", "C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe");
            driver = new ChromeDriver();
            break;
        case HtmlUnit:
            driver = new HtmlUnitDriver();
            break;
        default:
            throw new IllegalArgumentException("type is wrong");

        }
        return driver;
    }

    /**
     * 关闭driver
     * 
     * @param driver
     */
    public void close(WebDriver driver) {
        if (null == driver)
            return;
        driver.quit();
    }

    /**
     * you will not directly use this ,you should be use "checkCondition" method
     * implement ExpectedCondition
     */
    @Override
    public WebElement apply(WebDriver driver) {
        return driver.findElement(By.xpath(this.xpath));
    }

    /**
     * this method is synchronized to find and return element
     * 
     * @param driver
     * @param xpath
     *            locate element
     * @param timeOutSecond
     *            after this time ,will be throw a exception
     * @return
     */
    public WebElement checkCondition(WebDriver driver, String xpath) {
        this.xpath = xpath;
        WebDriverWait wait = new WebDriverWait(driver, BaseConst.OUT_TIME_SECOND);
        WebElement element = wait.until(this);
        return element;
    }
    
    /**
     * 自定义超时时间
     * @param driver
     * @param xpath
     * @param timeout 超时时间
     * @return
     */
    public WebElement checkCondition(WebDriver driver, String xpath,int timeout) {
        this.xpath = xpath;
        WebDriverWait wait = new WebDriverWait(driver,  timeout);
        WebElement element = wait.until(this);
        return element;
    }

    /**
     * repeat click element ,if click event have a exception will find the next
     * element in the list<WebElement> to click
     * 
     * @param elements
     * @param i
     *            start with the index = i element
     */
    public Boolean clickWebElement(List<WebElement> elements, int i) {
        int size = elements.size();
        if (i == size) {
            log.error("没有合适的按钮");
            return false;
        }
        try {
            elements.get(i).click();
            return true;
        } catch (Exception e) {
            ++i;
            this.clickWebElement(elements, i);
        }
        return true;
    }

    /**
     * 重新运行传入的方法，次数为failCout配置的数量
     * 会抛出MySeleniumException和MyInvokeException异常
     * @throws MyInvokeException
     */
    public void runBack(Method m, Throwable e, Object... args) throws MyExecuteMethodOutOfTimesException, MyInvokeException {
        if (this.failCout < 0) {
            throw new MyExecuteMethodOutOfTimesException(this.getClass()+m.getName()+ " 方法执行超过次数", e);
        }
        this.failCout--;
        try {
            m.invoke(this, args);
        } catch (IllegalAccessException | IllegalArgumentException | InvocationTargetException e1) {
            throw new MyInvokeException(this.getClass().getName()+m.getName()+"方法回滚异常",e1);
        }
    }

    /**
     * 弹出框点确认
     */
    public String AcceptAlertAndGetItsText(WebDriver driver) {
        String alertText = "";
        try {
            Alert alert = driver.switchTo().alert();
            alertText = alert.getText();
            if (acceptNextAlert) {
                alert.accept();
            } else {
                alert.dismiss();
            }
            return alertText;
        } catch (Exception e) {
            //不需要做处理
        } finally {
            acceptNextAlert = true;
        }
        return alertText;
    }

    /**
     * 关闭弹出框
     */
    public String closeAlertAndGetItsText(WebDriver driver) {
        String alertText = "";
        try {
            Alert alert = driver.switchTo().alert();
            alertText = alert.getText();
            alert.dismiss();
            return alertText;
        } catch (Exception e) {
            //不需要做处理
        }
        return alertText;
    }

    /**
     * 正则匹配替换URL的日期
     * 
     * @param url
     *            地址
     * @param regex
     *            匹配策略"\\d{6}"表示匹配6位数字
     * @param dateFormat
     *            日期格式20140404的格式为"yyyyMMdd"
     * @param addDay
     *            后移几天
     * @return
     */
    public String repalceUrl(String url, String regex, String dateFormat, int addDay) {
        String newUrl = url;
        List<String> matches = RegexUtil.matchRegex(newUrl, regex);
        for (String old : matches) {
            Date day = DateOpt.stringTypeToDateType(old, dateFormat);
            String reString = DateOpt.dateTypeToString(DateOpt.add(day, addDay), dateFormat);
            newUrl = newUrl.replace(old, reString);
        }
        return newUrl;
    }
    
    /**
     * 
     * @param ele  CHECKBOX元素
     *          
     */
    public void selectCheckBox(WebElement ele, boolean selected) {
        if (ele.isSelected()&&!selected||!ele.isSelected()&&selected) {
                ele.click();
        }

    }

    /**
     * 由于部分页面渲染速度慢，导致无法点击
     * @param second
     */
    public void delay(int second){
           try {
            Thread.sleep(second*1000);
        } catch (InterruptedException e) {
            log.error("BaseSeleniumManager的delay()方法报错",e);
        }
    }
    
    /**
     * 检查页面的title
     * @param driver
     * @param expect
     * @return
     */
    public boolean checkTitle(WebDriver driver,String expect){
        return driver.getTitle().contains(expect);
    }

}

package org.travelsky.autotest.selenium.service.manager;

import java.util.ArrayList;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadPoolExecutor;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.travelsky.autotest.selenium.service.TestCaseService;
/**
 * 
 * @author sunfan
 *
 */
public class ThreadPoolService {
    static Logger log = LoggerFactory.getLogger("ThreadPoolService");

    /**
     * 
     * @param testcases
     * @throws InterruptedException
     */
    public void executeTestCase(
            ArrayList<? extends Runnable> testcases)
            throws InterruptedException {
        ExecutorService executorService = Executors.newCachedThreadPool();
        for (Runnable testCase : testcases) {
            executorService.execute(testCase);
        }
    }

    /**
     * 获取当前线程池中正在运行的线程数
     * @param executorService
     * @return
     */
    public int getThreadCount(ExecutorService executorService) {
        int threadCount = ((ThreadPoolExecutor) executorService)
                .getActiveCount();
        log.info("当前正在运行的线程数：" + threadCount);
        return threadCount;
    }

}

package org.travelsky.autotest.selenium.service;

import java.util.List;

import javax.security.auth.login.LoginException;

import org.openqa.selenium.WebDriver;
import org.travelsky.autotest.selenium.config.StartUpType;
import org.travelsky.autotest.selenium.exception.MyBookException;
import org.travelsky.autotest.selenium.exception.MyExecuteMethodOutOfTimesException;
import org.travelsky.autotest.selenium.exception.MyFreePassengerException;
import org.travelsky.autotest.selenium.exception.MyInvokeException;
import org.travelsky.autotest.selenium.exception.MyLoginException;
import org.travelsky.autotest.selenium.exception.MyPassengerException;
import org.travelsky.autotest.selenium.exception.MyPaymentException;
import org.travelsky.autotest.selenium.exception.MyQueryFlightException;
import org.travelsky.autotest.selenium.exception.MySelectFlightException;
import org.travelsky.autotest.selenium.model.BaseSeleniumManager;
import org.travelsky.autotest.selenium.model.laputa.entity.QueryFlightInfo;
import org.travelsky.autotest.selenium.model.laputa.entity.User;
import org.travelsky.autotest.selenium.model.laputa.impl.LaputaDynamicAvation;
import org.travelsky.autotest.selenium.model.laputa.impl.LaputaLowPrice;
import org.travelsky.autotest.selenium.service.util.TestCaseServiceUtil;

/**
 * It‘s a entrance to provide some method to run testcase
 * 
 * @author sunfan
 * 
 */
public abstract class TestCaseService extends BaseSeleniumManager implements Runnable {
    TestCaseServiceUtil util = new TestCaseServiceUtil();
    WebDriver driver;
    
    /**
     * @param type   启动浏览器的方式， StartUpType.IE,StartUpType.FireFox
     * @param orgCity  出发城市
     * @param arrCity 到达城市
     * @param depdate  单程时间
     * @param rtDate   往返时间  （如果无往返可不填）
     * @param product 单程产品    "[productCd]", "BNGM"  "COMMON_INTER_Y_DIS", ""
     * @param rtProduct  往返产品（注意，如果为单程航班这里必须为""或者NULL）
     * @param tripType  单程还是往返
     * @param usr   登录用户
     * @param pw   登录账号
     * @param loginType  登录类型 填 ffp或者mobile
     * @throws MyQueryFlightException 
     * @throws MyPassengerException 
     * @throws MySelectFlightException 
     * @throws LoginException 
     * @throws MyBookException 
     * @throws SecurityException 
     * @throws NoSuchMethodException 
     * @throws Exception 
     */
    
    public String bookCommonFlight(StartUpType type,User user,QueryFlightInfo queryFlightInfo,List<?> paxs,boolean insurance,String testcase) throws MyLoginException,MyQueryFlightException,  MySelectFlightException,MyPassengerException, MyBookException  {
        driver = util.loginStep(type, user);
        util.queryFlightStep(driver, queryFlightInfo);
        util.selectFlightStep(driver, queryFlightInfo);
        util.passengerStep(driver, paxs, insurance);
        String res = util.bookStep(driver, testcase, user);
        return res;
    }
    
    public String bookWsdFlight(StartUpType type,User user,QueryFlightInfo queryFlightInfo,List<?> paxs,List<?> freePaxs,boolean insurance,String testcase) throws MyLoginException,MyQueryFlightException,  MySelectFlightException,MyPassengerException, MyBookException, MyFreePassengerException  {
        driver = util.loginStep(type, user);
        util.queryFlightStep(driver, queryFlightInfo);
        util.selectFlightStep(driver, queryFlightInfo);
        util.passengerStep(driver, paxs, insurance);
        util.freePassengerStep(driver, freePaxs, insurance);
        String res = util.bookStep(driver, testcase, user);
        return res;
    }
    
    public String bookAndIssueCommonTicket(StartUpType type,User user,QueryFlightInfo queryFlightInfo,List<?> paxs,boolean insurance,String payType,String testcase) throws MyLoginException,MyQueryFlightException,  MySelectFlightException,MyPassengerException, MyBookException, MyPaymentException{
        String res = this.bookCommonFlight(type, user, queryFlightInfo, paxs, insurance, testcase);
        if(res.contains("error code:"))
            return res;
        return res + "   "+util.pay(driver, payType,user);
    }
    
    public String bookAndIssueWsdTicket(StartUpType type,User user,QueryFlightInfo queryFlightInfo,List<?> paxs,List<?> freePaxs,boolean insurance,String payType,String testcase) throws MyLoginException,MyQueryFlightException,  MySelectFlightException,MyPassengerException, MyBookException, MyPaymentException, MyFreePassengerException{
        String res = this.bookWsdFlight(type, user, queryFlightInfo, paxs, freePaxs, insurance, testcase);
        if(res.contains("error code:"))
            return res;
        return res + "   "+util.pay(driver, payType,user);
    }
    
    /**
     * 航班动态
     * @param type
     * @param city1
     * @param city2
     * @return
     * @throws MyInvokeException 
     * @throws MyExecuteMethodOutOfTimesException 
     * @throws Exception
     */
    public String dynamicFlight(StartUpType type,String city1 ,String city2) throws MyExecuteMethodOutOfTimesException, MyInvokeException {
        driver = super.startUpBrowse(type);
        String res = new LaputaDynamicAvation().queryDynamicFlight(driver, city1, city2); 
        return util.addcheckResult(res, 17);
    }
    
    
    public String lowPrice(StartUpType type) throws MyExecuteMethodOutOfTimesException, MyInvokeException   {
        driver = super.startUpBrowse(type);
        String res =new LaputaLowPrice().showLowPrice(driver);
        return util.addcheckResult(res, 17);
    }

    /**
     * 实现多线程
     */
    public abstract void run();
    public abstract String execute();
    public void closeCurrentDriver() {
            super.close(this.driver);
    }

}

package org.travelsky.autotest.selenium.service.runable;

import java.util.List;

import org.travelsky.autotest.selenium.config.StartUpType;
import org.travelsky.autotest.selenium.exception.MyBookException;
import org.travelsky.autotest.selenium.exception.MyLoginException;
import org.travelsky.autotest.selenium.exception.MyPassengerException;
import org.travelsky.autotest.selenium.exception.MyPaymentException;
import org.travelsky.autotest.selenium.exception.MyQueryFlightException;
import org.travelsky.autotest.selenium.exception.MySelectFlightException;
import org.travelsky.autotest.selenium.model.laputa.entity.QueryFlightInfo;
import org.travelsky.autotest.selenium.model.laputa.entity.User;
import org.travelsky.autotest.selenium.service.TestCaseService;

public class BookAndIssueCommonTicketCase extends TestCaseService {

    StartUpType type;
    QueryFlightInfo queryFlightInfo;
    User user;
    String testcase;
    List<?> paxs;
    boolean insurance;
    String payType;

    public BookAndIssueCommonTicketCase(StartUpType type, User user, QueryFlightInfo queryFlightInfo, List<?> paxs, boolean insurance, String payType, String testcase) {
        this.type = type;
        this.queryFlightInfo = queryFlightInfo;
        this.user = user;
        this.paxs = paxs;
        this.testcase = testcase;
        this.insurance = insurance;
        this.payType = payType;
    }

    @Override
    public String execute() {
        String res = null;
        try {
            res = super.bookAndIssueCommonTicket(type, user, queryFlightInfo, paxs, insurance, payType, testcase);
        } catch (MyPaymentException e) {
            res = "支付或出票页面报错";
            log.error(res, e);
        } catch (MyQueryFlightException e) {
            res = "查询航班页面报错";
            log.error(res, e);
        } catch (MySelectFlightException e) {
            res = "选择航班页面报错";
            log.error(res, e);
        } catch (MyPassengerException e) {
            res = "旅客资料页面报错";
            log.error(res, e);
        } catch (MyBookException e) {
            res = "预定页面报错";
            log.error(res, e);
        } catch (MyLoginException e) {
            res = "登录页面报错";
            log.error(res, e);
        } finally {
            super.closeCurrentDriver();
        }
        return res;
    }

    @Override
    public void run() {
        try {
            super.bookAndIssueCommonTicket(type, user, queryFlightInfo, paxs, insurance, payType, testcase);
        } catch (Exception e) {
            log.error(this.getClass().getName(), e);
        } finally {
            super.closeCurrentDriver();
        }
    }
}

package org.travelsky.autotest.selenium.model.laputa.impl;

import java.lang.reflect.InvocationTargetException;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.travelsky.autotest.selenium.exception.MyExecuteMethodOutOfTimesException;
import org.travelsky.autotest.selenium.exception.MyInvokeException;
import org.travelsky.autotest.selenium.exception.MyQueryFlightException;
import org.travelsky.autotest.selenium.exception.MySeleniumException;
import org.travelsky.autotest.selenium.model.BaseSeleniumManager;
import org.travelsky.autotest.selenium.model.laputa.QueryFlight;
import org.travelsky.autotest.selenium.model.laputa.entity.QueryFlightInfo;
import org.travelsky.autotest.selenium.model.laputa.util.QueryFlightUtil;

/**
 * 
 * @author sunfan
 * 
 */
public class LaputaQueryFlight extends BaseSeleniumManager implements QueryFlight {

    QueryFlightUtil queryFlightUtil = new QueryFlightUtil(1);

    /**
     * 
     * @param driver
     * @param queryFlight
     * @param int input "1" = "http://easternmiles.ceair.com/flight2014/" or
     *        input "2" = http://easternmiles.ceair.com/flight/";
     * @throws MyExecuteMethodOutOfTimesException
     * @throws MyInvokeException
     * @throws MySeleniumException
     * @throws InterruptedException
     * @throws InstantiationException
     * @throws InvocationTargetException
     * @throws IllegalArgumentException
     * @throws IllegalAccessException
     * @throws SecurityException
     * @throws NoSuchMethodException
     */
    public WebDriver queryFlight(WebDriver driver, QueryFlightInfo queryFlight) throws MyExecuteMethodOutOfTimesException, MyInvokeException {
        try {
            delay(2);
            driver.switchTo().window(driver.getWindowHandle());// 前置当前窗口
            queryFlightUtil.inputQueryInfo(driver, queryFlight);// 输入查询航班内容
            queryFlightUtil.checkJF(driver, queryFlight);// 判断是否需要点击积分兑换按钮
            delay(1);
            driver.findElement(By.id("btn_member_search")).click();// 点击查询按钮
            delay(2);
            super.AcceptAlertAndGetItsText(driver);// 点击可能出现的弹出框
            queryFlightUtil.confirmSuccess(driver);// 确认是否成功
        } catch (Exception e) {
            try {
                super.runBack(
                        this.getClass().getMethod("queryFlight", org.openqa.selenium.WebDriver.class,
                                org.travelsky.autotest.selenium.model.laputa.entity.QueryFlightInfo.class), e, driver, queryFlight);
            } catch (NoSuchMethodException | SecurityException e1) {
                log.info("反射获取login()queryFlight", e1);
            }
        }
        return driver;
    }

}

package org.travelsky.autotest.config.spring;

import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.web.servlet.config.annotation.EnableWebMvc;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;

@Configuration
@ComponentScan(basePackages="org.travelsky.autotest.*")
@EnableWebMvc
@EnableScheduling
public class WebConfig extends WebMvcConfigurerAdapter{
    
    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        registry.addResourceHandler("/resource/**").addResourceLocations("/resource/");
    }
}

package org.travelsky.autotest.config.spring;

import java.util.EnumSet;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.DispatcherType;
import javax.servlet.FilterRegistration;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletRegistration;

import org.springframework.web.WebApplicationInitializer;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
import org.springframework.web.filter.CharacterEncodingFilter;
import org.springframework.web.servlet.DispatcherServlet;

public class WebInit implements WebApplicationInitializer {
    @Override
    public void onStartup(ServletContext sc) throws ServletException {
        // Create the ‘root‘ Spring application context
        AnnotationConfigWebApplicationContext application = new AnnotationConfigWebApplicationContext();
        application.scan("org.travelsky.autotest.config.spring");
        // Manages the lifecycle of the root application context
        sc.addListener(new ContextLoaderListener(application));
        //CharacterEncodingFilter 
        CharacterEncodingFilter characterEncodingFilter = new CharacterEncodingFilter();
        characterEncodingFilter.setEncoding("UTF-8");
        characterEncodingFilter.setForceEncoding(true);
        FilterRegistration filterRegistration = sc.addFilter("characterEncodingFilter", characterEncodingFilter);
        filterRegistration.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), false, "/*");

        AnnotationConfigWebApplicationContext webContext = new AnnotationConfigWebApplicationContext();
        ServletRegistration.Dynamic appServlet = sc.addServlet("appServlet", new DispatcherServlet(webContext));
        appServlet.setLoadOnStartup(1);
        appServlet.addMapping("/");
    }

}