干货！使用Java注解和反射实现Junit4中的用例调用（附源码）

前面我写了关于Java注解和反射的文章：

测试开发必须掌握的知识点：Java反射

测试开发必须掌握的重要知识点-Spring的核心知识点 -Java注解@

 

大家可能会问，这些高级的知识点在实际工作中有哪些应用呢？在这里我们来结合java注解和反射的知识做一个实际案例的编写，让大家对其有更深刻的理解！直接上实例！

实例需求

需求：模拟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());

}

}

}

运行代码，输出如下：

logout

测试用例名称是：logout

login

测试用例名称是：login

info

测试用例名称是：info

这个就是一个模拟Junit4执行用例的demo，当我们随意添加TestTag 中的测试用例时，执行runCase方法，都能够成功执行测试用例。相信通过这个例子大家一定会注解，和反射的概念有了更为深刻的理解！我们的实例目前只支持对于指定某一个类中的测试方法进行调用，大家可以考虑如何实现对于某一个包下所有类的测试用例都可以进行调用呢？