How do I change the default return value for Strings in Mockito?

Question

This issue from 2010 hints at what I'm trying to do. I'm working on a unit test which exercises code that requires many mocked objects to do what it needs to do (testing HTML + PDF rendering). For this test to be successful I need many mocked objects to be generated and each of these objects ultimately return some String data to the code being tested. I think I can do this by implementing either my own `Answer` class or `IMockitoConfiguration`, but I am not sure how to implement those so they only affect methods which return Strings. I feel that the following code is close to what I want. It throws a cast exception, `java.lang.ClassCastException: java.lang.String cannot be cast to com.mypackage.ISOCountry`. I think this means I need to somehow default or limit the `Answer` to only affect the defaults for `String`. ``` private Address createAddress(){ Address address = mock(Address.class, new StringAnswer() ); /* I want to replace repetitive calls like this, with a default string. I ju…

Accepted Answer

I was able to completely answer my own question.

In this example an address with the Locality of Louisville is generated while the other fields look like "address.getStreet1();".

```
private Address createAddress(){
    Address address = mock(Address.class, new StringAnswer() );

when(address.getLocality()).thenReturn("Louisville");

ISOCountry isoCountry = mock(ISOCountry.class);
    when(isoCountry.getIsocode()).thenReturn("US");
    when(address.getCountry()).thenReturn(isoCountry);

return address;
}

private class StringAnswer implements Answer<Object> {
    @Override
    public Object answer(InvocationOnMock invocation) throws Throwable {
           if( invocation.getMethod().getReturnType().equals(String.class)){
               return invocation.toString();
           }
           else{
               return Mockito.RETURNS_DEFAULTS.answer(invocation);
           }
       }
}
```

How do I change the default return value for Strings in Mockito?

Solution

Problem

Related problems