params);
+
+}
diff --git a/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/notify/NotifyTemplateServiceImpl.java b/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/notify/NotifyTemplateServiceImpl.java
new file mode 100644
index 000000000..5ee8b4bbc
--- /dev/null
+++ b/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/notify/NotifyTemplateServiceImpl.java
@@ -0,0 +1,164 @@
+package cn.iocoder.yudao.module.system.service.notify;
+
+import cn.hutool.core.util.ReUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.template.NotifyTemplateCreateReqVO;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.template.NotifyTemplatePageReqVO;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.template.NotifyTemplateUpdateReqVO;
+import cn.iocoder.yudao.module.system.convert.notify.NotifyTemplateConvert;
+import cn.iocoder.yudao.module.system.dal.dataobject.notify.NotifyTemplateDO;
+import cn.iocoder.yudao.module.system.dal.mysql.notify.NotifyTemplateMapper;
+import cn.iocoder.yudao.module.system.mq.producer.notify.NotifyProducer;
+import com.google.common.annotations.VisibleForTesting;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+import org.springframework.validation.annotation.Validated;
+
+import javax.annotation.PostConstruct;
+import javax.annotation.Resource;
+import java.util.List;
+import java.util.Map;
+import java.util.regex.Pattern;
+
+import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
+import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*;
+
+/**
+ * 站内信模版 Service 实现类
+ *
+ * @author xrcoder
+ */
+@Service
+@Validated
+@Slf4j
+public class NotifyTemplateServiceImpl implements NotifyTemplateService {
+
+ /**
+ * 正则表达式，匹配 {} 中的变量
+ */
+ private static final Pattern PATTERN_PARAMS = Pattern.compile("\\{(.*?)}");
+
+ @Resource
+ private NotifyTemplateMapper notifyTemplateMapper;
+
+ @Resource
+ private NotifyProducer notifyProducer;
+
+ /**
+ * 站内信模板缓存
+ * key：站内信模板编码 {@link NotifyTemplateDO#getCode()}
+ *
+ * 这里声明 volatile 修饰的原因是，每次刷新时，直接修改指向
+ */
+ private volatile Map notifyTemplateCache;
+
+ /**
+ * 初始化站内信模板的本地缓存
+ */
+ @Override
+ @PostConstruct
+ public void initLocalCache() {
+ // 第一步：查询数据
+ List templates = notifyTemplateMapper.selectList();
+ log.info("[initLocalCache][缓存站内信模版，数量为:{}]", templates.size());
+
+ // 第二步：构建缓存
+ notifyTemplateCache = CollectionUtils.convertMap(templates, NotifyTemplateDO::getCode);
+ }
+
+ @Override
+ public NotifyTemplateDO getNotifyTemplateByCodeFromCache(String code) {
+ return notifyTemplateCache.get(code);
+ }
+
+ @Override
+ public Long createNotifyTemplate(NotifyTemplateCreateReqVO createReqVO) {
+ // 校验站内信编码是否重复
+ checkNotifyTemplateCodeDuplicate(null, createReqVO.getCode());
+
+ // 插入
+ NotifyTemplateDO notifyTemplate = NotifyTemplateConvert.INSTANCE.convert(createReqVO);
+ notifyTemplate.setParams(parseTemplateContentParams(notifyTemplate.getContent()));
+ notifyTemplateMapper.insert(notifyTemplate);
+
+ // 发送刷新消息
+ notifyProducer.sendNotifyTemplateRefreshMessage();
+ return notifyTemplate.getId();
+ }
+
+ @Override
+ public void updateNotifyTemplate(NotifyTemplateUpdateReqVO updateReqVO) {
+ // 校验存在
+ validateNotifyTemplateExists(updateReqVO.getId());
+ // 校验站内信编码是否重复
+ checkNotifyTemplateCodeDuplicate(updateReqVO.getId(), updateReqVO.getCode());
+
+ // 更新
+ NotifyTemplateDO updateObj = NotifyTemplateConvert.INSTANCE.convert(updateReqVO);
+ updateObj.setParams(parseTemplateContentParams(updateObj.getContent()));
+ notifyTemplateMapper.updateById(updateObj);
+
+ // 发送刷新消息
+ notifyProducer.sendNotifyTemplateRefreshMessage();
+ }
+
+ @VisibleForTesting
+ public List parseTemplateContentParams(String content) {
+ return ReUtil.findAllGroup1(PATTERN_PARAMS, content);
+ }
+
+ @Override
+ public void deleteNotifyTemplate(Long id) {
+ // 校验存在
+ validateNotifyTemplateExists(id);
+ // 删除
+ notifyTemplateMapper.deleteById(id);
+ // 发送刷新消息
+ notifyProducer.sendNotifyTemplateRefreshMessage();
+ }
+
+ private void validateNotifyTemplateExists(Long id) {
+ if (notifyTemplateMapper.selectById(id) == null) {
+ throw exception(NOTIFY_TEMPLATE_NOT_EXISTS);
+ }
+ }
+
+ @Override
+ public NotifyTemplateDO getNotifyTemplate(Long id) {
+ return notifyTemplateMapper.selectById(id);
+ }
+
+ @Override
+ public PageResult getNotifyTemplatePage(NotifyTemplatePageReqVO pageReqVO) {
+ return notifyTemplateMapper.selectPage(pageReqVO);
+ }
+
+ @VisibleForTesting
+ public void checkNotifyTemplateCodeDuplicate(Long id, String code) {
+ NotifyTemplateDO template = notifyTemplateMapper.selectByCode(code);
+ if (template == null) {
+ return;
+ }
+ // 如果 id 为空，说明不用比较是否为相同 id 的字典类型
+ if (id == null) {
+ throw exception(NOTIFY_TEMPLATE_CODE_DUPLICATE, code);
+ }
+ if (!template.getId().equals(id)) {
+ throw exception(NOTIFY_TEMPLATE_CODE_DUPLICATE, code);
+ }
+ }
+
+ /**
+ * 格式化站内信内容
+ *
+ * @param content 站内信模板的内容
+ * @param params 站内信内容的参数
+ * @return 格式化后的内容
+ */
+ @Override
+ public String formatNotifyTemplateContent(String content, Map params) {
+ return StrUtil.format(content, params);
+ }
+}
diff --git a/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifyMessageServiceImplTest.java b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifyMessageServiceImplTest.java
new file mode 100644
index 000000000..1087918cc
--- /dev/null
+++ b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifyMessageServiceImplTest.java
@@ -0,0 +1,266 @@
+package cn.iocoder.yudao.module.system.service.notify;
+
+import cn.hutool.core.map.MapUtil;
+import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.enums.SqlConstants;
+import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.message.NotifyMessageMyPageReqVO;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.message.NotifyMessagePageReqVO;
+import cn.iocoder.yudao.module.system.dal.dataobject.notify.NotifyMessageDO;
+import cn.iocoder.yudao.module.system.dal.dataobject.notify.NotifyTemplateDO;
+import cn.iocoder.yudao.module.system.dal.mysql.notify.NotifyMessageMapper;
+import com.baomidou.mybatisplus.annotation.DbType;
+import org.junit.jupiter.api.Test;
+import org.springframework.context.annotation.Import;
+
+import javax.annotation.Resource;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+
+import static cn.hutool.core.util.RandomUtil.randomEle;
+import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildBetweenTime;
+import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildTime;
+import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
+import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
+import static org.junit.jupiter.api.Assertions.*;
+
+/**
+* {@link NotifyMessageServiceImpl} 的单元测试类
+*
+* @author 芋道源码
+*/
+@Import(NotifyMessageServiceImpl.class)
+public class NotifyMessageServiceImplTest extends BaseDbUnitTest {
+
+ @Resource
+ private NotifyMessageServiceImpl notifyMessageService;
+
+ @Resource
+ private NotifyMessageMapper notifyMessageMapper;
+
+ @Test
+ public void testCreateNotifyMessage_success() {
+ // 准备参数
+ Long userId = randomLongId();
+ Integer userType = randomEle(UserTypeEnum.values()).getValue();
+ NotifyTemplateDO template = randomPojo(NotifyTemplateDO.class);
+ String templateContent = randomString();
+ Map templateParams = randomTemplateParams();
+ // mock 方法
+
+ // 调用
+ Long messageId = notifyMessageService.createNotifyMessage(userId, userType,
+ template, templateContent, templateParams);
+ // 断言
+ NotifyMessageDO message = notifyMessageMapper.selectById(messageId);
+ assertNotNull(message);
+ assertEquals(userId, message.getUserId());
+ assertEquals(userType, message.getUserType());
+ assertEquals(template.getId(), message.getTemplateId());
+ assertEquals(template.getCode(), message.getTemplateCode());
+ assertEquals(template.getType(), message.getTemplateType());
+ assertEquals(template.getNickname(), message.getTemplateNickname());
+ assertEquals(templateContent, message.getTemplateContent());
+ assertEquals(templateParams, message.getTemplateParams());
+ assertEquals(false, message.getReadStatus());
+ assertNull(message.getReadTime());
+ }
+
+ @Test
+ public void testGetNotifyMessagePage() {
+ // mock 数据
+ NotifyMessageDO dbNotifyMessage = randomPojo(NotifyMessageDO.class, o -> { // 等会查询到
+ o.setUserId(1L);
+ o.setUserType(UserTypeEnum.ADMIN.getValue());
+ o.setTemplateCode("test_01");
+ o.setTemplateType(10);
+ o.setCreateTime(buildTime(2022, 1, 2));
+ o.setTemplateParams(randomTemplateParams());
+ });
+ notifyMessageMapper.insert(dbNotifyMessage);
+ // 测试 userId 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserId(2L)));
+ // 测试 userType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
+ // 测试 templateCode 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setTemplateCode("test_11")));
+ // 测试 templateType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setTemplateType(20)));
+ // 测试 createTime 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setCreateTime(buildTime(2022, 2, 1))));
+ // 准备参数
+ NotifyMessagePageReqVO reqVO = new NotifyMessagePageReqVO();
+ reqVO.setUserId(1L);
+ reqVO.setUserType(UserTypeEnum.ADMIN.getValue());
+ reqVO.setTemplateCode("est_01");
+ reqVO.setTemplateType(10);
+ reqVO.setCreateTime(buildBetweenTime(2022, 1, 1, 2022, 1, 10));
+
+ // 调用
+ PageResult pageResult = notifyMessageService.getNotifyMessagePage(reqVO);
+ // 断言
+ assertEquals(1, pageResult.getTotal());
+ assertEquals(1, pageResult.getList().size());
+ assertPojoEquals(dbNotifyMessage, pageResult.getList().get(0));
+ }
+
+ @Test
+ public void testGetMyNotifyMessagePage() {
+ // mock 数据
+ NotifyMessageDO dbNotifyMessage = randomPojo(NotifyMessageDO.class, o -> { // 等会查询到
+ o.setUserId(1L);
+ o.setUserType(UserTypeEnum.ADMIN.getValue());
+ o.setReadStatus(true);
+ o.setCreateTime(buildTime(2022, 1, 2));
+ o.setTemplateParams(randomTemplateParams());
+ });
+ notifyMessageMapper.insert(dbNotifyMessage);
+ // 测试 userId 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserId(2L)));
+ // 测试 userType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
+ // 测试 readStatus 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setReadStatus(false)));
+ // 测试 createTime 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setCreateTime(buildTime(2022, 2, 1))));
+ // 准备参数
+ Long userId = 1L;
+ Integer userType = UserTypeEnum.ADMIN.getValue();
+ NotifyMessageMyPageReqVO reqVO = new NotifyMessageMyPageReqVO();
+ reqVO.setReadStatus(true);
+ reqVO.setCreateTime(buildBetweenTime(2022, 1, 1, 2022, 1, 10));
+
+ // 调用
+ PageResult pageResult = notifyMessageService.getMyMyNotifyMessagePage(reqVO, userId, userType);
+ // 断言
+ assertEquals(1, pageResult.getTotal());
+ assertEquals(1, pageResult.getList().size());
+ assertPojoEquals(dbNotifyMessage, pageResult.getList().get(0));
+ }
+
+ @Test
+ public void testGetUnreadNotifyMessageList() {
+ SqlConstants.init(DbType.MYSQL);
+ // mock 数据
+ NotifyMessageDO dbNotifyMessage = randomPojo(NotifyMessageDO.class, o -> { // 等会查询到
+ o.setUserId(1L);
+ o.setUserType(UserTypeEnum.ADMIN.getValue());
+ o.setReadStatus(false);
+ o.setTemplateParams(randomTemplateParams());
+ });
+ notifyMessageMapper.insert(dbNotifyMessage);
+ // 测试 userId 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserId(2L)));
+ // 测试 userType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
+ // 测试 readStatus 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setReadStatus(true)));
+ // 准备参数
+ Long userId = 1L;
+ Integer userType = UserTypeEnum.ADMIN.getValue();
+ Integer size = 10;
+
+ // 调用
+ List list = notifyMessageService.getUnreadNotifyMessageList(userId, userType, size);
+ // 断言
+ assertEquals(1, list.size());
+ assertPojoEquals(dbNotifyMessage, list.get(0));
+ }
+
+ @Test
+ public void testGetUnreadNotifyMessageCount() {
+ SqlConstants.init(DbType.MYSQL);
+ // mock 数据
+ NotifyMessageDO dbNotifyMessage = randomPojo(NotifyMessageDO.class, o -> { // 等会查询到
+ o.setUserId(1L);
+ o.setUserType(UserTypeEnum.ADMIN.getValue());
+ o.setReadStatus(false);
+ o.setTemplateParams(randomTemplateParams());
+ });
+ notifyMessageMapper.insert(dbNotifyMessage);
+ // 测试 userId 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserId(2L)));
+ // 测试 userType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
+ // 测试 readStatus 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setReadStatus(true)));
+ // 准备参数
+ Long userId = 1L;
+ Integer userType = UserTypeEnum.ADMIN.getValue();
+
+ // 调用，并断言
+ assertEquals(1, notifyMessageService.getUnreadNotifyMessageCount(userId, userType));
+ }
+
+ @Test
+ public void testUpdateNotifyMessageRead() {
+ // mock 数据
+ NotifyMessageDO dbNotifyMessage = randomPojo(NotifyMessageDO.class, o -> { // 等会查询到
+ o.setUserId(1L);
+ o.setUserType(UserTypeEnum.ADMIN.getValue());
+ o.setReadStatus(false);
+ o.setReadTime(null);
+ o.setTemplateParams(randomTemplateParams());
+ });
+ notifyMessageMapper.insert(dbNotifyMessage);
+ // 测试 userId 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserId(2L)));
+ // 测试 userType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
+ // 测试 readStatus 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setReadStatus(true)));
+ // 准备参数
+ Collection ids = Arrays.asList(dbNotifyMessage.getId(), dbNotifyMessage.getId() + 1,
+ dbNotifyMessage.getId() + 2, dbNotifyMessage.getId() + 3);
+ Long userId = 1L;
+ Integer userType = UserTypeEnum.ADMIN.getValue();
+
+ // 调用
+ int updateCount = notifyMessageService.updateNotifyMessageRead(ids, userId, userType);
+ // 断言
+ assertEquals(1, updateCount);
+ NotifyMessageDO notifyMessage = notifyMessageMapper.selectById(dbNotifyMessage.getId());
+ assertTrue(notifyMessage.getReadStatus());
+ assertNotNull(notifyMessage.getReadTime());
+ }
+
+ @Test
+ public void testUpdateAllNotifyMessageRead() {
+ // mock 数据
+ NotifyMessageDO dbNotifyMessage = randomPojo(NotifyMessageDO.class, o -> { // 等会查询到
+ o.setUserId(1L);
+ o.setUserType(UserTypeEnum.ADMIN.getValue());
+ o.setReadStatus(false);
+ o.setReadTime(null);
+ o.setTemplateParams(randomTemplateParams());
+ });
+ notifyMessageMapper.insert(dbNotifyMessage);
+ // 测试 userId 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserId(2L)));
+ // 测试 userType 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setUserType(UserTypeEnum.MEMBER.getValue())));
+ // 测试 readStatus 不匹配
+ notifyMessageMapper.insert(cloneIgnoreId(dbNotifyMessage, o -> o.setReadStatus(true)));
+ // 准备参数
+ Long userId = 1L;
+ Integer userType = UserTypeEnum.ADMIN.getValue();
+
+ // 调用
+ int updateCount = notifyMessageService.updateAllNotifyMessageRead(userId, userType);
+ // 断言
+ assertEquals(1, updateCount);
+ NotifyMessageDO notifyMessage = notifyMessageMapper.selectById(dbNotifyMessage.getId());
+ assertTrue(notifyMessage.getReadStatus());
+ assertNotNull(notifyMessage.getReadTime());
+ }
+
+ private static Map randomTemplateParams() {
+ return MapUtil.builder().put(randomString(), randomString())
+ .put(randomString(), randomString()).build();
+ }
+
+}
diff --git a/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifySendServiceImplTest.java b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifySendServiceImplTest.java
new file mode 100644
index 000000000..ddc214443
--- /dev/null
+++ b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifySendServiceImplTest.java
@@ -0,0 +1,120 @@
+package cn.iocoder.yudao.module.system.service.notify;
+
+import cn.hutool.core.map.MapUtil;
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
+import cn.iocoder.yudao.framework.test.core.ut.BaseMockitoUnitTest;
+import cn.iocoder.yudao.module.system.dal.dataobject.notify.NotifyTemplateDO;
+import org.assertj.core.util.Lists;
+import org.junit.jupiter.api.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static cn.hutool.core.util.RandomUtil.randomEle;
+import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
+import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.NOTICE_NOT_FOUND;
+import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.NOTIFY_SEND_TEMPLATE_PARAM_MISS;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.*;
+
+class NotifySendServiceImplTest extends BaseMockitoUnitTest {
+
+ @InjectMocks
+ private NotifySendServiceImpl notifySendService;
+
+ @Mock
+ private NotifyTemplateService notifyTemplateService;
+ @Mock
+ private NotifyMessageService notifyMessageService;
+
+ /**
+ * 发送成功，当短信模板开启时
+ */
+ @Test
+ public void testSendSingleNotify_successWhenMailTemplateEnable() {
+ // 准备参数
+ Long userId = randomLongId();
+ Integer userType = randomEle(UserTypeEnum.values()).getValue();
+ String templateCode = randomString();
+ Map templateParams = MapUtil.builder().put("code", "1234")
+ .put("op", "login").build();
+ // mock NotifyTemplateService 的方法
+ NotifyTemplateDO template = randomPojo(NotifyTemplateDO.class, o -> {
+ o.setStatus(CommonStatusEnum.ENABLE.getStatus());
+ o.setContent("验证码为{code}, 操作为{op}");
+ o.setParams(Lists.newArrayList("code", "op"));
+ });
+ when(notifyTemplateService.getNotifyTemplateByCodeFromCache(eq(templateCode))).thenReturn(template);
+ String content = randomString();
+ when(notifyTemplateService.formatNotifyTemplateContent(eq(template.getContent()), eq(templateParams)))
+ .thenReturn(content);
+ // mock NotifyMessageService 的方法
+ Long messageId = randomLongId();
+ when(notifyMessageService.createNotifyMessage(eq(userId), eq(userType),
+ eq(template), eq(content), eq(templateParams))).thenReturn(messageId);
+
+ // 调用
+ Long resultMessageId = notifySendService.sendSingleNotify(userId, userType, templateCode, templateParams);
+ // 断言
+ assertEquals(messageId, resultMessageId);
+ }
+
+ /**
+ * 发送成功，当短信模板关闭时
+ */
+ @Test
+ public void testSendSingleMail_successWhenSmsTemplateDisable() {
+ // 准备参数
+ Long userId = randomLongId();
+ Integer userType = randomEle(UserTypeEnum.values()).getValue();
+ String templateCode = randomString();
+ Map templateParams = MapUtil.builder().put("code", "1234")
+ .put("op", "login").build();
+ // mock NotifyTemplateService 的方法
+ NotifyTemplateDO template = randomPojo(NotifyTemplateDO.class, o -> {
+ o.setStatus(CommonStatusEnum.DISABLE.getStatus());
+ o.setContent("验证码为{code}, 操作为{op}");
+ o.setParams(Lists.newArrayList("code", "op"));
+ });
+ when(notifyTemplateService.getNotifyTemplateByCodeFromCache(eq(templateCode))).thenReturn(template);
+
+ // 调用
+ Long resultMessageId = notifySendService.sendSingleNotify(userId, userType, templateCode, templateParams);
+ // 断言
+ assertNull(resultMessageId);
+ verify(notifyTemplateService, never()).formatNotifyTemplateContent(anyString(), anyMap());
+ verify(notifyMessageService, never()).createNotifyMessage(anyLong(), anyInt(), any(), anyString(), anyMap());
+ }
+
+ @Test
+ public void testCheckMailTemplateValid_notExists() {
+ // 准备参数
+ String templateCode = randomString();
+ // mock 方法
+
+ // 调用，并断言异常
+ assertServiceException(() -> notifySendService.checkNotifyTemplateValid(templateCode),
+ NOTICE_NOT_FOUND);
+ }
+
+ @Test
+ public void testCheckTemplateParams_paramMiss() {
+ // 准备参数
+ NotifyTemplateDO template = randomPojo(NotifyTemplateDO.class,
+ o -> o.setParams(Lists.newArrayList("code")));
+ Map templateParams = new HashMap<>();
+ // mock 方法
+
+ // 调用，并断言异常
+ assertServiceException(() -> notifySendService.checkTemplateParams(template, templateParams),
+ NOTIFY_SEND_TEMPLATE_PARAM_MISS, "code");
+ }
+
+
+}
diff --git a/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifyTemplateServiceImplTest.java b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifyTemplateServiceImplTest.java
new file mode 100644
index 000000000..5a6f5a508
--- /dev/null
+++ b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/notify/NotifyTemplateServiceImplTest.java
@@ -0,0 +1,146 @@
+package cn.iocoder.yudao.module.system.service.notify;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.template.NotifyTemplateCreateReqVO;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.template.NotifyTemplatePageReqVO;
+import cn.iocoder.yudao.module.system.controller.admin.notify.vo.template.NotifyTemplateUpdateReqVO;
+import cn.iocoder.yudao.module.system.dal.dataobject.notify.NotifyTemplateDO;
+import cn.iocoder.yudao.module.system.dal.mysql.notify.NotifyTemplateMapper;
+import cn.iocoder.yudao.module.system.mq.producer.notify.NotifyProducer;
+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 static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildBetweenTime;
+import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildTime;
+import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
+import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
+import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
+import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.NOTIFY_TEMPLATE_NOT_EXISTS;
+import static org.junit.jupiter.api.Assertions.*;
+import static org.mockito.Mockito.verify;
+
+/**
+* {@link NotifyTemplateServiceImpl} 的单元测试类
+*
+* @author 芋道源码
+*/
+@Import(NotifyTemplateServiceImpl.class)
+public class NotifyTemplateServiceImplTest extends BaseDbUnitTest {
+
+ @Resource
+ private NotifyTemplateServiceImpl notifyTemplateService;
+
+ @Resource
+ private NotifyTemplateMapper notifyTemplateMapper;
+
+ @MockBean
+ private NotifyProducer notifyProducer;
+
+ @Test
+ public void testCreateNotifyTemplate_success() {
+ // 准备参数
+ NotifyTemplateCreateReqVO reqVO = randomPojo(NotifyTemplateCreateReqVO.class,
+ o -> o.setStatus(randomCommonStatus()));
+
+ // 调用
+ Long notifyTemplateId = notifyTemplateService.createNotifyTemplate(reqVO);
+ // 断言
+ assertNotNull(notifyTemplateId);
+ // 校验记录的属性是否正确
+ NotifyTemplateDO notifyTemplate = notifyTemplateMapper.selectById(notifyTemplateId);
+ assertPojoEquals(reqVO, notifyTemplate);
+ verify(notifyProducer).sendNotifyTemplateRefreshMessage();
+ }
+
+ @Test
+ public void testUpdateNotifyTemplate_success() {
+ // mock 数据
+ NotifyTemplateDO dbNotifyTemplate = randomPojo(NotifyTemplateDO.class);
+ notifyTemplateMapper.insert(dbNotifyTemplate);// @Sql: 先插入出一条存在的数据
+ // 准备参数
+ NotifyTemplateUpdateReqVO reqVO = randomPojo(NotifyTemplateUpdateReqVO.class, o -> {
+ o.setId(dbNotifyTemplate.getId()); // 设置更新的 ID
+ o.setStatus(randomCommonStatus());
+ });
+
+ // 调用
+ notifyTemplateService.updateNotifyTemplate(reqVO);
+ // 校验是否更新正确
+ NotifyTemplateDO notifyTemplate = notifyTemplateMapper.selectById(reqVO.getId()); // 获取最新的
+ assertPojoEquals(reqVO, notifyTemplate);
+ verify(notifyProducer).sendNotifyTemplateRefreshMessage();
+ }
+
+ @Test
+ public void testUpdateNotifyTemplate_notExists() {
+ // 准备参数
+ NotifyTemplateUpdateReqVO reqVO = randomPojo(NotifyTemplateUpdateReqVO.class);
+
+ // 调用, 并断言异常
+ assertServiceException(() -> notifyTemplateService.updateNotifyTemplate(reqVO), NOTIFY_TEMPLATE_NOT_EXISTS);
+ }
+
+ @Test
+ public void testDeleteNotifyTemplate_success() {
+ // mock 数据
+ NotifyTemplateDO dbNotifyTemplate = randomPojo(NotifyTemplateDO.class);
+ notifyTemplateMapper.insert(dbNotifyTemplate);// @Sql: 先插入出一条存在的数据
+ // 准备参数
+ Long id = dbNotifyTemplate.getId();
+
+ // 调用
+ notifyTemplateService.deleteNotifyTemplate(id);
+ // 校验数据不存在了
+ assertNull(notifyTemplateMapper.selectById(id));
+ verify(notifyProducer).sendNotifyTemplateRefreshMessage();
+ }
+
+ @Test
+ public void testDeleteNotifyTemplate_notExists() {
+ // 准备参数
+ Long id = randomLongId();
+
+ // 调用, 并断言异常
+ assertServiceException(() -> notifyTemplateService.deleteNotifyTemplate(id), NOTIFY_TEMPLATE_NOT_EXISTS);
+ }
+
+ @Test
+ public void testGetNotifyTemplatePage() {
+ // mock 数据
+ NotifyTemplateDO dbNotifyTemplate = randomPojo(NotifyTemplateDO.class, o -> { // 等会查询到
+ o.setName("芋头");
+ o.setCode("test_01");
+ o.setStatus(CommonStatusEnum.ENABLE.getStatus());
+ o.setCreateTime(buildTime(2022, 2, 3));
+ });
+ notifyTemplateMapper.insert(dbNotifyTemplate);
+ // 测试 name 不匹配
+ notifyTemplateMapper.insert(cloneIgnoreId(dbNotifyTemplate, o -> o.setName("投")));
+ // 测试 code 不匹配
+ notifyTemplateMapper.insert(cloneIgnoreId(dbNotifyTemplate, o -> o.setCode("test_02")));
+ // 测试 status 不匹配
+ notifyTemplateMapper.insert(cloneIgnoreId(dbNotifyTemplate, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())));
+ // 测试 createTime 不匹配
+ notifyTemplateMapper.insert(cloneIgnoreId(dbNotifyTemplate, o -> o.setCreateTime(buildTime(2022, 1, 5))));
+ // 准备参数
+ NotifyTemplatePageReqVO reqVO = new NotifyTemplatePageReqVO();
+ reqVO.setName("芋");
+ reqVO.setCode("est_01");
+ reqVO.setStatus(CommonStatusEnum.ENABLE.getStatus());
+ reqVO.setCreateTime(buildBetweenTime(2022, 2, 1, 2022, 2, 5));
+
+ // 调用
+ PageResult pageResult = notifyTemplateService.getNotifyTemplatePage(reqVO);
+ // 断言
+ assertEquals(1, pageResult.getTotal());
+ assertEquals(1, pageResult.getList().size());
+ assertPojoEquals(dbNotifyTemplate, pageResult.getList().get(0));
+ }
+
+}
diff --git a/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/clean.sql b/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/clean.sql
index 1fb4cbaf9..1be69f28b 100644
--- a/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/clean.sql
+++ b/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/clean.sql
@@ -28,3 +28,5 @@ DELETE FROM "system_oauth2_code";
DELETE FROM "system_mail_account";
DELETE FROM "system_mail_template";
DELETE FROM "system_mail_log";
+DELETE FROM "system_notify_template";
+DELETE FROM "system_notify_message";
diff --git a/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/create_tables.sql b/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/create_tables.sql
index 0acd892c3..b00c9168e 100644
--- a/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/create_tables.sql
+++ b/yudao-module-system/yudao-module-system-biz/src/test/resources/sql/create_tables.sql
@@ -626,3 +626,43 @@ CREATE TABLE IF NOT EXISTS "system_mail_log" (
"deleted" bit NOT NULL DEFAULT FALSE,
PRIMARY KEY ("id")
) COMMENT '邮件日志表';
+
+-- 将该建表 SQL 语句，添加到 yudao-module-system-biz 模块的 test/resources/sql/create_tables.sql 文件里
+CREATE TABLE IF NOT EXISTS "system_notify_template" (
+ "id" bigint NOT NULL GENERATED BY DEFAULT AS IDENTITY,
+ "name" varchar NOT NULL,
+ "code" varchar NOT NULL,
+ "nickname" varchar NOT NULL,
+ "content" varchar NOT NULL,
+ "type" varchar NOT NULL,
+ "params" varchar,
+ "status" varchar NOT NULL,
+ "remark" varchar,
+ "creator" varchar DEFAULT '',
+ "create_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+ "updater" varchar DEFAULT '',
+ "update_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+ "deleted" bit NOT NULL DEFAULT FALSE,
+ PRIMARY KEY ("id")
+) COMMENT '站内信模板表';
+
+CREATE TABLE IF NOT EXISTS "system_notify_message" (
+ "id" bigint NOT NULL GENERATED BY DEFAULT AS IDENTITY,
+ "user_id" bigint NOT NULL,
+ "user_type" varchar NOT NULL,
+ "template_id" bigint NOT NULL,
+ "template_code" varchar NOT NULL,
+ "template_nickname" varchar NOT NULL,
+ "template_content" varchar NOT NULL,
+ "template_type" int NOT NULL,
+ "template_params" varchar NOT NULL,
+ "read_status" bit NOT NULL,
+ "read_time" varchar,
+ "creator" varchar DEFAULT '',
+ "create_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+ "updater" varchar DEFAULT '',
+ "update_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+ "deleted" bit NOT NULL DEFAULT FALSE,
+ "tenant_id" bigint not null default '0',
+ PRIMARY KEY ("id")
+) COMMENT '站内信消息表';