diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemReadListener.java b/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemReadListener.java index 3d680d640..5c80b3cbf 100644 --- a/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemReadListener.java +++ b/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemReadListener.java @@ -20,7 +20,6 @@ import java.util.Arrays; import java.util.Iterator; import java.util.List; -import org.springframework.batch.core.ChunkListener; import org.springframework.batch.core.ItemReadListener; /** @@ -34,10 +33,10 @@ public class CompositeItemReadListener implements ItemReadListener { /** * Public setter for the listeners. * - * @param listeners + * @param itemReadListeners */ - public void setListeners(ChunkListener[] listeners) { - this.listeners = Arrays.asList(listeners); + public void setListeners(ItemReadListener[] itemReadListeners) { + this.listeners = Arrays.asList(itemReadListeners); } /** diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemWriteListener.java b/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemWriteListener.java index 07a30111a..554fb048e 100644 --- a/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemWriteListener.java +++ b/spring-batch-core/src/main/java/org/springframework/batch/core/listener/CompositeItemWriteListener.java @@ -20,7 +20,6 @@ import java.util.Arrays; import java.util.Iterator; import java.util.List; -import org.springframework.batch.core.ChunkListener; import org.springframework.batch.core.ItemWriteListener; /** @@ -34,10 +33,10 @@ public class CompositeItemWriteListener implements ItemWriteListener { /** * Public setter for the listeners. * - * @param listeners + * @param itemWriteListeners */ - public void setListeners(ChunkListener[] listeners) { - this.listeners = Arrays.asList(listeners); + public void setListeners(ItemWriteListener[] itemWriteListeners) { + this.listeners = Arrays.asList(itemWriteListeners); } /** diff --git a/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemReadListenerTests.java b/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemReadListenerTests.java index e5a4d4d6d..152ad4fb0 100644 --- a/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemReadListenerTests.java +++ b/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemReadListenerTests.java @@ -64,4 +64,13 @@ public class CompositeItemReadListenerTests extends TestCase { compositeListener.onReadError(ex); listenerControl.verify(); } + + public void testSetListners() throws Exception { + compositeListener.setListeners(new ItemReadListener[] {listener}); + listener.beforeRead(); + listenerControl.replay(); + compositeListener.beforeRead(); + listenerControl.verify(); + } + } diff --git a/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemWriteListenerTests.java b/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemWriteListenerTests.java index 0df8faf40..e24f8bac2 100644 --- a/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemWriteListenerTests.java +++ b/spring-batch-core/src/test/java/org/springframework/batch/core/listener/CompositeItemWriteListenerTests.java @@ -19,7 +19,6 @@ import junit.framework.TestCase; import org.easymock.MockControl; import org.springframework.batch.core.ItemWriteListener; -import org.springframework.batch.core.listener.CompositeItemWriteListener; /** * @author Lucas Ward @@ -64,4 +63,14 @@ public class CompositeItemWriteListenerTests extends TestCase { compositeListener.onWriteError(ex, item); listenerControl.verify(); } + + public void testSetListners() throws Exception { + compositeListener.setListeners(new ItemWriteListener[] {listener}); + Object item = new Object(); + listener.beforeWrite(item); + listenerControl.replay(); + compositeListener.beforeWrite(item); + listenerControl.verify(); + } + }