mirror of
https://gitee.com/hhyykk/ipms-sjy.git
synced 2025-07-23 15:35:06 +08:00
增加 podam 随机生成对象,简化单元测试的成本
This commit is contained in:
@ -1,9 +1,14 @@
|
||||
package cn.iocoder.dashboard.util;
|
||||
|
||||
import cn.hutool.core.util.*;
|
||||
import cn.hutool.core.util.ArrayUtil;
|
||||
import cn.hutool.core.util.RandomUtil;
|
||||
import cn.iocoder.dashboard.modules.system.dal.dataobject.user.SysUserDO;
|
||||
import uk.co.jemos.podam.api.PodamFactory;
|
||||
import uk.co.jemos.podam.api.PodamFactoryImpl;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.Arrays;
|
||||
import java.util.Date;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
/**
|
||||
* 随机工具类
|
||||
@ -14,6 +19,10 @@ public class RandomUtils {
|
||||
|
||||
private static final int RANDOM_STRING_LENGTH = 10;
|
||||
|
||||
private static final int RANDOM_DATE_MAX = 30;
|
||||
|
||||
private static final PodamFactory PODAM_FACTORY = new PodamFactoryImpl();
|
||||
|
||||
public static String randomString() {
|
||||
return RandomUtil.randomString(RANDOM_STRING_LENGTH);
|
||||
}
|
||||
@ -26,41 +35,27 @@ public class RandomUtils {
|
||||
return RandomUtil.randomInt(0, Integer.MAX_VALUE);
|
||||
}
|
||||
|
||||
public static Date randomDate() {
|
||||
return RandomUtil.randomDay(0, RANDOM_DATE_MAX);
|
||||
}
|
||||
|
||||
public static Short randomShort() {
|
||||
return (short) RandomUtil.randomInt(0, Short.MAX_VALUE);
|
||||
}
|
||||
|
||||
public static SysUserDO randomUserDO() {
|
||||
SysUserDO user = randomObject(SysUserDO.class);
|
||||
return user;
|
||||
@SafeVarargs
|
||||
public static SysUserDO randomUserDO(Consumer<SysUserDO>... consumers) {
|
||||
return randomPojo(SysUserDO.class, consumers);
|
||||
}
|
||||
|
||||
private static <T> T randomObject(Class<T> clazz) {
|
||||
// 创建对象
|
||||
T object = ReflectUtil.newInstance(clazz);
|
||||
// 遍历属性,设置随机值
|
||||
for (Field field : ReflectUtil.getFields(clazz)) {
|
||||
// 数字类型
|
||||
if (field.getType() == Long.class) {
|
||||
ReflectUtil.setFieldValue(object, field, randomLong());
|
||||
continue;
|
||||
}
|
||||
if (field.getType() == Integer.class) {
|
||||
ReflectUtil.setFieldValue(object, field, randomInteger());
|
||||
continue;
|
||||
}
|
||||
if (field.getType() == Short.class) {
|
||||
ReflectUtil.setFieldValue(object, field, randomShort());
|
||||
continue;
|
||||
}
|
||||
// 字符串类型
|
||||
if (field.getType() == String.class) {
|
||||
ReflectUtil.setFieldValue(object, field, randomString());
|
||||
continue;
|
||||
}
|
||||
// System.out.println();
|
||||
@SafeVarargs
|
||||
private static <T> T randomPojo(Class<T> clazz, Consumer<T>... consumers) {
|
||||
T pojo = PODAM_FACTORY.manufacturePojo(clazz);
|
||||
// 非空时,回调逻辑。通过它,可以实现 Pojo 的进一步处理
|
||||
if (ArrayUtil.isNotEmpty(consumers)) {
|
||||
Arrays.stream(consumers).forEach(consumer -> consumer.accept(pojo));
|
||||
}
|
||||
return object;
|
||||
return pojo;
|
||||
}
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user