如何mock静态方法时传递参数

在Java开发中，经常会遇到需要mock静态方法的情况，例如测试一个类的方法时，该方法内部调用了一个静态方法。但是mock静态方法时，如何传递参数是一个常见的问题。本文将介绍如何使用Mockito和PowerMockito来mock静态方法并传递参数，以解决这个问题。

实际问题

假设我们有一个Calculator类，其中有一个divide方法，该方法调用了一个静态方法Util.divide进行除法运算。我们想要测试Calculator类中的divide方法，但是需要mock掉Util.divide方法，并且传递参数。

public class Calculator {
    public int divide(int dividend, int divisor) {
        return Util.divide(dividend, divisor);
    }
}

public class Util {
    public static int divide(int dividend, int divisor) {
        // 除法运算
        return dividend / divisor;
    }
}

解决方案

使用Mockito

首先，我们需要引入Mockito和PowerMockito的依赖：

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>x.x.x</version>
    <scope>test</scope>
</dependency>

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>x.x.x</version>
    <scope>test</scope>
</dependency>

然后，我们可以使用Mockito和PowerMockito来mock静态方法并传递参数。下面是一个示例：

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.anyInt;

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

    @Test
    public void testDivide() {
        PowerMockito.mockStatic(Util.class);

        PowerMockito.when(Util.divide(anyInt(), anyInt())).thenAnswer(invocation -> {
            int dividend = invocation.getArgument(0);
            int divisor = invocation.getArgument(1);
            return dividend / divisor;
        });

        Calculator calculator = new Calculator();
        int result = calculator.divide(10, 2);

        assertEquals(5, result);
    }
}

在上面的示例中，我们使用PowerMockito.mockStatic(Util.class)来mock静态方法Util.divide，然后使用PowerMockito.when(Util.divide(anyInt(), anyInt())).thenAnswer来定义模拟的行为，并传递参数。最后，我们通过断言验证了方法的正确性。

饼状图

pie
    title 静态方法mock
    "Mockito" : 60
    "PowerMockito" : 40

状态图

stateDiagram
    [*] --> Mocking
    Mocking --> Passed
    Mocking --> Failed

结论

通过本文的介绍，我们学习了如何使用Mockito和PowerMockito来mock静态方法并传递参数。这种方法可以帮助我们解决测试中的一些实际问题，提高代码的质量和可测试性。希望本文对你有所帮助！