模拟JWT UTIL以验证令牌

Question

我想为此endpoint创建JUnkt测试：

    @Autowired
    private JwtTokenProvider jwtTokenProvider;

    @PostMapping("reset_token")
    public ResponseEntity<?> resetToken(@Valid @RequestBody ResetPasswordTokenDTO resetPasswordTokenDTO, BindingResult bindResult) {

        final String login = jwtTokenProvider.getUsername(resetPasswordTokenDTO.getResetPasswordToken());
    }

完整代码：Github

JUnit测试：

@Test
public void resetTokenTest_NOT_FOUND() throws Exception {
    when(usersService.findByResetPasswordToken(anyString())).thenReturn(Optional.empty());

    mockMvc.perform(post("/users/reset_token")
            .contentType(MediaType.APPLICATION_JSON)
            .content(ResetPasswordTokenDTO))
            .andExpect(status().isNotFound());
}

当我运行代码时，我在这一行得到NPE：

final String login = jwtTokenProvider.getUsername(resetPasswordTokenDTO.getResetPasswordToken());

我如何正确地模拟jwtTokenProvider？正如你所见，我有一个包含测试数据的文件，我加载了该文件，但没有提取令牌。你知道我如何解决这个问题吗？

Answer 1

您可以考虑直接在测试类中使用@MockBean来模拟您的JwtTokenProvider。@MockBean注释是Spring式的，包含在Spring-boo-starter-test中。Spring Boot留档总结得很好：

Spring Boot包含@MockBean注释，可用于为ApplicationContext中的bean定义Mockito mock。可以使用注释添加新bean或替换单个现有bean定义。注释可以直接用于测试类、测试中的字段或@Configuration类和字段。在字段上使用时，还将注入已创建模拟的实例。模拟bean在每个测试方法之后都会自动重置。

@MockBean注释将使Spring在其应用程序上下文中查找类型为JwtTokenProvider的现有单个bean。如果存在，mock将替换该bean，如果不存在，则在应用程序上下文中添加新的mock。

您的测试类将如下所示：

import org.springframework.boot.test.mock.mockito.MockBean;

@MockBean
@Qualifier("xxx") //If there is more than one bean of type JwtTokenProvider
private JwtTokenProvider jwtTokenProvider;

@Test
public void resetTokenTest_NOT_FOUND() throws Exception {

    when(jwtTokenProvider.getUsername(anyString())).thenReturn(Optional.empty());

    mockMvc.perform(post("/users/reset_token")
            .contentType(MediaType.APPLICATION_JSON)
            .content(ResetPasswordTokenDTO))
            .andExpect(status().isNotFound());
}

你可能还想检查这个和这个。

Answer 2

最直接的方法是使用Mockito并创建mock实例，然后使用构造函数注入将其直接传递给控制器类。

然而，如果您不希望使用构造函数注入（我建议您使用它，因为它更显式），那么您需要在单独的测试配置类中定义bean

@Profile("test")
@Configuration

public class TestConfiguration {
    @Bean
    public JwtTokenProvider mockJwtTokenProvider() {
        return Mockito.mock(JwtTokenProvider.class);
    }

}

另外，通过@ActiveProfile（"test"）将正确的配置文件添加到您的测试类中