Making a mocked method return an argument that was passed to it

by

Since Mockito 1.9.5+ and Java 8+

You can use a lambda expression, like:

when(myMock.myFunction(anyString())).thenAnswer(i -> i.getArguments()[0]);

Where i is an instance of InvocationOnMock.

For older versions

You can create an Answer in Mockito. Let’s assume, we have an interface named MyInterface with a method myFunction.

public interface MyInterface {
    public String myFunction(String abc);
}

Here is the test method with a Mockito answer:

public void testMyFunction() throws Exception {
    MyInterface mock = mock(MyInterface.class);
    when(mock.myFunction(anyString())).thenAnswer(new Answer<String>() {
    @Override
    public String answer(InvocationOnMock invocation) throws Throwable {
        Object[] args = invocation.getArguments();
        return (String) args[0];
    }
    });

    assertEquals("someString",mock.myFunction("someString"));
    assertEquals("anotherString",mock.myFunction("anotherString"));
}

More Related Contents:

  1. Mocking static methods with Mockito
  2. How to mock a final class with mockito
  3. How to mock private method for testing using PowerMock?
  4. @RunWith(MockitoJUnitRunner.class) vs MockitoAnnotations.initMocks(this)
  5. Difference between @Mock, @MockBean and Mockito.mock()
  6. How to properly match varargs in Mockito
  7. Final method mocking
  8. Mockito: Trying to spy on method is calling the original method
  9. Mockito: InvalidUseOfMatchersException
  10. Is it possible to use Mockito in Kotlin?
  11. How to use ArgumentCaptor for stubbing?
  12. Is it discouraged to use @Spy and @InjectMocks on the same field?
  13. Unfinished Stubbing Detected in Mockito
  14. Mockito, JUnit and Spring
  15. What is the difference between mocking and spying when using Mockito?
  16. How to mock an enum singleton class using Mockito/Powermock?
  17. Mockito – NullpointerException when stubbing Method
  18. Mockito test a void method throws an exception
  19. Mockito.any() pass Interface with Generics
  20. Using Mockito to mock classes with generic parameters
  21. Mocking Logger and LoggerFactory with PowerMock and Mockito
  22. PowerMock, mock a static method, THEN call real methods on all other statics
  23. Mockito: Inject real objects into private @Autowired fields
  24. How do I mock a REST template exchange?
  25. Why are my mocked methods not calld when executing a unit test?
  26. How to mock RestTemplate in Java Spring?
  27. Java verify void method calls n times with Mockito
  28. Mockito’s Matcher vs Hamcrest Matcher?
  29. Powermock – java.lang.IllegalStateException: Failed to transform class
  30. Integration tests for AspectJ

Leave a Comment