How to use returnsLastArg method of org.mockito.AdditionalAnswers class

Best Mockito code snippet using org.mockito.AdditionalAnswers.returnsLastArg

Source:AdditionalAnswers.java

...13 *14 * <p>Currently offer answers that can return the parameter of an invocation at a certain position.15 *16 * <p>See factory methods for more information : {@link #returnsFirstArg}, {@link #returnsSecondArg},17 * {@link #returnsLastArg} and {@link #returnsArgAt}18 *19 * @since 1.9.520 */21@SuppressWarnings("unchecked")22public class AdditionalAnswers {23    private static final ReturnsArgumentAt RETURNS_FIRST_ARGUMENT = new ReturnsArgumentAt(0);24    private static final ReturnsArgumentAt RETURNS_SECOND_ARGUMENT = new ReturnsArgumentAt(1);25    private static final ReturnsArgumentAt RETURNS_LAST_ARGUMENT = new ReturnsArgumentAt(-1);26    /**27     * Returns the first parameter of an invocation.28     *29     * <p>30     *     This additional answer could be used at stub time using the31     *     <code>then|do|will{@link org.mockito.stubbing.Answer}</code> methods. For example :32     * </p>33     *34     * <pre class="code"><code class="java">given(carKeyFob.authenticate(carKey)).will(returnsFirstArg());35     * doAnswer(returnsFirstArg()).when(carKeyFob).authenticate(carKey)</code></pre>36     *37     * @param <T> Return type of the invocation.38     * @return Answer that will return the first argument of the invocation.39     *40     * @since 1.9.541     */42    public static <T> Answer<T> returnsFirstArg() {43        return (Answer<T>) RETURNS_FIRST_ARGUMENT;44    }45    /**46     * Returns the second parameter of an invocation.47     *48     * <p>49     *     This additional answer could be used at stub time using the50     *     <code>then|do|will{@link org.mockito.stubbing.Answer}</code> methods. For example :51     * </p>52     *53     * <pre class="code"><code class="java">given(trader.apply(leesFormula, onCreditDefaultSwap)).will(returnsSecondArg());54     * doAnswer(returnsSecondArg()).when(trader).apply(leesFormula, onCreditDefaultSwap)</code></pre>55     *56     * @param <T> Return type of the invocation.57     * @return Answer that will return the second argument of the invocation.58     *59     * @since 1.9.560     */61    public static <T> Answer<T> returnsSecondArg() {62        return (Answer<T>) RETURNS_SECOND_ARGUMENT;63    }64    /**65     * Returns the last parameter of an invocation.66     *67     * <p>68     *     This additional answer could be used at stub time using the69     *     <code>then|do|will{@link org.mockito.stubbing.Answer}</code> methods. For example :70     * </p>71     *72     * <pre class="code"><code class="java">given(person.remember(dream1, dream2, dream3, dream4)).will(returnsLastArg());73     * doAnswer(returnsLastArg()).when(person).remember(dream1, dream2, dream3, dream4)</code></pre>74     *75     * @param <T> Return type of the invocation.76     * @return Answer that will return the last argument of the invocation.77     *78     * @since 1.9.579     */80    public static <T> Answer<T> returnsLastArg() {81        return (Answer<T>) RETURNS_LAST_ARGUMENT;82    }83    /**84     * Returns the parameter of an invocation at the given position.85     *86     * <p>87     * This additional answer could be used at stub time using the88     * <code>then|do|will{@link org.mockito.stubbing.Answer}</code> methods. For example :89     * </p>90     *91     * <pre class="code"><code class="java">given(person.remember(dream1, dream2, dream3, dream4)).will(returnsArgAt(3));92     * doAnswer(returnsArgAt(3)).when(person).remember(dream1, dream2, dream3, dream4)</code></pre>93     *94     * @param <T> Return type of the invocation....

Source:WithAdditionalAnswers.java

...35    default<T> Answer<T> returnsSecondArg() {36        return AdditionalAnswers.returnsSecondArg();37    }38    /**39     * @see AdditionalAnswers#returnsLastArg()40     */41    default<T> Answer<T> returnsLastArg() {42        return AdditionalAnswers.returnsLastArg();43    }44    /**45     * @see AdditionalAnswers#returnsArgAt(int)46     */47    default<T> Answer<T> returnsArgAt(int position) {48        return AdditionalAnswers.returnsArgAt(position);49    }50    /**51     * @see AdditionalAnswers#delegatesTo(Object) 52     */53    default<T> Answer<T> delegatesTo(Object delegate) {54        return AdditionalAnswers.delegatesTo(delegate);55    }56    /**...

Source:BookServiceUnitTest.java

...34    public void givenCheckifEqualsMethodMocked_whenCheckifEqualsInvoked_ThenReturnLastArgument_UnitTest() {35        Book book1 = new Book(1L, "The Stranger", "Albert Camus", 456);36        Book book2 = new Book(2L, "Animal Farm", "George Orwell", 300);37        Book book3 = new Book(3L, "Romeo and Juliet", "William Shakespeare", 200);38        Mockito.when(bookRepository.selectRandomBook(any(Book.class), any(Book.class), any(Book.class))).then(AdditionalAnswers.returnsLastArg());39        Book lastBook = bookService.selectRandomBook(book1, book2, book3);40        assertEquals(lastBook, book3);41    }42    @Test43    public void givenCheckifEqualsMethodMocked_whenCheckifEqualsInvoked_ThenReturnArgumentAtIndex_UnitTest() {44        Book book1 = new Book(1L, "The Stranger", "Albert Camus", 456);45        Book book2 = new Book(2L, "Animal Farm", "George Orwell", 300);46        Book book3 = new Book(3L, "Romeo and Juliet", "William Shakespeare", 200);47        Mockito.when(bookRepository.selectRandomBook(any(Book.class), any(Book.class), any(Book.class))).then(AdditionalAnswers.returnsArgAt(1));48        Book bookOnIndex = bookService.selectRandomBook(book1, book2, book3);49        assertEquals(bookOnIndex, book2);50    }51}...

Source:ReturnsArgumentTest.java

...41        Person firstPerson = new Person("first");42        Person secondPerson = new Person("second");43        Person thirdPerson = new Person("third");44        Mockito.when(personRepository.select(firstPerson, secondPerson, thirdPerson))45          .then(AdditionalAnswers.returnsLastArg());46        Person actual = personService.select(firstPerson, secondPerson, thirdPerson);47        Assertions.assertThat(actual).isEqualTo(thirdPerson);48    }49    @Test50    public void shouldReturnArgAt() {51        Person firstPerson = new Person("first");52        Person secondPerson = new Person("second");53        Person thirdPerson = new Person("third");54        Mockito.when(personRepository.select(firstPerson, secondPerson, thirdPerson))55          .then(AdditionalAnswers.returnsArgAt(1));56        Person actual = personService.select(firstPerson, secondPerson, thirdPerson);57        Assertions.assertThat(actual).isEqualTo(secondPerson);58    }59}...

Source:StubbingWithAdditionalAnswers.java

...10import org.mockitousage.IMethods;11import static org.fest.assertions.Assertions.assertThat;12import static org.mockito.AdditionalAnswers.returnsArgAt;13import static org.mockito.AdditionalAnswers.returnsFirstArg;14import static org.mockito.AdditionalAnswers.returnsLastArg;15import static org.mockito.AdditionalAnswers.returnsSecondArg;16import static org.mockito.BDDMockito.given;17import static org.mockito.Matchers.anyInt;18import static org.mockito.Matchers.anyObject;19import static org.mockito.Matchers.anyString;20import static org.mockito.Matchers.anyVararg;21import static org.mockito.Matchers.eq;22@RunWith(MockitoJUnitRunner.class)23public class StubbingWithAdditionalAnswers {24    @Mock IMethods iMethods;25    @Test26    public void can_return_arguments_of_invocation() throws Exception {27        given(iMethods.objectArgMethod(anyObject())).will(returnsFirstArg());28        given(iMethods.threeArgumentMethod(eq(0), anyObject(), anyString())).will(returnsSecondArg());29        given(iMethods.threeArgumentMethod(eq(1), anyObject(), anyString())).will(returnsLastArg());30        assertThat(iMethods.objectArgMethod("first")).isEqualTo("first");31        assertThat(iMethods.threeArgumentMethod(0, "second", "whatever")).isEqualTo("second");32        assertThat(iMethods.threeArgumentMethod(1, "whatever", "last")).isEqualTo("last");33    }34    @Test35    public void can_return_expanded_arguments_of_invocation() throws Exception {36        given(iMethods.varargsObject(eq(1), anyVararg())).will(returnsArgAt(3));37        assertThat(iMethods.varargsObject(1, "bob", "alexander", "alice", "carl")).isEqualTo("alice");38    }39    @Test40    public void can_return_primitives_or_wrappers() throws Exception {41        given(iMethods.toIntPrimitive(anyInt())).will(returnsFirstArg());42        given(iMethods.toIntWrapper(anyInt())).will(returnsFirstArg());43        assertThat(iMethods.toIntPrimitive(1)).isEqualTo(1);...

returnsLastArg

Using AI Code Generation

1import static org.mockito.Mockito.mock;2import static org.mockito.Mockito.when;3import static org.mockito.AdditionalAnswers.returnsLastArg;4import java.util.List;5public class ReturnsLastArgExample {6   public static void main(String[] args) {7      List list = mock(List.class);8      when(list.add("one")).then(returnsLastArg());9      list.add("one");10      System.out.println(list.add("two"));11   }12}13import static org.mockito.Mockito.mock;14import static org.mockito.Mockito.when;15import static org.mockito.AdditionalAnswers.returnsFirstArg;16import java.util.List;17public class ReturnsFirstArgExample {18   public static void main(String[] args) {19      List list = mock(List.class);20      when(list.add("one")).then(returnsFirstArg());21      list.add("one");22      System.out.println(list.add("two"));23   }24}25import static org.mockito.Mockito.mock;26import static org.mockito.Mockito.when;27import static org.mockito.AdditionalAnswers.returnsSecondArg;28import java.util.List;29public class ReturnsSecondArgExample {30   public static void main(String[] args) {31      List list = mock(List.class);32      when(list.add("one")).then(returnsSecondArg());33      list.add("one");34      System.out.println(list.add("two"));35   }36}37import static org.mockito.Mockito.mock;38import static org.mockito.Mockito.when;39import static org.mockito.AdditionalAnswers.returnsArgAt;40import java.util.List;41public class ReturnsArgAtExample {42   public static void main(String[] args) {43      List list = mock(List.class);44      when(list.add("one")).then(returnsArgAt(1));45      list.add("one");46      System.out.println(list.add("two"));47   }48}49import static org.mockito.Mockito.mock;50import static org.mockito.Mockito.when;51import static org.mockito.AdditionalAnswers.returnsElements

returnsLastArg

Using AI Code Generation

1import org.mockito.AdditionalAnswers;2import org.mockito.Mock;3import org.mockito.Mockito;4import org.mockito.MockitoAnnotations;5import org.mockito.Spy;6import org.mockito.stubbing.Answer;7import org.testng.annotations.BeforeMethod;8import org.testng.annotations.Test;9import java.util.Arrays;10import java.util.List;11import static org.mockito.Mockito.*;12public class MockitoReturnsLastArgTest {13    private List<String> list;14    private List<String> spyList;15    public void setUp() {16        MockitoAnnotations.initMocks(this);17    }18    public void testReturnsLastArg() {19        when(list.add(anyString())).then(AdditionalAnswers.returnsLastArg());20        String result = list.add("test");21        System.out.println(result);22    }23    public void testReturnsLastArgWithSpy() {24        when(spyList.add(anyString())).then(AdditionalAnswers.returnsLastArg());25        String result = spyList.add("test");26        System.out.println(result);27    }28    public void testReturnsLastArgWithSpy2() {29        when(spyList.addAll(anyCollection())).then(AdditionalAnswers.returnsLastArg());30        List<String> result = spyList.addAll(Arrays.asList("test1", "test2"));31        System.out.println(result);32    }33    public void testReturnsLastArgWithSpy3() {34        doReturn(true).when(spyList).add(anyString());35        boolean result = spyList.add("test");36        System.out.println(result);37    }38    public void testReturnsLastArgWithSpy4() {39        doReturn(true).when(spyList).addAll(anyCollection());40        boolean result = spyList.addAll(Arrays.asList("test1", "test2"));41        System.out.println(result);42    }43    public void testReturnsLastArgWithSpy5() {44        doAnswer(AdditionalAnswers.returnsLastArg()).when(spyList).add(anyString());45        String result = spyList.add("test");46        System.out.println(result);47    }48    public void testReturnsLastArgWithSpy6() {

returnsLastArg

Using AI Code Generation

1import org.mockito.AdditionalAnswers;2import org.mockito.Mockito;3import java.util.List;4public class ReturnsLastArgExample {5    public static void main(String[] args) {6        List list = Mockito.mock(List.class);7        Mockito.when(list.get(Mockito.anyInt())).then(AdditionalAnswers.returnsLastArg());8        list.add("Hello");9        list.add("Mockito");10        System.out.println(list.get(0));11        System.out.println(list.get(1));12    }13}14Mockito returnsFirstArg() Method Example15Mockito returnsSecondArg() Method Example

returnsLastArg

Using AI Code Generation

1public class 1 {2    public static void main(String[] args) {3        List mock = mock(List.class);4        when(mock.add(anyString())).then(returnsLastArg());5        mock.add("one");6        mock.add("two");7        assertEquals("two", mock.add("three"));8    }9}10public class 2 {11    public static void main(String[] args) {12        List mock = mock(List.class);13        when(mock.add(anyString())).then(returnsFirstArg());14        mock.add("one");15        mock.add("two");16        assertEquals("one", mock.add("three"));17    }18}19public class 3 {20    public static void main(String[] args) {21        List mock = mock(List.class);22        when(mock.add(anyString())).then(returnsFirstArg());23        mock.add("one");24        mock.add("two");25        assertEquals("one", mock.add("three"));26    }27}28public class 4 {29    public static void main(String[] args) {30        List mock = mock(List.class);31        when(mock.add(anyString())).then(returnsArgAt(0));32        mock.add("one");33        mock.add("two");34        assertEquals("one", mock.add("three"));35    }36}37public class 5 {38    public static void main(String[] args) {39        List mock = mock(List.class);40        when(mock.add(anyString())).then(returnsArgAt(1));41        mock.add("one");42        mock.add("two");43        assertEquals("two", mock.add("three"));44    }45}46public class 6 {47    public static void main(String[] args) {48        List mock = mock(List.class);49        when(mock.add(anyString())).then(returnsArg

returnsLastArg

Using AI Code Generation

1import org.mockito.AdditionalAnswers;2import org.mockito.Mockito;3public class ReturnsLastArg {4    public static void main(String[] args) {5        SomeClass mock = Mockito.mock(SomeClass.class);6        Mockito.when(mock.someMethod(Mockito.anyInt(), Mockito.anyInt())).then(AdditionalAnswers.returnsLastArg());7        System.out.println(mock.someMethod(1, 2));8    }9}10class SomeClass {11    public int someMethod(int i, int j) {12        return i + j;13    }14}15Related posts: Mockito – How to use doThrow() method of Mockito class? Mockito – How to use doReturn() method of Mockito class? Mockito – How to use doCallRealMethod() metho

returnsLastArg

Using AI Code Generation

1import org.mockito.Mockito;2import org.mockito.stubbing.Answer;3public class ReturnsLastArgTest {4   public static void main(String[] args) {5      MyInterface myInterface = Mockito.mock(MyInterface.class);6      Answer<String> answer = AdditionalAnswers.returnsLastArg();7      Mockito.when(myInterface.someMethod(Mockito.anyString())).thenAnswer(answer);8      String result = myInterface.someMethod("Hello");9      System.out.println(result);10   }11}12import org.mockito.Mockito;13import org.mockito.stubbing.Answer;14public class ReturnsFirstArgTest {15   public static void main(String[] args) {16      MyInterface myInterface = Mockito.mock(MyInterface.class);17      Answer<String> answer = AdditionalAnswers.returnsFirstArg();18      Mockito.when(myInterface.someMethod(Mockito.anyString(), Mockito.anyString())).thenAnswer(answer);19      String result = myInterface.someMethod("Hello", "World");20      System.out.println(result);21   }22}23import org.mockito.Mockito;24import org.mockito.stubbing.Answer;25public class ReturnsSecondArgTest {26   public static void main(String[] args) {27      MyInterface myInterface = Mockito.mock(MyInterface.class);28      Answer<String> answer = AdditionalAnswers.returnsSecondArg();29      Mockito.when(myInterface.someMethod(Mockito.anyString(), Mockito.anyString())).thenAnswer(answer);30      String result = myInterface.someMethod("Hello", "World");31      System.out.println(result);32   }33}34import org.mockito.Mockito;35import org.mockito.stubbing.Answer;36public class ReturnsArgAtTest {37   public static void main(String[] args) {38      MyInterface myInterface = Mockito.mock(My

returnsLastArg

Using AI Code Generation

1public class Test {2    public void test() {3        List mockList = mock(List.class);4        doAnswer(AdditionalAnswers.returnsLastArg()).when(mockList).add(any());5        mockList.add("hello");6        assertEquals("hello", mockList.get(0));7    }8}9at org.junit.Assert.fail(Assert.java:88)10at org.junit.Assert.failNotEquals(Assert.java:834)11at org.junit.Assert.assertEquals(Assert.java:645)12at org.junit.Assert.assertEquals(Assert.java:631)13at Test.test(1.java:7)14at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)15at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)16at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)17at java.lang.reflect.Method.invoke(Method.java:498)18at org.junit.runners.model.FrameworkMethod$1.runReflectiveCall(FrameworkMethod.java:50)19at org.junit.internal.runners.model.ReflectiveCallable.run(ReflectiveCallable.java:12)20at org.junit.runners.model.FrameworkMethod.invokeExplosively(FrameworkMethod.java:47)21at org.junit.internal.runners.statements.InvokeMethod.evaluate(InvokeMethod.java:17)22at org.junit.internal.runners.statements.RunBefores.evaluate(RunBefores.java:26)23at org.junit.runners.ParentRunner.runLeaf(ParentRunner.java:325)24at org.junit.runners.BlockJUnit4ClassRunner.runChild(BlockJUnit4ClassRunner.java:78)25at org.junit.runners.BlockJUnit4ClassRunner.runChild(BlockJUnit4ClassRunner.java:57)26at org.junit.runners.ParentRunner$3.run(ParentRunner.java:290)27at org.junit.runners.ParentRunner$1.schedule(ParentRunner.java:71)28at org.junit.runners.ParentRunner.runChildren(ParentRunner.java:288)29at org.junit.runners.ParentRunner.access$000(ParentRunner.java:58)30at org.junit.runners.ParentRunner$2.evaluate(ParentRunner.java:268)31at org.junit.runners.ParentRunner.run(ParentRunner.java:363)32at org.eclipse.jdt.internal.junit4.runner.JUnit4TestReference.run(JUnit4TestReference.java:86)33at org.eclipse.jdt.internal.junit.runner.TestExecution.run(TestExecution.java:38)

returnsLastArg

Using AI Code Generation

1import org.mockito.AdditionalAnswers;2import org.mockito.Mockito;3import static org.mockito.Mockito.mock;4public class returnsLastArgExample {5    public static void main(String[] args) {6        MyInterface myInterface = mock(MyInterface.class);7        Mockito.when(myInterface.doSomething("Hello")).then(AdditionalAnswers.returnsLastArg());8        System.out.println(myInterface.doSomething("Hello"));9    }10}11interface MyInterface {12    String doSomething(String s);13}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.