Ensure RemotePartitioningWorkerStepBuilder override all methods of StepBuilder

Closes GH-5150

Signed-off-by: Yanming Zhou <[email protected]>

Author: quaff

spring-batch-integration/src/main/java/org/springframework/batch/integration/partition/RemotePartitioningWorkerStepBuilder.java

@@ -26,6 +26,7 @@
 import org.springframework.batch.core.partition.Partitioner;
 import org.springframework.batch.core.repository.JobRepository;
 import org.springframework.batch.core.step.StepLocator;
+import org.springframework.batch.core.step.builder.ChunkOrientedStepBuilder;
 import org.springframework.batch.core.step.builder.FlowStepBuilder;
 import org.springframework.batch.core.step.builder.JobStepBuilder;
 import org.springframework.batch.core.step.builder.PartitionStepBuilder;
@@ -62,7 +63,9 @@
  *
  * @since 4.1
  * @author Mahmoud Ben Hassine
+ * @author Yanming Zhou
  */
+@SuppressWarnings("removal")
 public class RemotePartitioningWorkerStepBuilder extends StepBuilder {
 
 	private static final String SERVICE_ACTIVATOR_METHOD_NAME = "handle";
@@ -162,6 +165,18 @@ public TaskletStepBuilder tasklet(Tasklet tasklet, PlatformTransactionManager tr
 		return super.tasklet(tasklet, transactionManager);
 	}
 
+    @Override
+    public TaskletStepBuilder tasklet(Tasklet tasklet) {
+        configureWorkerIntegrationFlow();
+        return super.tasklet(tasklet);
+    }
+
+    @Override
+    public <I, O> ChunkOrientedStepBuilder<I, O> chunk(int chunkSize) {
+        configureWorkerIntegrationFlow();
+		return super.chunk(chunkSize);
+	}
+
 	@Override
 	public <I, O> SimpleStepBuilder<I, O> chunk(int chunkSize, PlatformTransactionManager transactionManager) {
 		configureWorkerIntegrationFlow();

spring-batch-integration/src/test/java/org/springframework/batch/integration/partition/RemotePartitioningWorkerStepBuilderTests.java

@@ -16,18 +16,24 @@
 
 package org.springframework.batch.integration.partition;
 
+import java.lang.reflect.Method;
+import java.lang.reflect.Modifier;
+
 import org.junit.jupiter.api.Test;
 import org.mockito.Mock;
 
 import org.springframework.batch.core.repository.JobRepository;
+import org.springframework.batch.core.step.builder.StepBuilder;
 import org.springframework.batch.core.step.tasklet.Tasklet;
 import org.springframework.transaction.PlatformTransactionManager;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.fail;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 
 /**
  * @author Mahmoud Ben Hassine
+ * @author Yanming Zhou
  */
 class RemotePartitioningWorkerStepBuilderTests {
 
@@ -110,4 +116,17 @@ void testMandatoryInputChannel() {
 		assertThat(expectedException).hasMessage("An InputChannel must be provided");
 	}
 
+    @Test
+    void testAllMethodsAreOverridden() throws Exception {
+        for (Method method : StepBuilder.class.getDeclaredMethods()) {
+            if (!Modifier.isPublic(method.getModifiers())) {
+                continue;
+            }
+            try {
+                RemotePartitioningWorkerStepBuilder.class.getDeclaredMethod(method.getName(), method.getParameterTypes());
+            } catch (NoSuchMethodException ex) {
+                fail(RemotePartitioningWorkerStepBuilder.class.getName() + " should override method [" + method +"] to configure worker integration flow.");
+            }
+        }
+    }
 }