CODEWITHSUNDEEP
unit-testingmockitomocking
fnCzar
fnCzar

Reputation: 3213

How to only throw exception when the mocked method is called for the first time?

I have a method of a mocked object that can be called multiple times (think recursion). The method is defined like this:

public void doCommit() { }

In order to tell it to fail I use this convention:

doThrow(new RuntimeException()).when(mMockedObject).doCommit();

This though, makes the method throw this exception EVERY time it is called. How can I make it so that it only, for example, throws it the first and third time it is called? Which means that, for example, the second and forth time it just returns without throwing an exception. Please note that I am not the author of doCommit(), nor do I have source code that I can change.

Upvotes: 49

Views: 37632

Answers (2)

fnCzar
fnCzar

Reputation: 3213

I figured it out (with some hints from Igor). This is how you stub consecutive void method calls:

doThrow(new RuntimeException()).doNothing().doThrow(...).doNothing().when(mMockedObject).doCommit();

thanks Igor!

Upvotes: 73

Igor Zevaka
Igor Zevaka

Reputation: 76500

Reading Stubbing Consecutive Calls doco, something like this might do it:

when(mMockedObject.doCommit())
  .thenThrow(new RuntimeException())
  .thenCallRealMethod() 
  .thenThrow(new RuntimeException())
  .thenCallRealMethod();

If you don't want to actually call the underlying method then you should use thenAnswer instead of thenCallRealMethod method and provide an empty stub imlementation.

Upvotes: 24

Related Questions