package com.iailab.module.system.service.sms;
|
|
import com.iailab.framework.common.enums.CommonStatusEnum;
|
import com.iailab.framework.common.pojo.PageResult;
|
import com.iailab.framework.common.util.object.BeanUtils;
|
import com.iailab.module.system.framework.sms.core.client.SmsClient;
|
import com.iailab.module.system.framework.sms.core.client.SmsClientFactory;
|
import com.iailab.module.system.framework.sms.core.property.SmsChannelProperties;
|
import com.iailab.framework.test.core.ut.BaseDbUnitTest;
|
import com.iailab.module.system.controller.admin.sms.vo.channel.SmsChannelPageReqVO;
|
import com.iailab.module.system.controller.admin.sms.vo.channel.SmsChannelSaveReqVO;
|
import com.iailab.module.system.dal.dataobject.sms.SmsChannelDO;
|
import com.iailab.module.system.dal.mysql.sms.SmsChannelMapper;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.boot.test.mock.mockito.MockBean;
|
import org.springframework.context.annotation.Import;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
|
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.AssertUtils.assertServiceException;
|
import static com.iailab.framework.test.core.util.RandomUtils.*;
|
import static com.iailab.module.system.enums.ErrorCodeConstants.SMS_CHANNEL_HAS_CHILDREN;
|
import static com.iailab.module.system.enums.ErrorCodeConstants.SMS_CHANNEL_NOT_EXISTS;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.ArgumentMatchers.eq;
|
import static org.mockito.Mockito.*;
|
|
@Import(SmsChannelServiceImpl.class)
|
public class SmsChannelServiceTest extends BaseDbUnitTest {
|
|
@Resource
|
private SmsChannelServiceImpl smsChannelService;
|
|
@Resource
|
private SmsChannelMapper smsChannelMapper;
|
|
@MockBean
|
private SmsClientFactory smsClientFactory;
|
@MockBean
|
private SmsTemplateService smsTemplateService;
|
|
@Test
|
public void testCreateSmsChannel_success() {
|
// 准备参数
|
SmsChannelSaveReqVO reqVO = randomPojo(SmsChannelSaveReqVO.class, o -> o.setStatus(randomCommonStatus()))
|
.setId(null); // 防止 id 被赋值
|
|
// 调用
|
Long smsChannelId = smsChannelService.createSmsChannel(reqVO);
|
// 断言
|
assertNotNull(smsChannelId);
|
// 校验记录的属性是否正确
|
SmsChannelDO smsChannel = smsChannelMapper.selectById(smsChannelId);
|
assertPojoEquals(reqVO, smsChannel, "id");
|
// 断言 cache
|
assertNull(smsChannelService.getIdClientCache().getIfPresent(smsChannel.getId()));
|
assertNull(smsChannelService.getCodeClientCache().getIfPresent(smsChannel.getCode()));
|
}
|
|
@Test
|
public void testUpdateSmsChannel_success() {
|
// mock 数据
|
SmsChannelDO dbSmsChannel = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(dbSmsChannel);// @Sql: 先插入出一条存在的数据
|
// 准备参数
|
SmsChannelSaveReqVO reqVO = randomPojo(SmsChannelSaveReqVO.class, o -> {
|
o.setId(dbSmsChannel.getId()); // 设置更新的 ID
|
o.setStatus(randomCommonStatus());
|
o.setCallbackUrl(randomString());
|
});
|
|
// 调用
|
smsChannelService.updateSmsChannel(reqVO);
|
// 校验是否更新正确
|
SmsChannelDO smsChannel = smsChannelMapper.selectById(reqVO.getId()); // 获取最新的
|
assertPojoEquals(reqVO, smsChannel);
|
// 断言 cache
|
assertNull(smsChannelService.getIdClientCache().getIfPresent(smsChannel.getId()));
|
assertNull(smsChannelService.getCodeClientCache().getIfPresent(smsChannel.getCode()));
|
}
|
|
@Test
|
public void testUpdateSmsChannel_notExists() {
|
// 准备参数
|
SmsChannelSaveReqVO reqVO = randomPojo(SmsChannelSaveReqVO.class);
|
|
// 调用, 并断言异常
|
assertServiceException(() -> smsChannelService.updateSmsChannel(reqVO), SMS_CHANNEL_NOT_EXISTS);
|
}
|
|
@Test
|
public void testDeleteSmsChannel_success() {
|
// mock 数据
|
SmsChannelDO dbSmsChannel = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(dbSmsChannel);// @Sql: 先插入出一条存在的数据
|
// 准备参数
|
Long id = dbSmsChannel.getId();
|
|
// 调用
|
smsChannelService.deleteSmsChannel(id);
|
// 校验数据不存在了
|
assertNull(smsChannelMapper.selectById(id));
|
// 断言 cache
|
assertNull(smsChannelService.getIdClientCache().getIfPresent(dbSmsChannel.getId()));
|
assertNull(smsChannelService.getCodeClientCache().getIfPresent(dbSmsChannel.getCode()));
|
}
|
|
@Test
|
public void testDeleteSmsChannel_notExists() {
|
// 准备参数
|
Long id = randomLongId();
|
|
// 调用, 并断言异常
|
assertServiceException(() -> smsChannelService.deleteSmsChannel(id), SMS_CHANNEL_NOT_EXISTS);
|
}
|
|
@Test
|
public void testDeleteSmsChannel_hasChildren() {
|
// mock 数据
|
SmsChannelDO dbSmsChannel = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(dbSmsChannel);// @Sql: 先插入出一条存在的数据
|
// 准备参数
|
Long id = dbSmsChannel.getId();
|
// mock 方法
|
when(smsTemplateService.getSmsTemplateCountByChannelId(eq(id))).thenReturn(10L);
|
|
// 调用, 并断言异常
|
assertServiceException(() -> smsChannelService.deleteSmsChannel(id), SMS_CHANNEL_HAS_CHILDREN);
|
}
|
|
@Test
|
public void testGetSmsChannel() {
|
// mock 数据
|
SmsChannelDO dbSmsChannel = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(dbSmsChannel); // @Sql: 先插入出一条存在的数据
|
// 准备参数
|
Long id = dbSmsChannel.getId();
|
|
// 调用，并断言
|
assertPojoEquals(dbSmsChannel, smsChannelService.getSmsChannel(id));
|
}
|
|
@Test
|
public void testGetSmsChannelList() {
|
// mock 数据
|
SmsChannelDO dbSmsChannel01 = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(dbSmsChannel01);
|
SmsChannelDO dbSmsChannel02 = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(dbSmsChannel02);
|
// 准备参数
|
|
// 调用
|
List<SmsChannelDO> list = smsChannelService.getSmsChannelList();
|
// 断言
|
assertEquals(2, list.size());
|
assertPojoEquals(dbSmsChannel01, list.get(0));
|
assertPojoEquals(dbSmsChannel02, list.get(1));
|
}
|
|
@Test
|
public void testGetSmsChannelPage() {
|
// mock 数据
|
SmsChannelDO dbSmsChannel = randomPojo(SmsChannelDO.class, o -> { // 等会查询到
|
o.setSignature("iailab");
|
o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
o.setCreateTime(buildTime(2020, 12, 12));
|
});
|
smsChannelMapper.insert(dbSmsChannel);
|
// 测试 signature 不匹配
|
smsChannelMapper.insert(cloneIgnoreId(dbSmsChannel, o -> o.setSignature("源码")));
|
// 测试 status 不匹配
|
smsChannelMapper.insert(cloneIgnoreId(dbSmsChannel, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())));
|
// 测试 createTime 不匹配
|
smsChannelMapper.insert(cloneIgnoreId(dbSmsChannel, o -> o.setCreateTime(buildTime(2020, 11, 11))));
|
// 准备参数
|
SmsChannelPageReqVO reqVO = new SmsChannelPageReqVO();
|
reqVO.setSignature("平台");
|
reqVO.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
reqVO.setCreateTime(buildBetweenTime(2020, 12, 1, 2020, 12, 24));
|
|
// 调用
|
PageResult<SmsChannelDO> pageResult = smsChannelService.getSmsChannelPage(reqVO);
|
// 断言
|
assertEquals(1, pageResult.getTotal());
|
assertEquals(1, pageResult.getList().size());
|
assertPojoEquals(dbSmsChannel, pageResult.getList().get(0));
|
}
|
|
@Test
|
public void testGetSmsClient_id() {
|
// mock 数据
|
SmsChannelDO channel = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(channel);
|
// mock 参数
|
Long id = channel.getId();
|
// mock 方法
|
SmsClient mockClient = mock(SmsClient.class);
|
when(smsClientFactory.getSmsClient(eq(id))).thenReturn(mockClient);
|
|
// 调用
|
SmsClient client = smsChannelService.getSmsClient(id);
|
// 断言
|
assertSame(client, mockClient);
|
verify(smsClientFactory).createOrUpdateSmsClient(argThat(arg -> {
|
SmsChannelProperties properties = BeanUtils.toBean(channel, SmsChannelProperties.class);
|
return properties.equals(arg);
|
}));
|
}
|
|
@Test
|
public void testGetSmsClient_code() {
|
// mock 数据
|
SmsChannelDO channel = randomPojo(SmsChannelDO.class);
|
smsChannelMapper.insert(channel);
|
// mock 参数
|
String code = channel.getCode();
|
// mock 方法
|
SmsClient mockClient = mock(SmsClient.class);
|
when(smsClientFactory.getSmsClient(eq(code))).thenReturn(mockClient);
|
|
// 调用
|
SmsClient client = smsChannelService.getSmsClient(code);
|
// 断言
|
assertSame(client, mockClient);
|
verify(smsClientFactory).createOrUpdateSmsClient(argThat(arg -> {
|
SmsChannelProperties properties = BeanUtils.toBean(channel, SmsChannelProperties.class);
|
return properties.equals(arg);
|
}));
|
}
|
|
}
|
