package cn.iocoder.yudao.module.mes.service.energy;
|
|
import cn.iocoder.yudao.module.mes.dal.dataobject.energy.*;
|
import cn.iocoder.yudao.module.mes.dal.mysql.energy.*;
|
import cn.iocoder.yudao.module.system.api.notify.NotifyMessageSendApi;
|
import cn.iocoder.yudao.module.system.api.notify.dto.NotifySendSingleToUserReqDTO;
|
import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO;
|
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
|
import com.baomidou.mybatisplus.core.conditions.Wrapper;
|
import org.apache.ibatis.session.Configuration;
|
import org.junit.jupiter.api.*;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.mockito.*;
|
import org.mockito.junit.jupiter.MockitoExtension;
|
import java.math.BigDecimal;
|
import java.time.LocalDate;
|
import java.util.*;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.*;
|
|
@ExtendWith(MockitoExtension.class)
|
class EnergyAlertServiceTest {
|
@Mock EnergyAlertRuleMapper ruleMapper;
|
@Mock EnergyAlertLogMapper logMapper;
|
@Mock EnergyRecordMapper recordMapper;
|
@Mock AdminUserService userService;
|
@Mock NotifyMessageSendApi notifyApi;
|
@InjectMocks EnergyAlertService service;
|
final LocalDate date = LocalDate.of(2026, 10, 10);
|
|
private EnergyAlertRuleDO rule() {
|
var rule = new EnergyAlertRuleDO(); rule.setId(1L); rule.setEnabled(true); rule.setName("燕窝每日用水");
|
rule.setEnergyType("WATER"); rule.setMetric("QUANTITY"); rule.setUsageArea("燕窝生产区");
|
rule.setProductCategory("燕窝"); rule.setThreshold(new BigDecimal("100")); rule.setRecipientIds("7"); return rule;
|
}
|
private void prepare(EnergyAlertRuleDO rule, String actual) {
|
when(ruleMapper.selectList(any(Wrapper.class))).thenReturn(List.of(rule));
|
when(ruleMapper.lock(1L)).thenReturn(rule);
|
when(recordMapper.selectDailyValue(date, rule)).thenReturn(new BigDecimal(actual));
|
}
|
private void user() {
|
AdminUserDO user = new AdminUserDO(); user.setId(7L); user.setNickname("能耗管理员");
|
when(userService.getUser(7L)).thenReturn(user);
|
}
|
@Test void equalThresholdDoesNotNotify() {
|
prepare(rule(), "100.000"); assertEquals(0, service.check(date).get("sent"));
|
verifyNoInteractions(notifyApi, logMapper);
|
assertFalse(EnergyAlertService.exceeds(BigDecimal.ZERO, BigDecimal.ONE));
|
}
|
@Test void successCreatesMessageSnapshotAndRepeatedCheckDoesNotSendAgain() {
|
prepare(rule(), "101"); user(); when(notifyApi.sendSingleMessageToAdmin(any())).thenReturn(33L);
|
assertEquals(1, service.check(date).get("sent"));
|
ArgumentCaptor<EnergyAlertLogDO> history = ArgumentCaptor.forClass(EnergyAlertLogDO.class);
|
verify(logMapper).insert(history.capture());
|
var stored = history.getValue(); stored.setId(9L);
|
assertEquals("SENT", stored.getStatus()); assertEquals(33L, stored.getMessageId());
|
assertEquals(new BigDecimal("101"), stored.getActualValue());
|
ArgumentCaptor<NotifySendSingleToUserReqDTO> message = ArgumentCaptor.forClass(NotifySendSingleToUserReqDTO.class);
|
verify(notifyApi).sendSingleMessageToAdmin(message.capture());
|
assertEquals("m³", message.getValue().getTemplateParams().get("unit"));
|
assertEquals("2026-10-10", message.getValue().getTemplateParams().get("date"));
|
when(logMapper.selectOne(any(Wrapper.class))).thenReturn(stored);
|
assertEquals(0, service.check(date).get("sent"));
|
verify(notifyApi, times(1)).sendSingleMessageToAdmin(any());
|
}
|
@Test void disabledTemplateIsFailureAndLaterCheckRetriesSameHistory() {
|
prepare(rule(), "101"); user();
|
when(notifyApi.sendSingleMessageToAdmin(any())).thenReturn(null).thenReturn(44L);
|
assertEquals(1, service.check(date).get("failed"));
|
ArgumentCaptor<EnergyAlertLogDO> history = ArgumentCaptor.forClass(EnergyAlertLogDO.class);
|
verify(logMapper).insert(history.capture()); var stored = history.getValue();
|
assertEquals("FAILED", stored.getStatus()); assertNull(stored.getMessageId()); stored.setId(9L);
|
when(logMapper.selectOne(any(Wrapper.class))).thenReturn(stored);
|
assertEquals(1, service.check(date).get("sent"));
|
verify(logMapper).updateById(stored); assertEquals("SENT", stored.getStatus()); assertEquals("", stored.getErrorMessage());
|
}
|
@Test void removedOrDisabledRuleIsSkippedAfterLock() {
|
var rule = rule(); when(ruleMapper.selectList(any(Wrapper.class))).thenReturn(List.of(rule));
|
rule.setEnabled(false); when(ruleMapper.lock(1L)).thenReturn(rule);
|
assertEquals(0, service.check(date).get("sent")); verifyNoInteractions(recordMapper, notifyApi);
|
}
|
@Test void oneFailedRecipientDoesNotPreventOthersAndMoneyHasYuanUnit() {
|
var rule = rule(); rule.setRecipientIds("7,8"); rule.setMetric("AMOUNT"); prepare(rule, "150.55"); user();
|
AdminUserDO second = new AdminUserDO(); second.setNickname("主管"); when(userService.getUser(8L)).thenReturn(second);
|
when(notifyApi.sendSingleMessageToAdmin(any())).thenThrow(new IllegalStateException("test failure")).thenReturn(77L);
|
var result = service.check(date); assertEquals(1, result.get("sent")); assertEquals(1, result.get("failed"));
|
ArgumentCaptor<NotifySendSingleToUserReqDTO> message = ArgumentCaptor.forClass(NotifySendSingleToUserReqDTO.class);
|
verify(notifyApi, times(2)).sendSingleMessageToAdmin(message.capture());
|
assertEquals("元", message.getValue().getTemplateParams().get("unit"));
|
verify(logMapper, times(2)).insert(any(EnergyAlertLogDO.class));
|
}
|
@Test void dailySqlHasExactScopeFiltersAndBoundParameters() {
|
Configuration configuration = new Configuration(); configuration.addMapper(EnergyRecordMapper.class);
|
var rule = rule(); rule.setUsageArea("' OR 1=1 --"); rule.setMetric("AMOUNT");
|
String sql = configuration.getMappedStatement(EnergyRecordMapper.class.getName()+".selectDailyValue")
|
.getBoundSql(Map.of("date", date, "rule", rule)).getSql();
|
assertTrue(sql.contains("SUM( amount")); assertTrue(sql.contains("usage_area=?"));
|
assertTrue(sql.contains("deleted=0")); assertFalse(sql.contains(rule.getUsageArea()));
|
}
|
@Test void monthlyCostUsesFirstDayAndDeduplicatesDifferentDaysInSameMonth() {
|
var rule = rule(); rule.setMetric("MONTH_AMOUNT");
|
when(ruleMapper.selectList(any(Wrapper.class))).thenReturn(List.of(rule));
|
when(ruleMapper.lock(1L)).thenReturn(rule);
|
when(recordMapper.selectDailyValue(date.withDayOfMonth(1), rule)).thenReturn(new BigDecimal("150.55"));
|
user(); when(notifyApi.sendSingleMessageToAdmin(any())).thenReturn(55L);
|
assertEquals(1, service.check(date).get("sent"));
|
var history = ArgumentCaptor.forClass(EnergyAlertLogDO.class); verify(logMapper).insert(history.capture());
|
var stored = history.getValue(); stored.setId(9L);
|
assertEquals("MONTH", stored.getPeriodType()); assertEquals(date.withDayOfMonth(1), stored.getRecordDate());
|
var message = ArgumentCaptor.forClass(NotifySendSingleToUserReqDTO.class); verify(notifyApi).sendSingleMessageToAdmin(message.capture());
|
assertEquals("energy_monthly_warning", message.getValue().getTemplateCode());
|
assertEquals("2026-10", message.getValue().getTemplateParams().get("date"));
|
assertEquals("元", message.getValue().getTemplateParams().get("unit"));
|
when(logMapper.selectOne(any(Wrapper.class))).thenReturn(stored);
|
assertEquals(0, service.check(date.plusDays(1)).get("sent"));
|
verify(notifyApi, times(1)).sendSingleMessageToAdmin(any());
|
}
|
@Test void monthlyQuantitySqlUsesWholeCalendarMonthAndCostMetricSelectsAmount() {
|
var config = new Configuration(); config.addMapper(EnergyRecordMapper.class);
|
var statement = config.getMappedStatement(EnergyRecordMapper.class.getName() + ".selectDailyValue");
|
var rule = rule(); rule.setMetric("MONTH_QUANTITY");
|
String sql = statement.getBoundSql(Map.of("date", LocalDate.of(2028,2,1), "rule", rule)).getSql();
|
assertTrue(sql.contains("SUM( quantity")); assertTrue(sql.contains("DATE_ADD(?, INTERVAL 1 MONTH)"));
|
assertTrue(sql.contains("record_date >= ?")); assertTrue(sql.contains("record_date < DATE_ADD"));
|
rule.setMetric("MONTH_AMOUNT");
|
assertTrue(statement.getBoundSql(Map.of("date", LocalDate.of(2026,12,1), "rule", rule)).getSql().contains("SUM( amount"));
|
}
|
}
|
