Mockito实现增删改查

我会带着你远行 2022-01-21 05:49 605阅读 0赞

Mockito在springboot中的使用

> Mockito和Mock在编写代码测试类是存在相同点都是可以模拟相关的数据，不同之处在于Mockito添加的注解为MockitoJUnitRunner，而Mock添加的是SpringRunner.class

目的：

实现代码的功能

1、公共的测试类（后续的需要继承）

@RunWith(MockitoJUnitRunner.class)
    @AutoConfigureMockMvc
    @SpringBootTest
    @Rollback
    @Transactional
    
    @Ignore
    public class OperationCpProcessApplicationTests {
    }

2、控制器层面的测试（Controller）-- 增删改查

public class FeesControllerTest extends OperationCpProcessApplicationTests {
    
        @InjectMocks
        private FeesController feesController;
    
    /**
    
     * 如遇到test测试结束报:
         * org.springframework.beans.factory.BeanCreationNotAllowedException: Error
         * creating bean with name 'eurekaAutoServiceRegistration': Singleton bean...Do
         * not request a bean from a BeanFactory in a destroy method implementation!
    
    * 此处也可以使用补充中的补充一
    
    */
        @MockBean
        private EurekaAutoServiceRegistration eurekaAutoServiceRegistration;
    
        @Mock
        private FeesService feesService;
    
        @Test
        public void queryFees() {
            /** 正常查询*/
            feesController.queryFees(1,2,"个人所得税");
    
            /** 模拟分页*/
            PageInfo<LsfOperatingCmpname> pageInfo = new PageInfo<>();
            pageInfo.setTotal(100);
            List<LsfOperatingCmpname> LsfOperatingCmpnamesList = new ArrayList<>();
            LsfOperatingCmpname LsfOperatingCmpname = initLsfOperatingCmpname();
            LsfOperatingCmpnamesList.add(LsfOperatingCmpname);
            pageInfo.setList(LsfOperatingCmpnamesList);
    
    //  代码中如果有类似的feesService.queryFees进行的调用，类似该方法进行调用模拟
            Mockito.when(feesService.queryFees(1, 1, "个人所得税")).thenReturn(pageInfo);
            feesController.queryFees(1,1,"个人所得税");
    
            /** 异常模拟*/
            Mockito.doThrow(new RuntimeException("异常")).when(feesService).queryFees(1,2,"个人所得税");
        }
    
        @Test
        public void addFees() {
            /**  正常代码--init相关的为初始化数据源，可以查看补充二 */
            LsfOperatingCmpname Lsf = init();
            JwtCurrentUser jwtCurrentUser = initJwtCurrentUser();
            feesController.addFees(Lsf,jwtCurrentUser);
    
            /**  模拟税费名称不为空的 */
            Mockito.when(feesService.selectByName("个人所得税")).thenReturn(init());
            feesController.addFees(Lsf,jwtCurrentUser);
        }
    
        @Test
        public void editFees() {
            /** 正常编辑*/
            LsfOperatingCmpname Lsf = init();
            eduitLsfOperationCmpname();
            feesController.editFees(Lsf.getId(),eduitLsfOperationCmpname(),initJwtCurrentUser());
        }
    
          @Test
        public void delFees() {
            LsfOperatingCmpname Lsf = init();
            feesController.delFees(Lsf.getId(),initJwtCurrentUser());
        }

补充：

补充一：

/**
     * 如遇到test测试结束报:
     * org.springframework.beans.factory.BeanCreationNotAllowedException: Error
     * creating bean with name 'eurekaAutoServiceRegistration': Singleton bean...Do
     * not request a bean from a BeanFactory in a destroy method implementation!
     */
    @Component
    public class FeignBeanFactoryPostProcessor implements BeanFactoryPostProcessor {
    
       @Override
       public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
          BeanDefinition bd = beanFactory.getBeanDefinition("feignContext");
          bd.setDependsOn("eurekaServiceRegistry", "inetUtils");
       }
    }

补充二：初始化测试用的数据源

public static LsfOperatingCmpname init(){
        LsfOperatingCmpname LsfOperatingCmpname = new LsfOperatingCmpname();
        LsfOperatingCmpname.setId("666");
        LsfOperatingCmpname.setTaxationName("个人所得税");
        LsfOperatingCmpname.setTaxationType("888");
        LsfOperatingCmpname.setTaxAFeeTypeName("个人所得税_基本工资税务");
        LsfOperatingCmpname.setRemark("个人所得税中基本工资税务是最大的，但是为了确保税务的一致性，采取了该设置");
        LsfOperatingCmpname.setCreateTime(new Date());
        LsfOperatingCmpname.setCreatorId("1231");
        LsfOperatingCmpname.setDeleted(1);
        LsfOperatingCmpname.setDeleteTime(new Date());
        LsfOperatingCmpname.setDeletorId("13");
        LsfOperatingCmpname.setLastEditorId("13");
        LsfOperatingCmpname.setLastEditTime(new Date());
        return LsfOperatingCmpname;
    }
    
    public static JwtCurrentUser initJwtCurrentUser(){
        JwtCurrentUser jwtCurrentUser = new JwtCurrentUser();
        jwtCurrentUser.setId("123");
        jwtCurrentUser.setName("test");
        jwtCurrentUser.setCode("123");
        return jwtCurrentUser;
    }
    
    
    
    /**
    
    * 实现excel的导入和导出操作
    
    */
    
    @Test
    public void importExcel() {
        exportDataController.importExcel(null,null,null,CommonDataUtil.initJwtCurrentUser());
        exportDataController.importExcel(new MultipartFile[1],new String[]{"人力资源"},"1",CommonDataUtil.initJwtCurrentUser());
    }
    
    /**
     * 从文档库下载文件
     */
    @Test
    public void exportFile() {
        exportDataController.exportFile("1",CommonDataUtil.initJwtCurrentUser());
        exportDataController.exportFile(null,CommonDataUtil.initJwtCurrentUser());
    }

补充三：

有时注入的Service或者其他注入为空的情况，可以采用Mock的方式或者即使此时类中没有使用到也可以在测试类中进行注入，这个我在代码中确实遇到怎么也注入不了的情况，应为Mock可以模拟Spring构造容器，Mockito的这个功效就不太好了