package com.iailab.module.system.service.sms;
|
|
import cn.hutool.core.map.MapUtil;
|
import com.iailab.framework.common.enums.UserTypeEnum;
|
import com.iailab.framework.common.pojo.PageResult;
|
import com.iailab.framework.common.util.collection.ArrayUtils;
|
import com.iailab.framework.test.core.ut.BaseDbUnitTest;
|
import com.iailab.module.system.controller.admin.sms.vo.log.SmsLogPageReqVO;
|
import com.iailab.module.system.dal.dataobject.sms.SmsLogDO;
|
import com.iailab.module.system.dal.dataobject.sms.SmsTemplateDO;
|
import com.iailab.module.system.dal.mysql.sms.SmsLogMapper;
|
import com.iailab.module.system.enums.sms.SmsReceiveStatusEnum;
|
import com.iailab.module.system.enums.sms.SmsSendStatusEnum;
|
import com.iailab.module.system.enums.sms.SmsTemplateTypeEnum;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.context.annotation.Import;
|
|
import javax.annotation.Resource;
|
import java.time.LocalDateTime;
|
import java.util.Map;
|
import java.util.function.Consumer;
|
|
import static cn.hutool.core.util.RandomUtil.randomBoolean;
|
import static cn.hutool.core.util.RandomUtil.randomEle;
|
import static com.iailab.framework.common.util.date.LocalDateTimeUtils.buildBetweenTime;
|
import static com.iailab.framework.common.util.date.LocalDateTimeUtils.buildTime;
|
import static com.iailab.framework.common.util.object.ObjectUtils.cloneIgnoreId;
|
import static com.iailab.framework.test.core.util.AssertUtils.assertPojoEquals;
|
import static com.iailab.framework.test.core.util.RandomUtils.*;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
|
@Import(SmsLogServiceImpl.class)
|
public class SmsLogServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private SmsLogServiceImpl smsLogService;
|
|
@Resource
|
private SmsLogMapper smsLogMapper;
|
|
@Test
|
public void testGetSmsLogPage() {
|
// mock 数据
|
SmsLogDO dbSmsLog = randomSmsLogDO(o -> { // 等会查询到
|
o.setChannelId(1L);
|
o.setTemplateId(10L);
|
o.setMobile("15601691300");
|
o.setSendStatus(SmsSendStatusEnum.INIT.getStatus());
|
o.setSendTime(buildTime(2020, 11, 11));
|
o.setReceiveStatus(SmsReceiveStatusEnum.INIT.getStatus());
|
o.setReceiveTime(buildTime(2021, 11, 11));
|
});
|
smsLogMapper.insert(dbSmsLog);
|
// 测试 channelId 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setChannelId(2L)));
|
// 测试 templateId 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setTemplateId(20L)));
|
// 测试 mobile 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setMobile("18818260999")));
|
// 测试 sendStatus 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setSendStatus(SmsSendStatusEnum.IGNORE.getStatus())));
|
// 测试 sendTime 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setSendTime(buildTime(2020, 12, 12))));
|
// 测试 receiveStatus 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setReceiveStatus(SmsReceiveStatusEnum.SUCCESS.getStatus())));
|
// 测试 receiveTime 不匹配
|
smsLogMapper.insert(cloneIgnoreId(dbSmsLog, o -> o.setReceiveTime(buildTime(2021, 12, 12))));
|
// 准备参数
|
SmsLogPageReqVO reqVO = new SmsLogPageReqVO();
|
reqVO.setChannelId(1L);
|
reqVO.setTemplateId(10L);
|
reqVO.setMobile("156");
|
reqVO.setSendStatus(SmsSendStatusEnum.INIT.getStatus());
|
reqVO.setSendTime(buildBetweenTime(2020, 11, 1, 2020, 11, 30));
|
reqVO.setReceiveStatus(SmsReceiveStatusEnum.INIT.getStatus());
|
reqVO.setReceiveTime(buildBetweenTime(2021, 11, 1, 2021, 11, 30));
|
|
// 调用
|
PageResult<SmsLogDO> pageResult = smsLogService.getSmsLogPage(reqVO);
|
// 断言
|
assertEquals(1, pageResult.getTotal());
|
assertEquals(1, pageResult.getList().size());
|
assertPojoEquals(dbSmsLog, pageResult.getList().get(0));
|
}
|
|
@Test
|
public void testCreateSmsLog() {
|
// 准备参数
|
String mobile = randomString();
|
Long userId = randomLongId();
|
Integer userType = randomEle(UserTypeEnum.values()).getValue();
|
Boolean isSend = randomBoolean();
|
SmsTemplateDO templateDO = randomPojo(SmsTemplateDO.class,
|
o -> o.setType(randomEle(SmsTemplateTypeEnum.values()).getType()));
|
String templateContent = randomString();
|
Map<String, Object> templateParams = randomTemplateParams();
|
// mock 方法
|
|
// 调用
|
Long logId = smsLogService.createSmsLog(mobile, userId, userType, isSend,
|
templateDO, templateContent, templateParams);
|
// 断言
|
SmsLogDO logDO = smsLogMapper.selectById(logId);
|
assertEquals(isSend ? SmsSendStatusEnum.INIT.getStatus() : SmsSendStatusEnum.IGNORE.getStatus(),
|
logDO.getSendStatus());
|
assertEquals(mobile, logDO.getMobile());
|
assertEquals(userType, logDO.getUserType());
|
assertEquals(userId, logDO.getUserId());
|
assertEquals(templateDO.getId(), logDO.getTemplateId());
|
assertEquals(templateDO.getCode(), logDO.getTemplateCode());
|
assertEquals(templateDO.getType(), logDO.getTemplateType());
|
assertEquals(templateDO.getChannelId(), logDO.getChannelId());
|
assertEquals(templateDO.getChannelCode(), logDO.getChannelCode());
|
assertEquals(templateContent, logDO.getTemplateContent());
|
assertEquals(templateParams, logDO.getTemplateParams());
|
assertEquals(SmsReceiveStatusEnum.INIT.getStatus(), logDO.getReceiveStatus());
|
}
|
|
@Test
|
public void testUpdateSmsSendResult() {
|
// mock 数据
|
SmsLogDO dbSmsLog = randomSmsLogDO(
|
o -> o.setSendStatus(SmsSendStatusEnum.IGNORE.getStatus()));
|
smsLogMapper.insert(dbSmsLog);
|
// 准备参数
|
Long id = dbSmsLog.getId();
|
Boolean success = randomBoolean();
|
String apiSendCode = randomString();
|
String apiSendMsg = randomString();
|
String apiRequestId = randomString();
|
String apiSerialNo = randomString();
|
|
// 调用
|
smsLogService.updateSmsSendResult(id, success,
|
apiSendCode, apiSendMsg, apiRequestId, apiSerialNo);
|
// 断言
|
dbSmsLog = smsLogMapper.selectById(id);
|
assertEquals(success ? SmsSendStatusEnum.SUCCESS.getStatus() : SmsSendStatusEnum.FAILURE.getStatus(),
|
dbSmsLog.getSendStatus());
|
assertNotNull(dbSmsLog.getSendTime());
|
assertEquals(apiSendCode, dbSmsLog.getApiSendCode());
|
assertEquals(apiSendMsg, dbSmsLog.getApiSendMsg());
|
assertEquals(apiRequestId, dbSmsLog.getApiRequestId());
|
assertEquals(apiSerialNo, dbSmsLog.getApiSerialNo());
|
}
|
|
@Test
|
public void testUpdateSmsReceiveResult() {
|
// mock 数据
|
SmsLogDO dbSmsLog = randomSmsLogDO(
|
o -> o.setReceiveStatus(SmsReceiveStatusEnum.INIT.getStatus()));
|
smsLogMapper.insert(dbSmsLog);
|
// 准备参数
|
Long id = dbSmsLog.getId();
|
Boolean success = randomBoolean();
|
LocalDateTime receiveTime = randomLocalDateTime();
|
String apiReceiveCode = randomString();
|
String apiReceiveMsg = randomString();
|
|
// 调用
|
smsLogService.updateSmsReceiveResult(id, success, receiveTime, apiReceiveCode, apiReceiveMsg);
|
// 断言
|
dbSmsLog = smsLogMapper.selectById(id);
|
assertEquals(success ? SmsReceiveStatusEnum.SUCCESS.getStatus()
|
: SmsReceiveStatusEnum.FAILURE.getStatus(), dbSmsLog.getReceiveStatus());
|
assertEquals(receiveTime, dbSmsLog.getReceiveTime());
|
assertEquals(apiReceiveCode, dbSmsLog.getApiReceiveCode());
|
assertEquals(apiReceiveMsg, dbSmsLog.getApiReceiveMsg());
|
}
|
|
// ========== 随机对象 ==========
|
|
@SafeVarargs
|
private static SmsLogDO randomSmsLogDO(Consumer<SmsLogDO>... consumers) {
|
Consumer<SmsLogDO> consumer = (o) -> {
|
o.setTemplateParams(randomTemplateParams());
|
o.setTemplateType(randomEle(SmsTemplateTypeEnum.values()).getType()); // 保证 templateType 的范围
|
o.setUserType(randomEle(UserTypeEnum.values()).getValue()); // 保证 userType 的范围
|
o.setSendStatus(randomEle(SmsSendStatusEnum.values()).getStatus()); // 保证 sendStatus 的范围
|
o.setReceiveStatus(randomEle(SmsReceiveStatusEnum.values()).getStatus()); // 保证 receiveStatus 的范围
|
};
|
return randomPojo(SmsLogDO.class, ArrayUtils.append(consumer, consumers));
|
}
|
|
private static Map<String, Object> randomTemplateParams() {
|
return MapUtil.<String, Object>builder().put(randomString(), randomString())
|
.put(randomString(), randomString()).build();
|
}
|
}
|
