PowerMockito测试框架使用讲解

最新推荐文章于 2024-07-26 17:29:37 发布

qq_26893485

最新推荐文章于 2024-07-26 17:29:37 发布

阅读量1.9k

点赞数 3

CC 4.0 BY-SA版权

分类专栏： java PowerMokito

本文链接：https://blog.youkuaiyun.com/qq_26893485/article/details/115001955

java 同时被 2 个专栏收录

4 篇文章

订阅专栏

PowerMokito

1 篇文章

订阅专栏

本文介绍PowerMockito测试框架的基本注解及其应用场景，包括如何模拟静态方法、构造虚拟对象及替换new操作等。通过实例演示了如何针对不同情况选择合适的模拟策略。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

PowerMockito测试框架快速使用

1.注解讲解

@PrepareForTest //测试类测试原始类所需的静态类的声明
语法：@PrepareForTest（value = {A.class, B.class}）

@InjectMocks //测试类的注入注解，相当于Sping的@Autowired,
可以理解成new一个对象

@Mocks //模拟一个虚拟的对象，当测试类不走dao逻辑，可以在测试类中模拟一个虚拟的dao，进而测试原有代码的业务逻辑

@BeforeMethod //测试类执行之前执行

@AfterMethod //测试类运行之后执行

@Test //测试方法

2.简单用法

场景1：被测类中不使用new创建对象，如用getInstance()
解决方案：

首先在测试类中声明静态类对象@PrepareForTest
在测试类执行前初始化时，先mock一个虚拟的对象
当调用getInstance方法时，就使用@Mock注入的虚拟对象
这样调用到被测类中的静态类时，将会使用你模拟的那个虚拟类（既thenReturn的对象）

场景2：mock一些工具类的静态方法
解决方案：

首先要在测试类中声明一下要使用的工具类@PrepareForTest
声明完成后再代码中mockStatic一个对象
模拟掉被测类运行中调用工具类的静态方法
这样运行的时候就不会走工具类的静态方法，调用到时会默认返回一个值，在thenReturn中自行设置

场景3：当被测类中的方法内出现new对象的存在时，如何虚拟掉
5. 先创建一个虚拟的对象@Mock
在这里插入图片描述
2.测试方法执行前首先执行一行代码（代码意思是，当被测类方法内出现new某个对象时，使用步骤1被虚拟出来的对象）

场景四：方法讲解

判断输出结果与预计输出结果是否一样，不一样则报错（方法：assertEquals(source, target)）
断言方法：verify方法。判断此方法在测试的时候执行了几次

场景五：若dao中的方法没有返回类型
若mock的dao中的某个方法没有返回的参数，那么这个方法将不用mock

案例

注意：

创建测试对象时，命名格式，被测类后缀加Test
@Test运行前，先写一个方法@BeforeMethod，如下：
正常写测试方法即可。

案列：

/**
 * @Author: yyk
 * @Date: Created in 2021/3/18 17:29
 * @Description： 单元测试
 * @Modified By：
 * @Version: 1.0$
 * @Demind And Code:
 */
@PrepareForTest(value = {ArrearageAgeFinalDao.class, EsopUserManagerRelaDao.class, PermissionsMgmt.class, AcctEsopCommonMgt.class,
        DecimalPointLeaveUtil.class})
public class QueryArrearageAgeServiceTest extends Template {
    @InjectMocks
    private QueryArrearageAgeService queryArrearageAgeService;
    @Mock
    private ArrearageAgeFinalDao arrearageAgeFinalDaoMock;
    @Mock
    private EsopUserManagerRelaDao esopUserManagerRelaDaoMock;

    @BeforeMethod
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testCallService() throws Exception {
        //模拟掉对应的dao
        initPoverMockQueryArrearageAgeDao();
        queryArrearageAgeService.callService(initMessage());
    }

    /**初始化请求消息
     * @return
     */
    private Message initMessage() {
        Map<String, Object> reqMap = new HashMap();
        reqMap.put("accept_id", "200833491133");
        reqMap.put("request_source", "20");
        reqMap.put("request_time", "20210305143816");
        reqMap.put("operator_id", "jiangwk");
        Map<String, Object> orderContent = new HashMap();
        orderContent.put("order_type", "1700036");
        reqMap.put("order_content", orderContent);
        Map<String, Object> orderReq = new HashMap();
        orderReq.put("accountId", "1000122606");
        orderReq.put("customerId", "1000122606");
        orderReq.put("billMonthStart", "202101");
        orderReq.put("billMonthEnd", "202103");
        orderReq.put("operationType", "1");
        orderContent.put("order_req", orderReq);
        Message msg = new Message(reqMap);
        return msg;
    }

    /**模拟掉dao调用的方法
     */
    private void initPoverMockQueryArrearageAgeDao() {
//        模拟掉静态创建的dao
        PowerMockito.mockStatic(PermissionsMgmt.class);
        PowerMockito.when(PermissionsMgmt.getJkrOrgZc(Mockito.anyString(), Mockito.anyString(), Mockito.anyString())).thenReturn(initStrings());
        PowerMockito.mockStatic(ArrearageAgeFinalDao.class);
        PowerMockito.when(ArrearageAgeFinalDao.getInstance()).thenReturn(arrearageAgeFinalDaoMock);
        PowerMockito.mockStatic(EsopUserManagerRelaDao.class);
        PowerMockito.when(EsopUserManagerRelaDao.newInstance()).thenReturn(esopUserManagerRelaDaoMock);

        PowerMockito.mockStatic(AcctEsopCommonMgt.class);
        PowerMockito.when(AcctEsopCommonMgt.acctEsopCommon(Mockito.anyString(), Mockito.anyString(),
                Mockito.anyString(), Mockito.anyString())).thenReturn(initMap());

        PowerMockito.mockStatic(DecimalPointLeaveUtil.class);
        PowerMockito.when(DecimalPointLeaveUtil.getFloor(Mockito.anyString())).thenReturn("0.00");

//        模拟掉方法
        PowerMockito.when(arrearageAgeFinalDaoMock.queryAcctArrearagePage(Mockito.anyMap(), Mockito.anyInt(),
                Mockito.anyInt(), Mockito.anyString())).thenReturn(initPageAcct());
        PowerMockito.when(arrearageAgeFinalDaoMock.queryAcctArrearageAge(Mockito.anyMap(), Mockito.anyString())).thenReturn(initAgeBeanList());
        PowerMockito.when(esopUserManagerRelaDaoMock.checkPermissions(Mockito.anyString(), Mockito.anyString())).thenReturn(true);
    }

    /**初始化initMap
     * @return
     */
    private Map<String, String> initMap() {
        Map map = new HashMap<String, String>();
        map.put("customerName", "testCustomerName");
        return map;
    }

    /**初始化String集合
     * @return
     */
    private List<String> initStrings() {
        List<String> stringList = new ArrayList<String>();
        stringList.add("0000");
        return stringList;
    }

    /**模拟ageBean对象
     * @return
     */
    private List initAgeBeanList() {
        List<AcctArrearagrBean> ageBeanList = new ArrayList<>();
        AcctArrearagrBean acctArrearagrBean = new AcctArrearagrBean();
        acctArrearagrBean.setCustomerName("testCustomerName");
        ageBeanList.add(acctArrearagrBean);
        return ageBeanList;
    }

    /**
     * 模拟acct类
     * @return
     */
    private Page initPageAcct() {
        Page<AcctArrearagrBean> page = new Page<AcctArrearagrBean>();
        return page;
    }

}