在Java单元测试中，我们经常需要模拟（mock）依赖项以隔离被测代码。对于实例方法，Mockito提供了直观的Mockito.when().thenReturn()语法。然而，当涉及到静态方法时，传统的模拟方式会失效。例如，直接尝试使用Mockito.when(StaticClass.staticMethod(Mockito.any()))来模拟一个带参数的静态方法，通常会导致"You cannot use argument matchers outside of verification or stubbing."的错误。这是因为Mockito在设计上，其when语法主要用于实例方法，并且需要一个实际的实例来触发桩行为。对于静态方法，我们需要一种不同的机制来拦截其调用。
从Mockito 3.4.0版本开始，引入了MockedStatic接口，专门用于解决静态方法的模拟问题。MockedStatic允许我们在一个特定的作用域内，对某个类的所有静态方法进行桩或验证。
使用MockedStatic的关键在于try-with-resources语句。这确保了静态方法的模拟只在指定的代码块内有效，并在块结束后自动恢复其原始行为，从而避免对其他测试造成副作用。
以下是MockedStatic的基本结构：
import org.junit.jupiter.api.Test;
import org.mockito.MockedStatic;
import org.mockito.Mockito;
import static org.junit.jupiter.api.Assertions.assertEquals;
// 假设有一个静态类 Foo
class Foo {
public static String method() {
return "original";
}
public static String methodWithArg(String arg) {
return "original-" + arg;
}
}
public class StaticMethodMockingExample {
@Test
void testStaticMethodMocking() {
assertEquals("original", Foo.method()); // 原始方法调用
try (MockedStatic<Foo> mockedFoo = Mockito.mockStatic(Foo.class)) {
mockedFoo.when(Foo::method).thenReturn("mocked");
assertEquals("mocked", Foo.method()); // 模拟后的方法调用
mockedFoo.verify(Foo::method); // 验证方法是否被调用
}
assertEquals("original", Foo.method()); // 模拟作用域结束后，方法恢复原始行为
}
}对于本教程的核心问题——如何模拟带参数的静态方法，MockedStatic同样提供了优雅的解决方案。当静态方法带有参数，并且我们希望使用参数匹配器（如Mockito.any()）时，需要将桩定义放在一个lambda表达式中。
考虑以下场景：我们有一个AuthorizationHelper类，其中包含一个静态方法checkJwt，用于验证JsonWebToken。
import io.quarkus.security.identity.SecurityContext;
import io.smallrye.jwt.auth.principal.JsonWebToken;
import javax.ws.rs.core.Response;
// 辅助类
class AuthorizationHelper {
public static boolean checkJwt(JsonWebToken jsonWebToken) {
return jsonWebToken != null && jsonWebToken.getName() != null;
}
}
// 控制器代码片段
class MyController {
private SecurityContext securityContext; // 假设通过@Context注入
public Response getUsers() {
JsonWebToken jsonWebToken = (JsonWebToken) securityContext.getUserPrincipal();
if (!AuthorizationHelper.checkJwt(jsonWebToken)) {
return Response.status(Response.Status.UNAUTHORIZED).build();
}
// ... 其他逻辑
return Response.ok().build();
}
}为了在测试中模拟AuthorizationHelper.checkJwt方法，使其在接收任何JsonWebToken实例时都返回true，我们可以这样做：
import io.quarkus.security.identity.SecurityContext;
import io.smallrye.jwt.auth.principal.JsonWebToken;
import org.junit.jupiter.api.Test;
import org.mockito.MockedStatic;
import org.mockito.Mockito;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.mockito.Mockito.any;
public class AuthorizationHelperTest {
@Test
void testGetUsersWithMockedStaticAuthorization() {
// 模拟SecurityContext及其返回的JsonWebToken
SecurityContext securityContext = mock(SecurityContext.class);
JsonWebToken mockJwt = mock(JsonWebToken.class);
when(securityContext.getUserPrincipal()).thenReturn(mockJwt);
when(mockJwt.getName()).thenReturn("testUser"); // 确保Jwt本身是有效的
// 使用MockedStatic模拟AuthorizationHelper.checkJwt方法
try (MockedStatic<AuthorizationHelper> mockedAuthorizationHelper = Mockito.mockStatic(AuthorizationHelper.class)) {
// 关键：使用lambda表达式来定义带参数的静态方法的桩
mockedAuthorizationHelper.when(() -> AuthorizationHelper.checkJwt(any(JsonWebToken.class))).thenReturn(true);
// 实例化并调用控制器方法
MyController controller = new MyController();
// 注意：在实际Quarkus测试中，SecurityContext通常通过@InjectMock或类似机制注入
// 这里为了示例，假设可以直接设置
// Field injection example (not recommended for production code, but useful for testing private fields)
try {
java.lang.reflect.Field field = MyController.class.getDeclaredField("securityContext");
field.setAccessible(true);
field.set(controller, securityContext);
} catch (NoSuchFieldException | IllegalAccessException e) {
e.printStackTrace();
}
// 假设我们正在测试一个HTTP端点，这里简化为直接调用控制器方法
// 在实际Quarkus测试中，你会使用RestAssured等工具
// given()
// .when().get("/api/v1/users")
// .then()
// .statusCode(200);
// 示例：直接调用控制器方法并验证其行为
javax.ws.rs.core.Response response = controller.getUsers();
// 验证控制器返回的状态码
assertEquals(200, response.getStatus());
// 验证AuthorizationHelper.checkJwt是否被调用
mockedAuthorizationHelper.verify(() -> AuthorizationHelper.checkJwt(any(JsonWebToken.class)));
}
}
@Test
void testGetUsersWithMockedStaticAuthorization_Unauthorized() {
// 模拟SecurityContext及其返回的JsonWebToken
SecurityContext securityContext = mock(SecurityContext.class);
JsonWebToken mockJwt = mock(JsonWebToken.class);
when(securityContext.getUserPrincipal()).thenReturn(mockJwt);
when(mockJwt.getName()).thenReturn("testUser"); // 确保Jwt本身是有效的
// 使用MockedStatic模拟AuthorizationHelper.checkJwt方法
try (MockedStatic<AuthorizationHelper> mockedAuthorizationHelper = Mockito.mockStatic(AuthorizationHelper.class)) {
// 关键：使用lambda表达式来定义带参数的静态方法的桩，使其返回false
mockedAuthorizationHelper.when(() -> AuthorizationHelper.checkJwt(any(JsonWebToken.class))).thenReturn(false);
MyController controller = new MyController();
try {
java.lang.reflect.Field field = MyController.class.getDeclaredField("securityContext");
field.setAccessible(true);
field.set(controller, securityContext);
} catch (NoSuchFieldException | IllegalAccessException e) {
e.printStackTrace();
}
javax.ws.rs.core.Response response = controller.getUsers();
assertEquals(401, response.getStatus()); // 预期返回401 Unauthorized
mockedAuthorizationHelper.verify(() -> AuthorizationHelper.checkJwt(any(JsonWebToken.class)));
}
}
}在上述代码中：
通过MockedStatic接口，Mockito成功解决了静态方法模拟的难题，特别是对于那些需要参数匹配器的场景。掌握MockedStatic的正确用法，尤其是结合try-with-resources和lambda表达式来定义带参数静态方法的桩，将大大提升您在复杂Java应用中进行单元测试的能力。尽管如此，我们也应审视代码设计，尽量减少对静态方法的过度依赖，以促进更易于测试和维护的架构。
以上就是如何使用Mockito模拟带参数的静态方法的详细内容，更多请关注php中文网其它相关文章！
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移，垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是，许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
199
收藏
