From 2d17d60e472fdf68af409466d24b5df4d0c975e0 Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Wed, 15 Jun 2022 11:50:43 +0200
Subject: [PATCH] Fix batching test

---
 .../cloud/stream/function/FunctionBatchingTests.java  | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java b/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java
index d0284fb06..864e0d406 100644
--- a/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java
+++ b/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java
@@ -47,7 +47,7 @@ public class FunctionBatchingTests {
 	public void testMessageBatchConfiguration() {
 		TestChannelBinderConfiguration.applicationContextRunner(MessageBatchConfiguration.class)
 				.withPropertyValues("spring.jmx.enabled=false",
-						"spring.cloud.stream.function.definition=func",
+						"spring.cloud.stream.function.definition=func2",
 						"spring.cloud.stream.bindings.input.consumer.batch-mode=true")
 				.run(context -> {
 					InputDestination inputDestination = context.getBean(InputDestination.class);
@@ -282,6 +282,15 @@ public class FunctionBatchingTests {
 
 		@Bean
 		public Function<Message<List<Person>>, Person> func() {
+			return x -> {
+				Object o = x.getPayload().get(2);
+				assertThat(o).isNull();
+				return (Person) x.getPayload().get(0);
+			};
+		}
+
+		@Bean
+		public Function<Message<List<Person>>, Person> func2() {
 			return x -> x.getPayload().get(0);
 		}