java - 为Spring TestRestTemplate集成测试添加标题值

我正在使用TestRestTemplate对我们的产品进行集成测试。

我有一个看起来像这样的测试:

@Test
public void testDeviceQuery() {
    ResponseEntity<Page> deviceInfoPage = template.getForEntity(base, Page.class);

    // validation code here
}

此特定请求需要一个Header值。有人可以让我知道如何将头添加到TestRestTemplate调用吗？

最佳答案

如果您希望所有使用TestRestTemplate的请求都包含某些 header ，则可以在设置中添加以下内容:

testRestTemplate.getRestTemplate().setInterceptors(
        Collections.singletonList((request, body, execution) -> {
            request.getHeaders()
                    .add("header-name", "value");
            return execution.execute(request, body);
        }));