Java中的反射可以获取到Class,基于Class可以得到一些基本的元数据,根据模版文件生成单元测试类。最难的是根据方法中的分支条件生成单元测试内容,需要解析出方法调用栈、导致方法返回的条件语句生成对应测试语句。
使用javaparser或者JCTree 获取方法调用栈
描述类信息
ImportDescriptor 引入类包描述,包含单元测试、成员变量、方法内使用到的类型
UnitTestImportDescriptor
FieldImportDecriptor
MethodImportDescriptor
FieldDescriptor 类成员变量描述
MethodDescriptor 类方法描述
ClassDescriptor 类描述
PackageDescriptor 类包名描述
MethodCallDescriptor 方法调用栈描述,JCTree获取 JavaParser不知道是否可获取
查找生成测试用例的类
ClassResource 查找class
ClassPathResource 在ClassPath中查找class
FilePathResource 在文件路径中查找class
JarResource 在Jar包中查找class
InputStreamResource 在输入流中查找class
工具类
FileUtil 创建文件及目录
ClassLoadUtil 类加载
单元测试模版文件(Freemarker):生成的只是基本内容,可以根据实际情况修改成符合自己规范的模版。
package ${package};
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import ${package}.${className};
<#list imports as import>
import ${import};
</#list>
/**
* @author ${author}
* @version 1.0
* @description: TODO
* @date Create by ${author} on ${date}
*/
@RunWith(MockitoJUnitRunner.class)
public class ${className}Test {
@InjectMocks
private ${className} ${className?uncap_first};
<#list declaredFields as field>
@Mock
private ${field.type.simpleName} ${field.name};
</#list>
@Before
public void before() throws Exception {
}
@After
public void after() throws Exception {
}
<#list methods as method>
/**
* @return void
* @description TODO
* @author ${author}
* @date ${date}
*/
@Test
public void test${method.name?cap_first}() {
}
</#list>
}