如何在SpringBoot Mockito与JUnit4测试中排查API调用用户身份

嘿，我来帮你一步步搞定这个问题！你现在用JUnit4和Mockito测试部分类时碰到了API认证错误，想要切换成ADMIN身份调用，核心就是先搞清楚当前测试里到底用的是什么用户身份发起的请求，下面是具体的排查步骤：

1. 检查测试类的用户身份配置 #

先看看你的测试代码里有没有设置用户身份的逻辑：

• 如果是Spring Security环境的测试，有没有用@WithMockUser注解？比如要模拟ADMIN的话应该写成@WithMockUser(username = "ADMIN", roles = {"ADMIN"})，如果注解里写的是USER，那肯定会用普通用户身份发起请求。
• 如果是自定义的认证逻辑，看看有没有类似SecurityContextHolder.getContext().setAuthentication(...)的代码，确认这里传入的认证对象是不是ADMIN身份的。
• 要是用Mockito模拟了认证相关的依赖（比如用户信息服务），检查模拟的返回值是不是ADMIN用户的数据。

2. 打印/断言当前用户身份 #

在发起API调用的代码前，加一段调试代码，直接把当前身份打出来，一目了然：

// 在测试方法里，API调用前添加
Authentication currentAuth = SecurityContextHolder.getContext().getAuthentication();
System.out.println("当前登录用户: " + currentAuth.getName());
System.out.println("用户权限列表: " + currentAuth.getAuthorities());

也可以用断言来强制验证身份，避免遗漏：

assertEquals("ADMIN", currentAuth.getName());
assertTrue(currentAuth.getAuthorities().contains(new SimpleGrantedAuthority("ROLE_ADMIN")));

3. 检查API请求的认证信息 #

如果是调用真实API，抓个请求看看请求头里的认证字段（比如Authorization）对应的是不是ADMIN用户的令牌；如果是用Mockito模拟API客户端，检查模拟的请求参数里有没有携带正确的ADMIN身份标识，比如请求头、用户ID参数等。

4. 排查部分类的身份获取逻辑 #

你测试的是部分类（比如用@InjectMocks注入的类），要确认这个类里是怎么获取用户身份的：

• 是从SecurityContext里直接取？还是通过某个注入的服务获取？
• 如果是方法参数传入身份信息，那测试时有没有给这个方法传入ADMIN身份的参数？

另外要注意，JUnit4里的SecurityContext是线程绑定的，测试方法之间如果不清理，可能会串用之前的用户身份，所以可以在测试类里加个清理逻辑：

@After
public void cleanUpSecurityContext() {
    SecurityContextHolder.clearContext();
}

按照这些步骤走，你就能很快定位到当前用的是什么身份，然后调整成ADMIN身份就可以解决认证错误啦～

内容的提问来源于stack exchange，提问作者Aiyush Saini