How do I return different values on different calls to a mock?

java mocking mockito powermockito
user2032118 picture user2032118 · Sep 7, 2015 · Viewed 13.4k times · Source

I have the following code which is getting the current counter value from DB. Then it updates the counter in DB and then again it retrieves the value.

int current = DBUtil.getCurrentCount();
DBUtil.updateCount(50);// it updates the current count by adding 50
int latest = DBUtil.getCurrentCount();

I want to mock the static methods in such a way that the first call should return 100 and the second call should return 150. How can I use PowerMockito to achieve this? I am using TestNG, Mockito along with PowerMock.

Answer

durron597 picture durron597 · Sep 7, 2015

Mockito supports changing the returned value; this support extends to PowerMockito. Just use OngoingStubbing.thenReturn(T value, T... values)

OngoingStubbing<T> thenReturn(T value, T... values)

Sets consecutive return values to be returned when the method is called.
E.g: 

when(mock.someMethod()).thenReturn(1, 2, 3);

Last return value in the sequence (in example: 3) determines the behavior of further consecutive calls.

So, in this case, you would do:

PowerMockito.when(DBUtil.getCurrentCount()).thenReturn(100, 150);

Note: this answer assumes you already know how to mock static methods. If you do not, see this question.

Related questions

@RunWith(PowerMockRunner.class) vs @RunWith(MockitoJUnitRunner.class)

In the usual mocking with @Mock and @InjectMocks annotations, the class under testing should be run with @RunWith(MockitoJUnitRunner.class). @RunWith(MockitoJUnitRunner.class) public class ReportServiceImplTestMockito { @Mock private TaskService mockTaskService; @InjectMocks private ReportServiceImpl service; // Some tests } but in some example …

Read More
Unit test for Runnable with Mockito

I have a code like this for which I would like to write unit test. public class TestClass { private final Executor executor; private final Handler handler; TestClass(Executor executor, Handler handler) { this.executor = executor; this.handler = handler; } void doSomething(String …

Read More
How to mock void methods with Mockito

How to mock methods with void return type? I implemented an observer pattern but I can't mock it with Mockito because I don't know how. And I tried to find an example on the Internet but didn't succeed. My class …

Read More