Clash Royale CLAN TAG

Using @MockBean to mock an ItemReader with @BeforeStep Throws Exception to many methods with @Beforestep anotation. How do I solve it?

I have my own Implementation of an ItemReader with two methods.

public class Reader implements ItemReader<Integer> {

private final Logger logger = LoggerFactory.getLogger(getClass());
private Iterator<Integer> iterator;

@Override
public Integer read() throws UnexpectedInputException, ParseException, NonTransientResourceException {
if(iterator.hasNext()){
return iterator.next();
}
return null;
}

@BeforeStep
public void init(StepExecution stepExecution){
List<Integer> integerList = (List<Integer>)stepExecution.getJobExecution().getExecutionContext().get(CKEY_ERROREVENT_IDS);
this.iterator = integerList.iterator();
}
}


When I try to run this within a spring-batch context and mocking the ItemReader with @MockBean the application context throws:

@MockBean

java.lang.IllegalArgumentException: found more than one method on target class [Reader$MockitoMock$368106910] with the annotation type [BeforeStep].

Below is how I start the job.

@MockBean
private Reader reader;

@Test
public void readerTest(){
JobParameters jobParameters = new JobParametersBuilder()
.addString("triggerId", UUID.randomUUID().toString()).toJobParameters();
JobExecution jobExecution = jobLauncher.run(processEventJob, jobParameters);
}


2 Answers
2

I've been able to reproduce your issue. After that, I've replaced the mock with a child class of your Reader.

@Component
public class ChildReader extends Reader{

public void init(StepExecution stepExecution){
super.init(stepExecution);
}
}


This gives the same exception.
At runtime, Mockito also creates as subclass of your Reader. I think that this is causing your problem.

In this post, the same issue as described:
http://forum.spring.io/forum/spring-projects/batch/98067-beforestep-in-abstract-class
I cannot find any reference to it being fixed.

What you can do to resolve this issue is extract the code you wish to mock to a separate class and mock that class. One logical place for it would be a StepExecutionListener.
https://docs.spring.io/spring-batch/trunk/apidocs/org/springframework/batch/core/StepExecutionListener.html

Hope this helps....

implement stepexecutionlistener interface in your class and override before and after step method this solves my problem dont use @beforestep

public class Reader implements ItemReader<Integer>,StepExecutionListener{

private final Logger logger = LoggerFactory.getLogger(getClass());
private Iterator<Integer> iterator;

@Override
public Integer read() throws UnexpectedInputException, ParseException, NonTransientResourceException {
if(iterator.hasNext()){
return iterator.next();
}
return null;
}

@Override
public void beforeStep(StepExecution stepExecution){
List<Integer> integerList = (List<Integer>)stepExecution.getJobExecution().getExecutionContext().get(CKEY_ERROREVENT_IDS);
this.iterator = integerList.iterator();
}

@Override
public ExitStatus afterStep(StepExecution stepExecution) {
return null;
}
}


By clicking "Post Your Answer", you acknowledge that you have read our updated terms of service, privacy policy and cookie policy, and that your continued use of the website is subject to these policies.