模拟具有不同参数的相同方法

unit-testing java junit mockito

2022-09-01 01:01:11

我正在使用mockito来测试我的业务服务，它使用我想要模拟的实用程序。每个服务方法中至少有 2-3 个调用，用于具有不同参数的实用程序。

是否有任何推荐的方法可以对相同方法但不同的参数使用多个方法？when(...).thenReturn(...)

我也想在里面使用游行者。可能吗？any()

更新：示例代码。

@Test
public void myTest() {
  when(service.foo(any(), new ARequest(1, "A"))).thenReturn(new AResponse(1, "passed"));
  when(service.foo(any(), new ARequest(2, "2A"))).thenReturn(new AResponse(2, "passed"));
  when(service.foo(any(), new BRequest(1, "B"))).thenReturn(new BResponse(112, "passed"));

  c.execute();
}

public class ClassUnderTest {
  Service service = new Service();
  public void execute() {
    AResponse ar = (AResponse) service.foo("A1", new ARequest(1, "A"));
    AResponse ar2 = (AResponse) service.foo("A2", new ARequest(2, "2A"));
    BResponse br = (BResponse) service.foo("B1", new BRequest(1, "B"));
  }
}

public class Service {
  public Object foo(String firstArgument, Object obj) {
    return null; //return something
  }
}

答案 1

一种方法是避免对参数过于严格，以便仅通过一个调用提供所有预期结果。thenReturn

例如，假设我想模拟此方法：

public String foo(String firstArgument, Object obj) {
    return "Something";
}

然后，您可以通过提供任意数量的结果来模拟它，如下所示：

// Mock the call of foo of any String to provide 3 results
when(mock.foo(anyString(), anyObject())).thenReturn("val1", "val2", "val3");

调用任何参数将分别提供“”、“”，则任何后续调用都将提供“”。fooval1val2val3

如果您确实关心传递的值，但不想依赖于调用顺序，则可以使用该答案来提供与第二个参数匹配的答案，就像您当前所做的那样，但具有3个不同的。假设您已覆盖该方法。thenAnswerthenReturnequals(Object o)

when(mock.foo(anyString(), anyObject())).thenAnswer(
    invocation -> {
        Object argument = invocation.getArguments()[1];
        if (argument.equals(new ARequest(1, "A"))) {
            return new AResponse(1, "passed");
        } else if (argument.equals(new ARequest(2, "2A"))) {
            return new AResponse(2, "passed");
        } else if (argument.equals(new BRequest(1, "B"))) {
            return new BResponse(112, "passed");
        }
        throw new InvalidUseOfMatchersException(
            String.format("Argument %s does not match", argument)
        );
    }
);

或者简单地说，使用这些方法并作为争论进行者。假设您已覆盖该方法。anyStringeqequals(Object o)

when(service.foo(anyString(), eq(new ARequest(1, "A"))))
    .thenReturn(new AResponse(1, "passed"));
when(service.foo(anyString(), eq(new ARequest(2, "2A"))))
    .thenReturn(new AResponse(2, "passed"));
when(service.foo(anyString(), eq(new BRequest(1, "B"))))
    .thenReturn(new BResponse(112, "passed"));

答案 2

正确的方法是使用匹配参数，但如果您不想这样做，则可以只记录多个返回值。eq()

when(someService.doSomething(any(SomeParam.class))).thenReturn(
  firstReturnValue, secondReturnValue, thirdReturnValue
);

现在第一个调用将返回，第二个和所有后续调用。firstValuesecondValuethirdValue