| 提交 | 用户 | 时间 | ||
| e7c126 | 1 | package com.iailab.framework.test.core.util; |
| H | 2 | |
| 3 | import cn.hutool.core.util.ArrayUtil; | |
| 4 | import cn.hutool.core.util.ReflectUtil; | |
| 5 | import com.iailab.framework.common.exception.ErrorCode; | |
| 6 | import com.iailab.framework.common.exception.ServiceException; | |
| 7 | import com.iailab.framework.common.exception.util.ServiceExceptionUtil; | |
| 8 | import org.junit.jupiter.api.Assertions; | |
| 9 | import org.junit.jupiter.api.function.Executable; | |
| 10 | ||
| 11 | import java.lang.reflect.Field; | |
| 12 | import java.util.Arrays; | |
| 13 | import java.util.Objects; | |
| 14 | ||
| 15 | import static org.junit.jupiter.api.Assertions.assertThrows; | |
| 16 | ||
| 17 | /** | |
| 18 | * 单元测试，assert 断言工具类 | |
| 19 | * | |
| 20 | * @author iailab | |
| 21 | */ | |
| 22 | public class AssertUtils { | |
| 23 | ||
| 24 | /** | |
| 25 | * 比对两个对象的属性是否一致 | |
| 26 | * | |
| 27 | * 注意，如果 expected 存在的属性，actual 不存在的时候，会进行忽略 | |
| 28 | * | |
| 29 | * @param expected 期望对象 | |
| 30 | * @param actual 实际对象 | |
| 31 | * @param ignoreFields 忽略的属性数组 | |
| 32 | */ | |
| 33 | public static void assertPojoEquals(Object expected, Object actual, String... ignoreFields) { | |
| 34 | Field[] expectedFields = ReflectUtil.getFields(expected.getClass()); | |
| 35 | Arrays.stream(expectedFields).forEach(expectedField -> { | |
| 36 | // 忽略 jacoco 自动生成的 $jacocoData 属性的情况 | |
| 37 | if (expectedField.isSynthetic()) { | |
| 38 | return; | |
| 39 | } | |
| 40 | // 如果是忽略的属性，则不进行比对 | |
| 41 | if (ArrayUtil.contains(ignoreFields, expectedField.getName())) { | |
| 42 | return; | |
| 43 | } | |
| 44 | // 忽略不存在的属性 | |
| 45 | Field actualField = ReflectUtil.getField(actual.getClass(), expectedField.getName()); | |
| 46 | if (actualField == null) { | |
| 47 | return; | |
| 48 | } | |
| 49 | // 比对 | |
| 50 | Assertions.assertEquals( | |
| 51 | ReflectUtil.getFieldValue(expected, expectedField), | |
| 52 | ReflectUtil.getFieldValue(actual, actualField), | |
| 53 | String.format("Field(%s) 不匹配", expectedField.getName()) | |
| 54 | ); | |
| 55 | }); | |
| 56 | } | |
| 57 | ||
| 58 | /** | |
| 59 | * 比对两个对象的属性是否一致 | |
| 60 | * | |
| 61 | * 注意，如果 expected 存在的属性，actual 不存在的时候，会进行忽略 | |
| 62 | * | |
| 63 | * @param expected 期望对象 | |
| 64 | * @param actual 实际对象 | |
| 65 | * @param ignoreFields 忽略的属性数组 | |
| 66 | * @return 是否一致 | |
| 67 | */ | |
| 68 | public static boolean isPojoEquals(Object expected, Object actual, String... ignoreFields) { | |
| 69 | Field[] expectedFields = ReflectUtil.getFields(expected.getClass()); | |
| 70 | return Arrays.stream(expectedFields).allMatch(expectedField -> { | |
| 71 | // 如果是忽略的属性，则不进行比对 | |
| 72 | if (ArrayUtil.contains(ignoreFields, expectedField.getName())) { | |
| 73 | return true; | |
| 74 | } | |
| 75 | // 忽略不存在的属性 | |
| 76 | Field actualField = ReflectUtil.getField(actual.getClass(), expectedField.getName()); | |
| 77 | if (actualField == null) { | |
| 78 | return true; | |
| 79 | } | |
| 80 | return Objects.equals(ReflectUtil.getFieldValue(expected, expectedField), | |
| 81 | ReflectUtil.getFieldValue(actual, actualField)); | |
| 82 | }); | |
| 83 | } | |
| 84 | ||
| 85 | /** | |
| 86 | * 执行方法，校验抛出的 Service 是否符合条件 | |
| 87 | * | |
| 88 | * @param executable 业务异常 | |
| 89 | * @param errorCode 错误码对象 | |
| 90 | * @param messageParams 消息参数 | |
| 91 | */ | |
| 92 | public static void assertServiceException(Executable executable, ErrorCode errorCode, Object... messageParams) { | |
| 93 | // 调用方法 | |
| 94 | ServiceException serviceException = assertThrows(ServiceException.class, executable); | |
| 95 | // 校验错误码 | |
| 96 | Assertions.assertEquals(errorCode.getCode(), serviceException.getCode(), "错误码不匹配"); | |
| 97 | String message = ServiceExceptionUtil.doFormat(errorCode.getCode(), errorCode.getMsg(), messageParams); | |
| 98 | Assertions.assertEquals(message, serviceException.getMessage(), "错误提示不匹配"); | |
| 99 | } | |
| 100 | ||
| 101 | } | |
