From 465c5ea3f685e999dd3c32043692c0076df1ed0d Mon Sep 17 00:00:00 2001 From: Chris Bono Date: Mon, 15 Aug 2022 17:46:03 -0500 Subject: [PATCH] Polish "Pulsar template schema detection improvements" * Adjust AssertJ completable future usage in newly added test. --- .../org/springframework/pulsar/core/PulsarTemplateTests.java | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/spring-pulsar/src/test/java/org/springframework/pulsar/core/PulsarTemplateTests.java b/spring-pulsar/src/test/java/org/springframework/pulsar/core/PulsarTemplateTests.java index 65afb903..fb5073a1 100644 --- a/spring-pulsar/src/test/java/org/springframework/pulsar/core/PulsarTemplateTests.java +++ b/spring-pulsar/src/test/java/org/springframework/pulsar/core/PulsarTemplateTests.java @@ -132,9 +132,8 @@ class PulsarTemplateTests extends AbstractContainerBaseTests { pulsarTemplate.setSchema(Schema.JSON(Foo.class)); Foo foo = new Foo("Foo-" + UUID.randomUUID(), "Bar-" + UUID.randomUUID()); pulsarTemplate.send(foo); - - CompletableFuture> receiveMsgFuture = consumer.receiveAsync(); - assertThat(receiveMsgFuture).isCompleted().isCompletedWithValueMatching(m -> m.getValue().equals(foo)); + assertThat(consumer.receiveAsync()).succeedsWithin(Duration.ofSeconds(3)).extracting(Message::getValue) + .isEqualTo(foo); } } }