如何在单元测试SpringBoot中解析JWT令牌

Question

我有一个带有 Spring boot 的微服务设置和带有 JWT 的 OAuth 2。 我的 JWT 令牌中有其他字段。

我的大多数服务都调用一个静态方法，该方法具有令牌中附加字段的本地线程。
如何为此类服务编写单元测试？
即使我尝试注入一个模拟用户，它也不起作用，而且我找不到发送 JWT 的方法，因为我没有测试控制器。

代码：

SecurityUtils static Calss (also check the package for other relevant JWT handler) .

Example on a method that will call the static class (Line 79).

方法：

public CommonResponse saveUpdateProfile(ProfileRequest request) {

    String authUsername = SecurityUtils.getUsername();

    Optional<ProfileEntity> optionalProfile = findProfileByUsername(authUsername);

    ProfileEntity profile;
    if (optionalProfile.isPresent()) {
        profile = optionalProfile.get();
    } else {
        profile = ProfileEntity.builder()
                .username(authUsername)
                .build();
    }

    profile.setFirstName(request.getFirstName());
    profile.setLastName(request.getLastName());

    ProfileEntity savedProfile = profileRepository.save(profile);

    if (savedProfile == null) {
        throw new RuntimeException("Failed to save user in database");
    }

    return CommonResponse.ok(savedProfile);
}

感谢所有帮助。

Answer 1

好的，这是使用静态方法时的常见问题。你不能轻易地覆盖它们，例如在测试中。我想我要做的就是把你的SecurityUtils 类变成一个服务并让它实现一个接口。然后将此接口注入到需要使用它的任何其他服务中，而不是调用静态方法。然后，您可以轻松地为您的测试提供另一个实现。

所以你会有类似的东西：

interface SecurityUtils {
    String getUsername();
    ...
}

@Service
class MySecurityUtils immplements SecurityUtils {
    private JwtToken getJwtToken() {
        return MySecurityContextHolder.getContext().getJwtToken();
    }

    public String getUsername() {
        return getJwtToken().getUsername();
    }
    ...
}

然后在单元测试中，您可以将SecurityUtils 的模拟实现注入您正在测试的任何类。