如何测试 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. 如何获取/测试其请求正文（= my "example"）？我可以打电话request.bodyPublisher()来获取HttpRequest.BodyPublisher，但后来我被卡住了。

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

Answer 1

如果您对主体在处理程序中的外观感兴趣，您可以在 HttpRequest.BodyPublisher Subscriber 的帮助下了解它。我们致电subscription.request以便接收并收集所有遗体物品。

我们的自定义订阅者：

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);
}