有没有办法获取 HttpHeaders 中设置的 bearerToken 以在测试中执行验证？

Question

我有一个构建 HttpHeaders 的方法。它在登录后获取 bearerToken 并将其设置在标头中。我想验证是否在单元测试中设置了令牌，但我无法找到任何方法从标头获取令牌。

这是构建标头的方法：

public static HttpHeaders buildHeaders() {
    HttpHeaders headers = new HttpHeaders();

    headers.setAccept(Collections.singletonList(MediaType.APPLICATION_JSON));
    String bearerToken = getToken();
    headers.setBearerAuth(bearerToken);

    return headers;
}

这是我的测试：

    @Test
void shouldBuildHeaders() {
    HttpHeaders httpHeaders = Util.buildHeaders();
    assertNotNull(httpHeaders);
    assertEquals(httpHeaders.getAccept(), Collections.singletonList(MediaType.APPLICATION_JSON));

    //I would like to do something like this to verify if the token is being set.
    // String bearerToken = httpHeaders.getBearerToken();
    // assertNotNull(bearerToken);
}

有什么方法可以获取令牌，或者我测试的方式有什么问题吗？

Answer 1

您可以尝试循环遍历标头以查看可用的键，如下所示：

httpHeaders.forEach((key,value)->{
    System.out.println(key+"-->"+value);
});

这允许您检查拥有的所有标头。

但是，在您的情况下，您可以通过查看下面的文档来查找您想要的特定标头（我假设您正在使用 Spring）：
https://docs.spring.io/spring-framework/docs/current/javadoc-api/org/springframework/http/HttpHeaders.html#setBearerAuth-java.lang.String-

标头键实际上是“Authorization”，这意味着您可以使用以下代码访问它的值：

String bearerToken = httpHeaders.getFirst(HttpHeaders.AUTHORIZATION);