Testing Spring-Integration with subscriber channel - testing

Question

I have the following:
Direct-Channel => Splitter => PublishSubscribeChannel
I would like to send data to the direct-channel and test the outcome in the publish-subscribe-channel
What I have so far partly taken from Spring.io (and it works only if I step through debug mode) is:
DirectChannel incomeChannel
PublishSubscribeChannel subscribeChannel
#Test
public void test() {
final AtomicInteger count = new AtomicInteger()
assert true == subscribeChannel.subscribe(new MessageHandler() {
void handleMessage(Message<?> message) throws MessagingException {
count.getAndIncrement();
Entity response = message.getPayload()
assert response != null
// assertions ...
}
})
def request = MessageBuilder.withPayload(entities).build()
assert incomeChannel.send(request) == true
Thread.sleep(10000)
assert 0 < count.get()
}

Answer 1

You don't show your configuration but, if your pub sub channel has a task executor, you need to add a latch; you also should do the asserts on the main thread...
#Test
public void test() {
final AtomicReference<Message<?>> messageRef = new AtomicReference<>();
final CountDownLatch latch = new CountDownLatch(1);
assert true == subscriberChannel.subscribe(new MessageHandler() {
void handleMessage(Message<?> message) throws MessagingException {
messageRef.set(message);
latch.countDown();
}
})
def request = MessageBuilder.withPayload(entities).build()
assert incomeChannel.send(request) == true
assert true == latch.await(10, TimeUnit.SECONDS)
Entity response = msg.get().getPayload()
assert response != null
// assertions ...
}