一、安装JUnitGenerator V2.0。
1、通过网络安装。如下
2、手动下载插件安装。插件下载：https://plugins.jetbrains.com/idea/plugin/3064-junitgenerator-v2-0 。安装如下图所示：
3、安装完成重启IDEA，即可使用。
二、修改JUnitGenerator V2.0的配置。
1、自动生成测试代码和java类在同一包下，不匹配maven项目标准测试目录。
修改Output Path为：${SOURCEPATH}/../../test/java/${PACKAGE}/${FILENAME}，
Default Template选择JUnit 4。
2、修改测试用例模板。模板中生成的package的包名需去掉test。
三、生成JUnit4测试用例。
方法一：在待编写测试的java类源码块上按快捷键Alt + Insert。选择JUnit Test->JUnit 4。
方法二、在待编写测试的java类源码块上按快捷键Ctrl + Shift + T。
根据需要修改相关测试用例
方法三：光标定位到待编写测试的java类源码块，选择code->Generate，后面的步骤和方法1一样。
最终生成的用例如下：
四、注意。
IDEA自带的JUnit插件和JUnitGeneratorV2.0插件都要勾选上，若只勾选JUnit可能导致无法自动生成测试文件，
若只勾选JUnitGenerator V2.0可能导致生成的测试文件无法运行
实体类junit测试用例生成
,,,
package com.wms;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.Map;
/**
* Created by Administrator on 2017/4/8.
*/
public class Test {
private static final StringBuilder sb = new StringBuilder();
public static void writeClass(String classStr, OutputStreamWriter out) {
try {
Class clz = Class.forName(classStr);
// Method[] methods = clz.getDeclaredMethods();
// for (Method method : methods) {
// String methodName = method.getName();
// Class[] params = method.getParameterTypes();
// for (Class param : params) {
// System.out.println(param.getName());
// }
// }
Field[] fields = clz.getDeclaredFields();
for (Field field : fields) {
Class c = field.getType();
// System.out.println(c.isInterface());
// System.out.println(c.getName());
// Class cc = Class.forName(c.getName());
// Object o = cc.newInstance();
//只写入set方法即可
out.write(setMethod(classStr, getParam(c, field.getName()), field.getName()));
}
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
public static Map
Map
boolean isReference = false;
String result = "";
String className = clz.getName();
String clzName = className;
if (className.contains(".")) {
clzName = className.substring(className.lastIndexOf(".") + 1);
}
clearSb(sb);
if (clz.isInterface()) {
result = null;
} else if (className.startsWith("[L")) {
clzName = clzName.replaceAll(";", "");
sb.append(clzName).append("[] o = ").append("new ").append(clzName).append("[]{};");
result = sb.toString();
isReference = true;
} else if (clz == String.class) {
sb.append("\"").append(field).append("\"");
result = sb.toString();
} else if (clz == Integer.class || clz == int.class) {
result = "1";
} else if (clz == Long.class || clz == long.class) {
result = "1L";
} else if (clz == Float.class || clz == float.class) {
result = "1.1f";
} else if (clz == Double.class || clz == double.class) {
result = "1.1";
} else if (clz == Short.class || clz == short.class) {
result = "(short)1";
} else if (clz == Boolean.class || clz == boolean.class) {
result = "true";
} else if (clz == char.class || clz == Character.class) {
result = "\'1\'";
} else if (clz == byte.class || clz == Byte.class) {
result = "(byte)1";
} else {
sb.append(clzName).append(" o = ").append("new ").append(clzName).append("();");
result = sb.toString();
isReference = true;
}
map.put("result", result);
map.put("isReference", isReference);
return map;
}
public static String setMethod(String classStr, Map
clearSb(sb);
classStr = classStr.substring(classStr.lastIndexOf(".") + 1);
boolean isReference = (Boolean) map.get("isReference");
String result = (String) map.get("result");
String str = firstLowerToUpper(field);
sb.append("\t");
sb.append("@Test").append("\n");
sb.append("\t");
sb.append("public void testSet").append(str).append("()").append("\n");
sb.append("\t");
sb.append("{").append("\n");
sb.append("\t").append("\t");
sb.append(classStr).append(" obj = new ").append(classStr).append("();").append("\n");
sb.append("\t").append("\t");
if (isReference) {
sb.append(result).append("\n");
result = "o";
sb.append("\t").append("\t");
}
sb.append("obj.set").append(str).append("(").append(result).append(");").append("\n");
sb.append("\t").append("\t");
sb.append("assertTrue(").append(result).append(" == obj.get").append(str).append("());")
.append("\n");
sb.append("\t");
sb.append("}").append("\n").append("\n");
return sb.toString();
}
public static String firstLowerToUpper(String field) {
return field.substring(0, 1).toUpperCase() + field.substring(1);
}
public static void createTestFile(String projectPath, String[] pkgs, boolean rebuild) {
OutputStreamWriter out = null;
for (String pkg : pkgs) {
String testPkgPath = checkTestPackage(projectPath, pkg);
String srcPkgPath = getJavaPackagePath(projectPath, pkg);
File file = new File(srcPkgPath);
for (File file1 : file.listFiles()) {
String className = getClassName(file1.getName());
File testFile = new File(getTestFile(testPkgPath, className));
if (!testFile.exists() || rebuild) {
try {
//0. 创建测试用例文件
testFile.createNewFile();
//1. 写入测试用例类的包名
out = new FileWriter(testFile);
out.write(packageLine(pkg));
//2. 写入junit测试用例用到的包
out.write(importJunitClassLine());
//3. 写入测试用例类名
out.write(startClassNameLine(className + "Test"));
//4. before,after方法写入
// out.write(beforeMethod());
// out.write(afterMethod());
//5. 写入测试方法
writeClass(getClass(pkg, className), out);
// 类结束
out.write(endClassNameLine());
out.flush();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (out != null) {
try {
out.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
}
}
/**
* 检查测试类路径下的指定包是否存在，不存在则创建
*
* @param projectPath
* @param pkg
* @return
*/
public static String checkTestPackage(String projectPath, String pkg) {
String dstPath = getTestPackagePath(projectPath, pkg);
File f = new File(dstPath);
if (!f.exists()) {
f.mkdirs();
}
return dstPath;
}
/**
* 获取测试类的全路径名
*
* @param dstPackage
* @param className
* @return
*/
public static String getTestFile(String dstPackage, String className) {
clearSb(sb);
sb.append(dstPackage).append(File.separator).append(className).append("Test.java");
return sb.toString();
}
/**
* 包声明行
*
* @param pkg
* @return
*/
public static String packageLine(String pkg) {
clearSb(sb);
sb.append("package ").append(pkg).append(";");
sb.append("\n").append("\n");
return sb.toString();
}
/**
* junit类导入行
*
* @return
*/
public static String importJunitClassLine() {
clearSb(sb);
sb.append("import org.junit.Test;").append("\n");
sb.append("import org.junit.Before;").append("\n");
sb.append("import org.junit.After;").append("\n").append("\n").append("\n");
sb.append("import static org.junit.Assert.*;").append("\n").append("\n");
return sb.toString();
}
/**
* 类开始
*
* @param fileName
* @return
*/
public static String startClassNameLine(String className) {
clearSb(sb);
sb.append("public class ").append(className).append("\n");
sb.append("{").append("\n");
return sb.toString();
}
/**
* 类结束
*
* @return
*/
public static String endClassNameLine() {
return "\n}";
}
/**
* beforeMethod
*
* @return
*/
public static String beforeMethod() {
clearSb(sb);
sb.append("\n").append("\t").append("@Before").append("\n");
sb.append("\t").append("public void before()").append("\n");
sb.append("\t").append("{").append("\n");
sb.append("\t").append("}").append("\n");
return sb.toString();
}
/**
* afterMethod
*
* @return
*/
public static String afterMethod() {
clearSb(sb);
sb.append("\n").append("\t").append("@After").append("\n");
sb.append("\t").append("public void after()").append("\n");
sb.append("\t").append("{").append("\n");
sb.append("\t").append("}").append("\n");
return sb.toString();
}
/**
* 获取类全名，含有包
*
* @param pkg
* @param className
* @return
*/
public static String getClass(String pkg, String className) {
return pkg + "." + className;
}
/**
* 获取类名，不含包
*
* @param fileName
* @return
*/
public static String getClassName(String fileName) {
return fileName.replace(".java", "");
}
/**
* 获取java源文件路径下的指定包
*
* @param projectPath
* @param pkg
* @return
*/
public static String getJavaPackagePath(String projectPath, String pkg) {
String packagePath = pkg.replaceAll("\\.", "/");
return projectPath + "/src/main/java" + File.separator + packagePath;
}
/**
* 获取java测试类路径下的指定包
*
* @param projectPath
* @param pkg
* @return
*/
public static String getTestPackagePath(String projectPath, String pkg) {
String packagePath = pkg.replaceAll("\\.", "/");
return projectPath + "/src/test/java" + File.separator + packagePath;
}
/**
* 清空StringBuilder
*
* @param sb
*/
public static void clearSb(StringBuilder sb) {
sb.delete(0, sb.length());
}
public static void main(String[] args) throws IOException {
String projectPath = "D:/data/IDEA workspace/wms-web";
String[] pkgs = new String[]{"com.wms.junit"};
createTestFile(projectPath, pkgs, true);
//loadClass("com.wms.junit.Person", null);
}
}
,,,