package cn.iocoder.yudao.module.crm.service.invoice;
|
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.module.bpm.api.task.BpmProcessInstanceApi;
|
import cn.iocoder.yudao.module.crm.controller.admin.invoice.vo.invoice.CrmInvoiceSaveReqVO;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.contract.CrmContractDO;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.invoice.CrmInvoiceDO;
|
import cn.iocoder.yudao.module.crm.dal.mysql.invoice.CrmInvoiceMapper;
|
import cn.iocoder.yudao.module.crm.service.contract.CrmContractService;
|
import cn.iocoder.yudao.module.system.api.storage.StorageAttachmentApi;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.params.ParameterizedTest;
|
import org.junit.jupiter.params.provider.CsvSource;
|
import org.junit.jupiter.params.provider.ValueSource;
|
import org.mockito.ArgumentCaptor;
|
import org.mockito.InjectMocks;
|
import org.mockito.Mock;
|
import org.mockito.junit.jupiter.MockitoExtension;
|
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
|
import static cn.iocoder.yudao.module.crm.enums.ErrorCodeConstants.INVOICE_UPDATE_FAIL_EDITING_PROHIBITED;
|
import static cn.iocoder.yudao.module.crm.enums.ErrorCodeConstants.INVOICE_CREATE_FAIL_PRICE_EXCEEDS_LIMIT;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.*;
|
|
@ExtendWith(MockitoExtension.class)
|
class CrmInvoiceEditStatusTest {
|
@Mock private CrmInvoiceMapper invoiceMapper;
|
@Mock private CrmContractService contractService;
|
@Mock private StorageAttachmentApi storageAttachmentApi;
|
@Mock private BpmProcessInstanceApi bpmProcessInstanceApi;
|
@InjectMocks private CrmInvoiceServiceImpl service;
|
|
@ParameterizedTest
|
@CsvSource({"20,审核通过", "30,审核不通过", "40,已取消", "50,已作废", "99,未知状态"})
|
void lockedInvoiceReportsStatusBeforeAmountChecks(int status, String name) {
|
when(invoiceMapper.selectById(1L)).thenReturn(invoice(status));
|
ServiceException error = assertThrows(ServiceException.class, () -> service.updateInvoice(request()));
|
assertEquals(INVOICE_UPDATE_FAIL_EDITING_PROHIBITED.getCode(), error.getCode());
|
assertTrue(error.getMessage().contains(name));
|
assertTrue(error.getMessage().contains("仅未提交、审批中"));
|
verify(invoiceMapper, never()).updateById(any(CrmInvoiceDO.class));
|
verifyNoInteractions(contractService, storageAttachmentApi);
|
}
|
|
@ParameterizedTest
|
@ValueSource(ints = {0, 10})
|
void editableInvoiceKeepsApprovalAndRelationshipFields(int status) {
|
verifySuccessfulUpdate(status);
|
}
|
|
@ParameterizedTest
|
@ValueSource(ints = {0, 10, 20, 30, 40, 50})
|
void autoApprovalAllowsEditingEveryInvoiceStatus(int status) {
|
when(bpmProcessInstanceApi.isGlobalAutoApprove()).thenReturn(true);
|
verifySuccessfulUpdate(status);
|
}
|
|
@Test
|
void autoApprovalStillValidatesContractAmount() {
|
when(bpmProcessInstanceApi.isGlobalAutoApprove()).thenReturn(true);
|
when(invoiceMapper.selectById(1L)).thenReturn(invoice(20));
|
var contract = new CrmContractDO().setId(2L).setTotalPrice(new BigDecimal("50"));
|
when(contractService.validateContract(2L)).thenReturn(contract);
|
when(invoiceMapper.selectListByContractIdAndStatus(eq(2L), anyCollection())).thenReturn(new ArrayList<>());
|
var error = assertThrows(ServiceException.class, () -> service.updateInvoice(request()));
|
assertEquals(INVOICE_CREATE_FAIL_PRICE_EXCEEDS_LIMIT.getCode(), error.getCode());
|
verify(invoiceMapper, never()).updateById(any(CrmInvoiceDO.class));
|
verifyNoInteractions(storageAttachmentApi);
|
}
|
|
private void verifySuccessfulUpdate(int status) {
|
when(invoiceMapper.selectById(1L)).thenReturn(invoice(status));
|
var contract = new CrmContractDO().setId(2L).setTotalPrice(new BigDecimal("500"));
|
when(contractService.validateContract(2L)).thenReturn(contract);
|
when(invoiceMapper.selectListByContractIdAndStatus(eq(2L), anyCollection())).thenReturn(new ArrayList<>());
|
service.updateInvoice(request());
|
var update = ArgumentCaptor.forClass(CrmInvoiceDO.class);
|
verify(invoiceMapper).updateById(update.capture());
|
assertNull(update.getValue().getAuditStatus());
|
assertEquals(2L, update.getValue().getContractId());
|
assertEquals(3L, update.getValue().getCustomerId());
|
assertEquals(4L, update.getValue().getOwnerUserId());
|
verify(storageAttachmentApi).updateAttachments("file", "crm_invoice", 1L, null);
|
}
|
|
private static CrmInvoiceDO invoice(int status) {
|
return new CrmInvoiceDO().setId(1L).setContractId(2L).setCustomerId(3L)
|
.setOwnerUserId(4L).setAuditStatus(status).setPrice(new BigDecimal("100"));
|
}
|
|
private static CrmInvoiceSaveReqVO request() {
|
return new CrmInvoiceSaveReqVO().setId(1L).setContractId(999L).setCustomerId(999L)
|
.setOwnerUserId(999L).setPrice(new BigDecimal("100"));
|
}
|
}
|
