package jnpf.dmsService.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
import jnpf.dmsEntity.DmsWenjianBianhaoGuizeEntity;
|
import jnpf.dmsEntity.DmsWenjianBianhaoGuizeFeiqiEntity;
|
import jnpf.dmsMapper.DmsWenjianBianhaoGuizeFeiqiMapper;
|
import jnpf.dmsMapper.DmsWenjianBianhaoGuizeMapper;
|
import jnpf.exception.DataException;
|
import org.apache.ibatis.builder.MapperBuilderAssistant;
|
import org.apache.ibatis.session.Configuration;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.mockito.ArgumentCaptor;
|
import org.mockito.InjectMocks;
|
import org.mockito.Mock;
|
import org.mockito.junit.jupiter.MockitoExtension;
|
|
import java.util.Collections;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.isNull;
|
import static org.mockito.Mockito.times;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
@ExtendWith(MockitoExtension.class)
|
class DmsWenjianBianhaoGuizeFeiqiServiceImplTest {
|
|
@Mock private DmsWenjianBianhaoGuizeFeiqiMapper feiqiMapper;
|
@Mock private DmsWenjianBianhaoGuizeMapper guizeMapper;
|
|
@InjectMocks
|
private DmsWenjianBianhaoGuizeFeiqiServiceImpl service;
|
|
@BeforeEach
|
void initTableInfo() {
|
initTableInfo(DmsWenjianBianhaoGuizeFeiqiEntity.class);
|
initTableInfo(DmsWenjianBianhaoGuizeEntity.class);
|
}
|
|
@Test
|
void writesFeiqiIdBackToTheRelatedRuleBeforeDisablingIt() throws DataException {
|
DmsWenjianBianhaoGuizeFeiqiEntity relation = new DmsWenjianBianhaoGuizeFeiqiEntity();
|
relation.setId("feiqi-1");
|
relation.setGuizeId("guize-1");
|
when(feiqiMapper.selectList(any())).thenReturn(Collections.singletonList(relation));
|
when(guizeMapper.selectCount(any())).thenReturn(1L);
|
|
service.feiqiGuize("flow-1", "task-1");
|
|
ArgumentCaptor<LambdaUpdateWrapper> updateCaptor = ArgumentCaptor.forClass(LambdaUpdateWrapper.class);
|
verify(guizeMapper, times(2)).update(isNull(), updateCaptor.capture());
|
List<LambdaUpdateWrapper> updates = updateCaptor.getAllValues();
|
|
LambdaUpdateWrapper feiqiIdUpdate = updates.get(0);
|
assertTrue(feiqiIdUpdate.getSqlSet().contains("feiqi_id"));
|
assertTrue(feiqiIdUpdate.getParamNameValuePairs().containsValue("feiqi-1"));
|
assertTrue(feiqiIdUpdate.getCustomSqlSegment().contains("f_id"));
|
|
assertTrue(updates.get(1).getSqlSet().contains("biz_enabled"));
|
}
|
|
private void initTableInfo(Class<?> entityClass) {
|
if (TableInfoHelper.getTableInfo(entityClass) == null) {
|
TableInfoHelper.initTableInfo(new MapperBuilderAssistant(new Configuration(), ""), entityClass);
|
}
|
}
|
}
|
