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 1 package cn.iocoder.yudao.module.system.service.sms;
2 2  
3 3 import cn.hutool.core.map.MapUtil;
  4 +import cn.iocoder.yudao.module.system.dal.dataobject.sms.SmsChannelDO;
4 5 import cn.iocoder.yudao.module.system.dal.dataobject.sms.SmsTemplateDO;
5 6 import cn.iocoder.yudao.module.system.mq.message.sms.SmsSendMessage;
6 7 import cn.iocoder.yudao.module.system.mq.producer.sms.SmsProducer;
... ... @@ -36,11 +37,14 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest {
36 37 private SmsSendServiceImpl smsService;
37 38  
38 39 @Mock
  40 + private SmsChannelService smsChannelService;
  41 + @Mock
39 42 private SmsTemplateService smsTemplateService;
40 43 @Mock
41 44 private SmsLogService smsLogService;
42 45 @Mock
43 46 private SmsProducer smsProducer;
  47 +
44 48 @Mock
45 49 private SmsClientFactory smsClientFactory;
46 50  
... ... @@ -66,6 +70,9 @@ public class SmsSendServiceTest extends BaseMockitoUnitTest {
66 70 String content = randomString();
67 71 when(smsTemplateService.formatSmsTemplateContent(eq(template.getContent()), eq(templateParams)))
68 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 76 // mock SmsLogService 的方法
70 77 Long smsLogId = randomLongId();
71 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 110 String content = randomString();
104 111 when(smsTemplateService.formatSmsTemplateContent(eq(template.getContent()), eq(templateParams)))
105 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 116 // mock SmsLogService 的方法
107 117 Long smsLogId = randomLongId();
108 118 when(smsLogService.createSmsLog(eq(mobile), eq(userId), eq(userType), eq(Boolean.FALSE), eq(template),
... ...