BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 83295a10aa2f39cd055dee4ee466017216a712c7

Authored by YunaiV
1 parent d3b6a0ca

1. 修复 SmsSendServiceTest 单测的报错

Inline Side-by-side
Showing 1 changed file with 10 additions and 0 deletions
yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/sms/SmsSendServiceTest.java
  View file @83295a1
1 package cn.iocoder.yudao.module.system.service.sms; 1 package cn.iocoder.yudao.module.system.service.sms;
2 2
3 import cn.hutool.core.map.MapUtil; 3 import cn.hutool.core.map.MapUtil;
  4 +import cn.iocoder.yudao.module.system.dal.dataobject.sms.SmsChannelDO;
4 import cn.iocoder.yudao.module.system.dal.dataobject.sms.SmsTemplateDO; 5 import cn.iocoder.yudao.module.system.dal.dataobject.sms.SmsTemplateDO;
5 import cn.iocoder.yudao.module.system.mq.message.sms.SmsSendMessage; 6 import cn.iocoder.yudao.module.system.mq.message.sms.SmsSendMessage;
6 import cn.iocoder.yudao.module.system.mq.producer.sms.SmsProducer; 7 import cn.iocoder.yudao.module.system.mq.producer.sms.SmsProducer;
@@ -36,11 +37,14 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest { @@ -36,11 +37,14 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest {
36 private SmsSendServiceImpl smsService; 37 private SmsSendServiceImpl smsService;
37 38
38 @Mock 39 @Mock
  40 + private SmsChannelService smsChannelService;
  41 + @Mock
39 private SmsTemplateService smsTemplateService; 42 private SmsTemplateService smsTemplateService;
40 @Mock 43 @Mock
41 private SmsLogService smsLogService; 44 private SmsLogService smsLogService;
42 @Mock 45 @Mock
43 private SmsProducer smsProducer; 46 private SmsProducer smsProducer;
  47 +
44 @Mock 48 @Mock
45 private SmsClientFactory smsClientFactory; 49 private SmsClientFactory smsClientFactory;
46 50
@@ -66,6 +70,9 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest { @@ -66,6 +70,9 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest {
66 String content = randomString(); 70 String content = randomString();
67 when(smsTemplateService.formatSmsTemplateContent(eq(template.getContent()), eq(templateParams))) 71 when(smsTemplateService.formatSmsTemplateContent(eq(template.getContent()), eq(templateParams)))
68 .thenReturn(content); 72 .thenReturn(content);
  73 + // mock SmsChannelService 的方法
  74 + SmsChannelDO smsChannel = randomPojo(SmsChannelDO.class, o -> o.setStatus(CommonStatusEnum.ENABLE.getStatus()));
  75 + when(smsChannelService.getSmsChannel(eq(template.getChannelId()))).thenReturn(smsChannel);
69 // mock SmsLogService 的方法 76 // mock SmsLogService 的方法
70 Long smsLogId = randomLongId(); 77 Long smsLogId = randomLongId();
71 when(smsLogService.createSmsLog(eq(mobile), eq(userId), eq(userType), eq(Boolean.TRUE), eq(template), 78 when(smsLogService.createSmsLog(eq(mobile), eq(userId), eq(userType), eq(Boolean.TRUE), eq(template),
@@ -103,6 +110,9 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest { @@ -103,6 +110,9 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest {
103 String content = randomString(); 110 String content = randomString();
104 when(smsTemplateService.formatSmsTemplateContent(eq(template.getContent()), eq(templateParams))) 111 when(smsTemplateService.formatSmsTemplateContent(eq(template.getContent()), eq(templateParams)))
105 .thenReturn(content); 112 .thenReturn(content);
  113 + // mock SmsChannelService 的方法
  114 + SmsChannelDO smsChannel = randomPojo(SmsChannelDO.class, o -> o.setStatus(CommonStatusEnum.ENABLE.getStatus()));
  115 + when(smsChannelService.getSmsChannel(eq(template.getChannelId()))).thenReturn(smsChannel);
106 // mock SmsLogService 的方法 116 // mock SmsLogService 的方法
107 Long smsLogId = randomLongId(); 117 Long smsLogId = randomLongId();
108 when(smsLogService.createSmsLog(eq(mobile), eq(userId), eq(userType), eq(Boolean.FALSE), eq(template), 118 when(smsLogService.createSmsLog(eq(mobile), eq(userId), eq(userType), eq(Boolean.FALSE), eq(template),