Examples with TestUtils.b com.rabbitmq.stream.impl.TestUtils.b used on opensource projects

Example 1 with TestUtils.b

use of com.rabbitmq.stream.impl.TestUtils.b in project rabbitmq-stream-java-client by rabbitmq.

the class ClientTest method closingPublisherWhilePublishingShouldNotCloseConnection.

@ParameterizedTest
@ValueSource(strings = { "", "some-publisher-reference" })
void closingPublisherWhilePublishingShouldNotCloseConnection(String publisherReference) {
    AtomicReference<CountDownLatch> confirmLatch = new AtomicReference<>(new CountDownLatch(500_000));
    CountDownLatch closedLatch = new CountDownLatch(1);
    Semaphore semaphore = new Semaphore(10000);
    Client client = cf.get(new ClientParameters().shutdownListener(shutdownContext -> closedLatch.countDown()).publishConfirmListener((publisherId, publishingId) -> {
        semaphore.release();
        confirmLatch.get().countDown();
    }));
    Response response = client.declarePublisher(TestUtils.b(0), publisherReference, stream);
    assertThat(response.isOk()).isTrue();
    List<Thread> threads = IntStream.range(0, 10).mapToObj(i -> {
        Thread thread = new Thread(() -> {
            while (!Thread.currentThread().isInterrupted()) {
                List<Message> messages = IntStream.range(0, 100).mapToObj(j -> client.messageBuilder().addData("hello".getBytes(StandardCharsets.UTF_8)).build()).collect(Collectors.toList());
                semaphore.acquireUninterruptibly(100);
                client.publish(b(0), messages);
            }
        });
        thread.start();
        return thread;
    }).collect(Collectors.toList());
    try {
        assertThat(latchAssert(confirmLatch)).completes();
        response = client.deletePublisher(b(0));
        assertThat(response.isOk()).isTrue();
        assertThat(latchAssert(closedLatch)).doesNotComplete(1);
    } finally {
        threads.forEach(Thread::interrupt);
    }
}