怎么使用Java注解和反射实现Junit4调用

WBOY转载: 2023-05-04 13:37:06386浏览

实例需求

需求：模拟Junit4中的用例结构，自定义一个标签作为测试用例的标识。

在Junit4中注解@Test表示测试用例，每一个测试用例的本质就是测试类中的一个方法，即：

@Test public void test() { fail("Not yet implemented"); }

具体要求：

测试类是默认构造方法

通过标签MyTag作为方法是否为用例的标识

作为用例的方法必须是无参的

写一个方法runCase(String pkgName)，使其能够完成对指定类中测试用例的调用工作

注解设计

首先需要定义一个注解MyTag，代码如下：

import java.lang.annotation.Retention; import java.lang.annotation.Target; import java.lang.annotation.ElementType; import java.lang.annotation.RetentionPolicy; @Retention(RetentionPolicy.RUNTIME) @Target(ElementType.METHOD) public @interface MyTag{ String name(); }

测试用例设计

模拟junit4框架，写一个测试类，其中login、info、logout是无参方法，而test是有参方法

public class TestTag{ @MyTag(name="case1") public void login(){ System.out.println("login"); } @MyTag(name="case2") public void info(){ System.out.println("info");  } @MyTag(name="case3") public void logout(){ System.out.println("logout"); } @MyTag(name="case4") public void test(String p){ System.out.println("logout");  } }

运行类的设计思路

因为测试类是默认构造方法，所以使用如下API，完成类实例化

Class<?> clazz = Class.forName(pkgName) Object obj = clazz.newInstance();

因为测试类中有很多方法，我们需要获取所有的方法并按照规则进行过滤，代码如下：

Method[] methods = clazz.getMethods(); for (Method method : methods) { //过滤规则  }

判断方法的标签是否为MyTag，代码如下：

If（method.getAnnotation(MyTag.class) != null）

判断方法是否没有参数，代码如下：

If（method.getParameterCount()==0)

运行方法，代码如下：

method.invoke(obj)

完整代码

public static void runCase(String pkgName) throws IllegalAccessException,IllegalArgumentException, InvocationTargetException, InstantiationException,ClassNotFoundException {  Class<?> clazz = Class.forName(pkgName);  Object obj = clazz.newInstance();  Method[] methods = clazz.getMethods(); for (Method method : methods) {        if(method.getAnnotation(MyTag.class) != null&& method.getParameterCount()==0) {                                 method.invoke(obj); //调用方法                System.out.println("测试用例名称是："+method.getName()); } } }

运行代码，输出如下：