Hello Friends,
I am facing a particular problem where I have to test a method that reads a list of objects from a cache and then sorts them based upon priority. The list of objects as well as the cache has to be mock objects. However, once the method returns I have to test that these mock objects that are returned back by this method is sorted correctly based upon expected priority. Here is a very very simplified code sample which mimics what I am trying to do. In this code sample I have 3 Priority Beans which are mocked and I am trying to sort and return it back in ArrayComparator.sortBeans() method. However, the problems are obvious. The call to pb.getPriority() is not instance based but invocation count based and hence when the comparator calls getPriority on the bean for more than 3 times I get an UnexpectedInvocation error. Even I get through that in the assertEquals() method call pbArray[x].getPriority() method still calls the mock object. What I need are real stubs which are unique objects which can later on be compared. How do I do that in JMockit?
package com.salil.jmockit.mockarray;
import java.util.Arrays;
import java.util.Comparator;
public class ArrayComparator {
public PriorityBean[] sortBeans(PriorityBean[] beans){
Arrays.sort(beans,new Comparator<PriorityBean>() {
public int compare(PriorityBean o1, PriorityBean o2) {
return o2.getPriority() - o1.getPriority();
}
});
return beans;
}
}
package com.salil.jmockit.mockarray;
public class PriorityBean {
private int priority;
PriorityBean(int priority){
this.priority = priority;
}
public int getPriority(){
return priority;
}
}
package com.salil.jmockit.mockarray;
import static org.junit.Assert.assertEquals;
import mockit.Expectations;
import org.junit.Test;
public class MockArrayComparatorTest {
@Test
public void testSortBeans(final PriorityBean pb) {
new Expectations(){
{
pb.getPriority();
returns(5,10,0);
}
};
ArrayComparator ac = new ArrayComparator();
PriorityBean[] pbArray = ac.sortBeans(new PriorityBean[]{pb,pb,pb});
assertEquals(3,pbArray.length);
assertEquals(10,pbArray[0].getPriority());
assertEquals(5,pbArray[1].getPriority());
assertEquals(0,pbArray[2].getPriority());
}
}