9

我正在尝试测试使用新 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()，但没有真正的实现，所以比较总是错误的。
4

1 回答 1

5

如果您对处理程序中的主体外观感兴趣，您可以在 HttpRequest.BodyPublisher 订阅者的帮助下了解它。我们打电话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);
}
于 2019-12-15T19:26:23.040 回答