Increase wait timeout in the AbsCorMHandlerTests

https://build.spring.io/browse/INT-MASTERSPRING40-414/

Author: artembilan

spring-integration-core/src/test/java/org/springframework/integration/aggregator/AbstractCorrelatingMessageHandlerTests.java

@@ -96,7 +96,7 @@ protected void afterRelease(MessageGroup group, Collection<Message<?>> completed
 			waitReapCompleteLatch.countDown();
 		});
 
-		final List<Message<?>> outputMessages = new ArrayList<Message<?>>();
+		final List<Message<?>> outputMessages = new ArrayList<>();
 		handler.setOutputChannel((message, timeout) -> {
 			/*
 			 * Executes when group 'bar' completes normally
@@ -137,7 +137,7 @@ protected void afterRelease(MessageGroup group, Collection<Message<?>> completed
 				.build();
 		handler.handleMessage(message);
 
-		assertTrue(waitReapCompleteLatch.await(10, TimeUnit.SECONDS));
+		assertTrue(waitReapCompleteLatch.await(20, TimeUnit.SECONDS));
 		// Before INT-2751 we got bar + bar + qux
 		assertEquals(2, outputMessages.size()); // bar + qux
 		// normal release
@@ -154,7 +154,7 @@ public void testReaperReapsAnEmptyGroup() throws Exception {
 		final MessageGroupStore groupStore = new SimpleMessageStore();
 		AggregatingMessageHandler handler = new AggregatingMessageHandler(group -> group, groupStore);
 
-		final List<Message<?>> outputMessages = new ArrayList<Message<?>>();
+		final List<Message<?>> outputMessages = new ArrayList<>();
 		handler.setOutputChannel((message, timeout) -> {
 			/*
 			 * Executes when group 'bar' completes normally
@@ -181,7 +181,7 @@ public void testReaperReapsAnEmptyGroupAfterConfiguredDelay() throws Exception {
 		final MessageGroupStore groupStore = new SimpleMessageStore();
 		AggregatingMessageHandler handler = new AggregatingMessageHandler(group -> group, groupStore);
 
-		final List<Message<?>> outputMessages = new ArrayList<Message<?>>();
+		final List<Message<?>> outputMessages = new ArrayList<>();
 		handler.setOutputChannel((message, timeout) -> {
 			/*
 			 * Executes when group 'bar' completes normally
@@ -233,8 +233,8 @@ public void testReapWithChangeInSameMillisecond() throws Exception {
 		Method forceComplete =
 				AbstractCorrelatingMessageHandler.class.getDeclaredMethod("forceComplete", MessageGroup.class);
 		forceComplete.setAccessible(true);
-		GenericMessage<String> secondMessage = new GenericMessage<String>("bar");
-		mgs.addMessagesToGroup("foo", new GenericMessage<String>("foo"), secondMessage);
+		GenericMessage<String> secondMessage = new GenericMessage<>("bar");
+		mgs.addMessagesToGroup("foo", new GenericMessage<>("foo"), secondMessage);
 		MessageGroup group = mgs.getMessageGroup("foo");
 		// remove a message
 		mgs.removeMessagesFromGroup("foo", secondMessage);
@@ -256,7 +256,7 @@ public void testDontReapIfAlreadyComplete() throws Exception {
 		QueueChannel outputChannel = new QueueChannel();
 		handler.setOutputChannel(outputChannel);
 		MessageGroupStore mgs = TestUtils.getPropertyValue(handler, "messageStore", MessageGroupStore.class);
-		mgs.addMessagesToGroup("foo", new GenericMessage<String>("foo"));
+		mgs.addMessagesToGroup("foo", new GenericMessage<>("foo"));
 		mgs.completeGroup("foo");
 		mgs = spy(mgs);
 		new DirectFieldAccessor(handler).setPropertyValue("messageStore", mgs);
@@ -282,7 +282,7 @@ public void testDontReapIfAlreadyCompleteAfterRefetch() throws Exception {
 		QueueChannel outputChannel = new QueueChannel();
 		handler.setOutputChannel(outputChannel);
 		MessageGroupStore mgs = TestUtils.getPropertyValue(handler, "messageStore", MessageGroupStore.class);
-		mgs.addMessagesToGroup("foo", new GenericMessage<String>("foo"));
+		mgs.addMessagesToGroup("foo", new GenericMessage<>("foo"));
 		MessageGroup group = new SimpleMessageGroup(mgs.getMessageGroup("foo"));
 		mgs.completeGroup("foo");
 		mgs = spy(mgs);
@@ -311,7 +311,7 @@ public void testDontReapIfNewGroupFoundDuringRefetch() throws Exception {
 		QueueChannel outputChannel = new QueueChannel();
 		handler.setOutputChannel(outputChannel);
 		MessageGroupStore mgs = TestUtils.getPropertyValue(handler, "messageStore", MessageGroupStore.class);
-		mgs.addMessagesToGroup("foo", new GenericMessage<String>("foo"));
+		mgs.addMessagesToGroup("foo", new GenericMessage<>("foo"));
 		MessageGroup group = new SimpleMessageGroup(mgs.getMessageGroup("foo"));
 		mgs = spy(mgs);
 		new DirectFieldAccessor(handler).setPropertyValue("messageStore", mgs);
@@ -383,7 +383,7 @@ public void testScheduleRemoveAnEmptyGroupAfterConfiguredDelay() throws Exceptio
 		final MessageGroupStore groupStore = new SimpleMessageStore();
 		AggregatingMessageHandler handler = new AggregatingMessageHandler(group -> group, groupStore);
 
-		final List<Message<?>> outputMessages = new ArrayList<Message<?>>();
+		final List<Message<?>> outputMessages = new ArrayList<>();
 		handler.setOutputChannel((message, timeout) -> {
 			/*
 			 * Executes when group 'bar' completes normally
@@ -422,7 +422,7 @@ public void testScheduleRemoveAnEmptyGroupAfterConfiguredDelay() throws Exceptio
 	}
 
 	@Test
-	public void testDontReapMessageOfOtherHandler() throws Exception {
+	public void testDontReapMessageOfOtherHandler() {
 		MessageGroupStore groupStore = new SimpleMessageStore();
 
 		AggregatingMessageHandler handler1 = new AggregatingMessageHandler(group -> group, groupStore);
@@ -443,7 +443,8 @@ public void testDontReapMessageOfOtherHandler() throws Exception {
 
 		groupStore.expireMessageGroups(0);
 
-		assertTrue(handler1DiscardChannel.getQueueSize() == 2);
-		assertTrue(handler2DiscardChannel.getQueueSize() == 1);
+		assertEquals(2, handler1DiscardChannel.getQueueSize());
+		assertEquals(1, handler2DiscardChannel.getQueueSize());
 	}
+
 }