使模拟方法返回传递给它的参数 [英] Making a mocked method return an argument that was passed to it

问题描述

考虑如下方法签名:

public String myFunction(String abc);

Mockito 能否帮助返回与方法接收到的相同的字符串?

Can Mockito help return the same string that the method received?

推荐答案

您可以在 Mockito 中创建答案.假设我们有一个名为 Application 的接口，其中包含一个 myFunction 方法.

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

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

这是带有 Mockito 答案的测试方法:

Here is the test method with a Mockito answer:

public void testMyFunction() throws Exception {
  Application mock = mock(Application.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"));
}

从 Mockito 1.9.5 和 Java 8 开始，您还可以使用 lambda 表达式:

Since Mockito 1.9.5 and Java 8, you can also use a lambda expression:

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

这篇关于使模拟方法返回传递给它的参数的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！