Java 使用注解进行简单的测试框架

简单的测试框架

• 要求：
  当主方法执行，会自动执行被检查的所有方法（加上check注解的方法）
  判断方法是否有异常，然后记录到文件中

注解Check.java

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Check {
}

计算器类Calculator

public class Calculator {

    @Check
    public void add(){//加
        System.out.println("1 + 0 ="+(1+0));
    }

    @Check
    public void sub(){//减
        System.out.println("1 - 0 ="+(1-0));
    }

    @Check
    public void mul(){//乘
        System.out.println("1 * 0 ="+(1*0));
    }

    @Check
    public void div(){//除
        System.out.println("1 / 0 ="+(1/0));
    }
}

测试类

public class TestCheck {
    public static void main(String[] args) throws IOException {
        //1.创建计算器对象
        Calculator c = new Calculator();
        //2.获取字节码文件对象：期望获取所有的方法
        Class cls = c.getClass();
        //3.获取所有方法
        Method[] methods = cls.getMethods();

        int number = 0;//异常出现的次数
        BufferedWriter bw = new BufferedWriter(new FileWriter("bug.txt"));

        for (Method method : methods) {
            if (method.isAnnotationPresent(Check.class)){
                //5.有就执行
                try {
                    method.invoke(c);
                } catch (Exception e) {
                    //6.捕获异常记录到文件中
                    number++;
                    bw.write(method.getName()+"方法出异常了");
                    bw.newLine();
                    bw.write("异常的名称"+e.getCause().getClass().getSimpleName());
                    bw.newLine();
                    bw.write("异常的原因"+e.getCause().getMessage());
                    bw.newLine();
                    bw.write("---------------------");
                }
            }


        }

        bw.write("本次测试一个出现"+number+"次异常");
        bw.flush();
        bw.close();
    }
}

结果

div方法出异常了
异常的名称ArithmeticException
异常的原因/ by zero
---------------------本次测试一个出现1次异常