java 单元测试静态mock

原创

mob649e8160f07c 2024-05-10 03:31:50 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8160f07c的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java单元测试静态Mock

在Java的单元测试中，我们经常会遇到需要mock掉某些依赖的情况，以便于专注于测试目标方法的逻辑。通常我们使用的是Mockito等工具来实现对实例方法的mock，但是有时候我们也会遇到需要mock静态方法的情况。本文将介绍如何在Java单元测试中使用PowerMockito来对静态方法进行mock。

PowerMockito简介

PowerMockito是Mockito的一个扩展库，它提供了对静态方法、final类和构造函数的mock支持。通过PowerMockito，我们可以轻松地mock掉静态方法，从而实现对静态方法的单元测试。

使用示例

假设我们有一个简单的类Calculator，其中有一个静态方法add用于对两个数进行相加操作。我们想要对这个静态方法进行单元测试，下面是示例代码：

public class Calculator {
    public static int add(int a, int b) {
        return a + b;
    }
}

现在我们需要对Calculator类中的add方法进行单元测试，我们可以使用PowerMockito来实现对静态方法的mock。

首先，我们需要在pom.xml中添加PowerMockito的依赖：

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>

然后，我们可以编写单元测试代码：

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import static org.powermock.api.mockito.PowerMockito.mockStatic;
import static org.powermock.api.mockito.PowerMockito.when;
import static org.junit.Assert.assertEquals;

@RunWith(PowerMockRunner.class)
@PrepareForTest(Calculator.class)
public class CalculatorTest {

    @Test
    public void testAdd() {
        mockStatic(Calculator.class);
        when(Calculator.add(2, 3)).thenReturn(5);

        int result = Calculator.add(2, 3);

        assertEquals(5, result);
    }
}

在上面的代码中，我们首先使用mockStatic方法来mock静态类Calculator，然后使用when方法来定义当传入参数为2和3时，返回值为5。最后通过断言来验证结果是否符合预期。

序列图

下面是一个使用PowerMockito对静态方法进行mock的序列图示例：

sequenceDiagram
    participant Test
    participant Calculator
    Test->>Calculator: mockStatic(Calculator.class)
    Test->>Calculator: when(Calculator.add(2,3)).thenReturn(5)
    Test->>Calculator: Calculator.add(2,3)
    Calculator-->>Test: return 5