java-testng-selenium优化

由于项目中webui测试的需要，是用testng+selenium的方式，其中遇到过几个问题，记录下，方便以后查看

1.重复运行多次case

因为是selenium，所以有的时候需要运行多次，方法是写一个Retry的类，继承testng的retry类就可以。

public class Retry implements IRetryAnalyzer {
    private int retryCount = 0;
    private int maxRetryCount = 3;

    public boolean retry(ITestResult result) {

        if (retryCount < maxRetryCount) {
            retryCount++;
            System.out.println(String.format("test failed!try %d time!",retryCount+1));
            return true;
        }
        return false;
    }
运行的时候就在case上加上
@Test(retryAnalyzer=Retry.class)
就可以拉

2.自定义annotions

虽然testng自带有很多注释，很好用，但是在项目中，组长要求在每个case前面能够描述一下这个case的作用，以后当case失败的时候，可以输出这个case的作用和失败原因，要外人不看代码也可以看的懂，所以就想自定义一个annotation，添加个decsription属性。

 //UIMessage.java
 import static com.myhexin.common.Assert.AssertEquals;

 import java.lang.annotation.Retention;
 import java.lang.annotation.RetentionPolicy;
 import java.lang.annotation.Target;
 import java.lang.annotation.ElementType;

 import org.testng.annotations.Parameters;
 import org.testng.annotations.Test;
 @Retention(RetentionPolicy.RUNTIME)
 @Target({ElementType.METHOD})
 public @interface UIMessage {
      String description();
 }

 //应用
 @UIMessage(description = "我是来试试的")@Test(retryAnalyzer=Retry.class)
 @Parameters()
 public void testDengLuZhangHao(){
 /*
  * 函数体
  */
 }
 //如何获取annotation的注释消息,我是用反射来获取的

 Method[] methods=Class.forName("core.Member").getDeclaredMethods();
             for(Method method:methods)
            {
                  if(method.isAnnotationPresent( UIMessage. class))
                 {
                      System. out.println(method.getAnnotation( UIMessage. class).desciption());
                 }
            }

3.selenium截图

目前该截图方式当在remote执行selenium时候，我只是在firefox下成功过，ie打死不行，场景是当case运行失败的时候截图，主要是重写下assert，当断言失败时候，截图，直接上，截图的代码

    public  void screenShot(WebDriver driver,int i)
    {
            String dir_name="./pic";
          if (!(new File(dir_name).isDirectory())) {  // 判断是否存在该目录
             new File(dir_name).mkdir();  // 如果不存在则新建一个目录
          }

          SimpleDateFormat sdf = new SimpleDateFormat("-HHmmss");
          String time = sdf.format(new Date());
         WebDriver augmentedDriver = new Augmenter().augment(driver);

          try {
             File screenshot = ((TakesScreenshot)augmentedDriver).getScreenshotAs(OutputType.FILE); // 关键代码，执行屏幕截图，默认会把截图保存到temp目录
             FileUtils.copyFile(screenshot, new File(dir_name + File.separator + i+ time  + ".png"));  // 这里将截图另存到我们需要保存
          } catch (IOException e) {
             e.printStackTrace();
          }
    }