使用@InjectMocks注释的对象上的SpringBoot单元测试注入属性

时间:2018-09-05 18:30:00

标签: spring unit-testing spring-boot junit mockito

我有一个处理jwt创建的类:

public class JwtTokenHandler {
    @Value("${jwtSecret}")
    private String jwtSecret;

    @Value("${jwtExpirationInMs}")
    private int jwtExpirationInMs;

    public String generateToken(String subject) {
        Date now = new Date();
        Date expiryDate = new Date(now.getTime() + jwtExpirationInMs);

        return Jwts.builder()
                .setSubject(subject)
                .setIssuedAt(new Date())
                .setExpiration(expiryDate)
                .signWith(SignatureAlgorithm.HS512, jwtSecret)
                .compact();
    }
}

这需要 resources 文件夹中的属性文件中的 jwtSecret jwtExpirationInMs

在我的单元测试中,我只想测试该类是否按预期工作并生成JWT令牌,我面临的问题是,即使我在其中有另一个属性文件,在运行测试时这两个属性也不会注入测试/资源文件夹。

我发现有使用 ReflectionTestUtils.setField 的解决方案,但这更像是一种解决方法。

我的单元测试如下:

@RunWith(SpringRunner.class)
public class JwtTokenProviderTest {

    @InjectMocks
    private JwtTokenProvider jwtTokenProvider;

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testGenerateToken() {
        String result = jwtTokenProvider.generateToken("Test");
        assertThat(result).isNotBlank();
    }

1 个答案:

答案 0 :(得分:0)

建议不要使用Spring Boot's @MockedBean注释,而不要使用@InjectMocks。伪造豆要容易得多。