Mockito - 如何模拟平等？

Question

我有以下需要模拟的条件语句。 是否可以使用 mockito 来模拟它？

if (events.size() == rowsAffected) {
// app logic
}

这里的events是List<Map<String, Object>>类型，将从 function 返回。

rowsAffected是int类型，它也将由方法返回。

List<Map<String, Object>> events = service.getEvents();
int rowsAffected = service.updateEventStatus(events);

Answer 1

当您需要模拟具有自定义行为的方法时，可以使用answer 。 例子：

public class MockMe {
    public String hello(String name) {
        return "Hello " + name;
    }
}

@Test
public void testCase() {
    List<Map<String, Object>> events = service.getEvents();
    int rowsAffected = service.updateEventStatus(events);
    Answer<String> myMockedAnswer = invocation -> {
        if (events.size() == rowsAffected) {
            return "mock good";
        }
        return "mock bad";
    };
    MockMe mockMe = mock(mockMe);
    doAnswer(myMockedAnswer)
        .when(mockMe).hello(anyString());
    assertEquals("mock good", mockMe.hello("anything"));
}