如何测试 java.net.http (Java 11) 请求 BodyPublisher？

Question

我正在尝试测试使用新的 Java 11 java.net.http.HttpClient代码。

在我的生产代码中，我有这样的事情：

HttpClient httpClient = ... (gets injected)

HttpRequest request = HttpRequest.newBuilder().uri(URI.create("http://localhost:1234"))
        .POST(HttpRequest.BodyPublishers.ofByteArray("example".getBytes()))
        .build();

return httpClient.send(request, HttpResponse.BodyHandlers.ofByteArray());

在我的测试中，我模拟了HttpClient并获得了java.net.http.HttpRequest 。 如何获取/测试其请求正文（= 我的"example" ）？ 我可以调用request.bodyPublisher()来获取HttpRequest.BodyPublisher ，但后来我被卡住了。

• 我试图将它转换为jdk.internal.net.http.RequestPublishers.ByteArrayPublisher （它实际上是），但它不会编译，因为模块没有导出相应的包。
• 我已经检查了HttpRequest.BodyPublisher （ .contentLength() , .subscribe(subscriber) ）中的可用方法，但我想它们不可能。
• 我试图创建一个新的BodyPublisher并使用.equals()比较它们，但没有真正的实现，所以比较总是错误的。

Answer 1

如果您对处理程序中 body 的外观感兴趣，您可以在 HttpRequest.BodyPublisher Subscriber 的帮助下了解它。 我们调用subscription.request以接收所有正文项目并收集它们。

我们的 custrom 订户：

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.Flow;

public class FlowSubscriber<T> implements Flow.Subscriber<T> {
    private final CountDownLatch latch = new CountDownLatch(1);
    private List<T> bodyItems = new ArrayList<>();

    public List<T> getBodyItems() {
        try {
            this.latch.await();
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        return bodyItems;
    }

    @Override
    public void onSubscribe(Flow.Subscription subscription) {
        //Retrieve all parts
        subscription.request(Long.MAX_VALUE);
    }

    @Override
    public void onNext(T item) {
        this.bodyItems.add(item);
    }

    @Override
    public void onError(Throwable throwable) {
        this.latch.countDown();
    }

    @Override
    public void onComplete() {
        this.latch.countDown();
    }
}

测试中的用法：

@Test
public void test() {
    byte[] expected = "example".getBytes();

    HttpRequest.BodyPublisher bodyPublisher =
            HttpRequest.BodyPublishers.ofByteArray(expected);

    FlowSubscriber<ByteBuffer> flowSubscriber = new FlowSubscriber<>();
    bodyPublisher.subscribe(flowSubscriber);

    byte[] actual = flowSubscriber.getBodyItems().get(0).array();

    Assert.assertArrayEquals(expected, actual);
}