Spring/spring-integration-samples
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix race condition in MQTT ApplicationTest

Browse Source 
We mock a `MessageHandler` on the receiver side, but still send the
message over MQTT.

Add `CountDownLatch` to the mocked method and wait for its count before
asserting a received message
This commit is contained in:
Artem Bilan
2019-04-08 11:06:31 -04:00
parent 8eb85d29b3
commit 47ebbee8f3
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
basic/mqtt/src/test/java/org/springframework/integration/samples/mqtt/ApplicationTest.java
View File

@@ -22,6 +22,9 @@ import static org.mockito.Mockito.verify;
import static org.springframework.integration.test.mock.MockIntegration.messageArgumentCaptor;
import static org.springframework.integration.test.mock.MockIntegration.mockMessageHandler;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import org.junit.ClassRule;
import org.junit.Test;
import org.junit.runner.RunWith;
@@ -58,14 +61,16 @@ public class ApplicationTest {
private IntegrationFlow mqttOutFlow;
@Test
public void test() {
public void test() throws InterruptedException {
ArgumentCaptor<Message<?>> captor = messageArgumentCaptor();
MessageHandler mockMessageHandler = mockMessageHandler(captor).handleNext(m -> { });
CountDownLatch receiveLatch = new CountDownLatch(1);
MessageHandler mockMessageHandler = mockMessageHandler(captor).handleNext(m -> receiveLatch.countDown());
this.mockIntegrationContext
.substituteMessageHandlerFor(
"mqttInFlow.org.springframework.integration.config.ConsumerEndpointFactoryBean#1",
mockMessageHandler);
this.mqttOutFlow.getInputChannel().send(new GenericMessage<>("foo"));
assertThat(receiveLatch.await(10, TimeUnit.SECONDS)).isTrue();
verify(mockMessageHandler).handleMessage(any());
assertThat(captor.getValue().getPayload())
.isEqualTo("foo sent to MQTT, received from MQTT");