Skip to content

Commit 0ea4048

Browse files
committed
Check queue content in test
1 parent 2e0a6ba commit 0ea4048

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

src/test/java/com/rabbitmq/client/amqp/impl/ClusterTest.java

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@
2727
import com.rabbitmq.client.amqp.*;
2828
import com.rabbitmq.client.amqp.impl.TestUtils.Sync;
2929
import java.util.List;
30+
import java.util.Set;
31+
import java.util.concurrent.ConcurrentHashMap;
3032
import java.util.concurrent.atomic.AtomicInteger;
3133
import java.util.function.Consumer;
3234
import org.junit.jupiter.api.*;
@@ -131,7 +133,22 @@ void publishToMovingQq() {
131133
publisher.publish(publisher.message().messageId(3L), ctx -> publishSync.down());
132134
assertThat(publishSync).completes();
133135

134-
assertThat(queueInfo()).hasMessageCount(3);
136+
int messageCount = 3;
137+
assertThat(queueInfo()).hasMessageCount(messageCount);
138+
Sync consumeSync = sync(messageCount);
139+
Set<Long> messageIds = ConcurrentHashMap.newKeySet(3);
140+
connection
141+
.consumerBuilder()
142+
.queue(q)
143+
.messageHandler(
144+
(ctx, msg) -> {
145+
messageIds.add(msg.messageIdAsLong());
146+
consumeSync.down();
147+
ctx.accept();
148+
})
149+
.build();
150+
assertThat(consumeSync).completes();
151+
assertThat(messageIds).containsOnly(1L, 2L, 3L);
135152
} finally {
136153
management.queueDeletion().delete(q);
137154
}

0 commit comments

Comments
 (0)