如果别人的一个代码,需要你去测试,但是方法特别多,一个一个测试会非常不方便,这个时候使用注解。
例如:小明定义了一个计算机类,明显看出有错误,(假设方法特别多的时候)我们不可能一眼看出错误
我们给需要测试的方法加上注解 @Check:
package cn.itcast.exercise_01;
/**
* 小明定义的计算器类
*/
public class Calculator {
//加法
@Check
public void add(){
// String str = null;
// str.toString();
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 void show(){
System.out.println("永无bug...");
}
}
新建一个注解:
package cn.itcast.exercise_01;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Check {
}
对自定义的注解进行测试:
要求:*当主方法执行后,会自动进行被检测的所有方法(加了Check()注解的方法),判断方法是否有异常,记录到文件中
package cn.itcast.exercise_01;
/**
* 简单的测试框架
*
* 当主方法执行后,会自动进行被检测的所有方法(加了Check()注解的方法),判断方法是否有异常,记录到文件中
*/
import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
public class CheckTest {
public static void main(String[] args) throws IOException {
//1.创建计算器对象
Calculator c = new Calculator();
//2.获取字节码文件对象
Class aclass= c.getClass();
//3.获取所有方法
Method[] methods = aclass.getMethods();
//出现异常的次数
int num = 0;
//生成文档
BufferedWriter bw = new BufferedWriter(new FileWriter("bug.txt"));
for (Method method : methods) {
//判断方法是否是Check()注解
if (method.isAnnotationPresent(Check.class)){
//有,则执行。 捕获异常
try {
method.invoke(c);
} catch (Exception e) {
num++;
//记录到文件中
bw.write(method.getName()+"方法异常了");
bw.newLine();
bw.write("异常的名称:"+e.getCause().getClass().getSimpleName());
bw.newLine();
bw.write("异常的原因:"+e.getCause().getMessage());
bw.newLine();
}
}
}
bw.write("本次一共出现"+num+"次异常");
bw.flush();
bw.close();
}
}