package cn.iocoder.yudao.module.mes.service.safety;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.risk.SafetyRiskSourceSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.safety.SafetyRiskSourceDO;
import cn.iocoder.yudao.module.mes.dal.mysql.safety.SafetyRiskSourceMapper;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import org.springframework.test.util.ReflectionTestUtils;
import java.math.BigDecimal;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;

class SafetyRiskSourceServiceTest {
    @Test void savingAboveLimitCreatesAlarmWithObservationTime() {
        var service=new SafetyRiskSourceService();var mapper=mock(SafetyRiskSourceMapper.class);
        ReflectionTestUtils.setField(service,"mapper",mapper);
        var req=new SafetyRiskSourceSaveReqVO();req.setName("测试危险源");req.setCategory("MAJOR");req.setMetricName("温度");
        req.setUpperLimit(BigDecimal.TEN);req.setCurrentValue(new BigDecimal("11"));
        service.save(req);
        var captor=ArgumentCaptor.forClass(SafetyRiskSourceDO.class);verify(mapper).insert(captor.capture());
        assertEquals("ALARM",captor.getValue().getRiskStatus());assertNotNull(captor.getValue().getObservedAt());
    }
    @Test void invalidRangeDoesNotInsert() {
        var service=new SafetyRiskSourceService();var mapper=mock(SafetyRiskSourceMapper.class);
        ReflectionTestUtils.setField(service,"mapper",mapper);
        var req=new SafetyRiskSourceSaveReqVO();req.setLowerLimit(BigDecimal.TEN);req.setUpperLimit(BigDecimal.ONE);
        assertThrows(RuntimeException.class,()->service.save(req));verify(mapper,never()).insert(any(SafetyRiskSourceDO.class));
    }
}