How to mock a builder with mockito

by

The problem with using RETURN_DEEP_STUBS is that you’ll get a different mock each time you call a method. I think from your question that you want to use a default Answer that actually returns the mock on which it was called, for each method that has the right return type. This could look something like the following. Note that I haven’t tested this, so it may contain typos, but I hope that the intention is clear in any case.

import static org.mockito.Mockito.RETURNS_DEFAULTS;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

public class SelfReturningAnswer implements Answer<Object> {

    public Object answer(InvocationOnMock invocation) throws Throwable {
        Object mock = invocation.getMock();
        if(invocation.getMethod().getReturnType().isInstance(mock)){
            return mock;
        }
        return RETURNS_DEFAULTS.answer(invocation);
    }
}

Then, when you create your mock, specify this as your default answer. This will make your mock return itself from each method that it can; but it will behave like an ordinary mock when you call a method whose return type is wrong for the mock.

Create your mock like this

Builder mockBuilder = mock( Builder.class, new SelfReturningAnswer());

or create a constant for this class and write something like

@Mock(answer = SELF_RETURNING) private Builder mockBuilder;

Hope that helps.

More Related Contents:

  1. Mocking static methods with Mockito
  2. How to mock void methods with Mockito
  3. Difference between @Mock and @InjectMocks
  4. Mockito – difference between doReturn() and when()
  5. Using Mockito to test abstract classes
  6. Test class with a new() call in it with Mockito
  7. Final method mocking
  8. How to use Mockito with JUnit5
  9. Mockito : how to verify method was called on an object created within a method?
  10. How to tell a Mockito mock object to return something different the next time it is called?
  11. Mockito: InvalidUseOfMatchersException
  12. How to use ArgumentCaptor for stubbing?
  13. Unit testing with mockito for constructors
  14. How to mock new Date() in java using Mockito
  15. Mockito, JUnit and Spring
  16. Mockito – @Spy vs @Mock
  17. Mockito verify order / sequence of method calls
  18. mock or stub for chained call
  19. How to verify that a specific method was not called using Mockito?
  20. Mock final class with Mockito 2
  21. Why are my mocked methods not calld when executing a unit test?
  22. Forming Mockito “grammars”
  23. What’s the difference between Mockito Matchers isA, any, eq, and same?
  24. How to mock RestTemplate in Java Spring?
  25. Java verify void method calls n times with Mockito
  26. Can Mockito verify parameters based on their values at the time of method call?
  27. Why are my mocked methods not called when executing a unit test?
  28. Assert equals between 2 Lists in Junit
  29. AssertEquals 2 Lists ignore order
  30. Mocking Java enum to add a value to test fail case

Leave a Comment