使用 Mockito 为 Java 8 个 lambda 表达式编写存根

Question

我有一个使用 Java 8 lambda 表达式的方法。在为此方法编写单元测试时，我需要为该表达式编写存根。我们怎么写呢？

public List<User> methodA(String userId) {
    List<User> users = new ArrayList<>();
    userRepository.findAll().forEach(users::add);
    List<User> usersFilterByUserId = users.stream().filter(u -> u.getUserid().equalsIgnoreCase(userId)).collect(Collectors.toList());

    some other stuff ....
}

我尝试了给定的解决方案 -

@Test
public void testMethodA() {
    ArrayList<User> mockUsers = mock(ArrayList.class);
    PowerMockito.whenNew(ArrayList.class).withNoArguments().thenReturn(mockUsers);
    User user = mock(User.class);
    Iterable<User> mockIt = mock(Iterable.class);
    when(userRepository.findAll()).thenReturn(mockIt);
    doNothing().when(mockIt).forEach(any());

    // This gives compiler error which is obvious 
    // The method thenReturn(Stream<User>) in the type 
    // OngoingStubbing<Stream<User>> is not applicable for the arguments (ArrayList<User>)
    ArrayList<User> mockUsersFilterByUserId = mock(ArrayList.class);
    when(mockUsers.stream()).thenReturn(mockUsersFilterByUserId);
    ...
}

Answer 1

实际上，您在单元测试中模拟了所有内容。
测试变得复杂并失去了它的价值。

在 methodA 中，您应该模拟的唯一一件事是依赖项：

userRepository.findAll().

通过模拟对此方法的调用，您将在 lambda 主体中使用模拟数据，因为它使用 findAll().

的结果

List<User> usersByMock = new ArrayList<>();
usersByMock.add(...);
usersByMock.add(...);
usersByMock.add(...);
...
when(userRepository.findAll()).thenReturn(usersByMock);

使用 Mockito 为 Java 8 个 lambda 表达式编写存根

Use Mockito to write stub for Java 8 lambda expressions

java

lambda

unit-testing

mockito

powermockito