java - Mockito:如何模拟方法，其结果取决于其参数之一？

Question

有没有办法将参数传递给模拟函数并在内部使用该参数值。示例(通知名称作为参数):

Mockito.when(clientRepo.registerNewClient(Mockito.any(String.class) as name))    .thenReturn(        dslContext            .insertInto(CLIENT)            .set(CLIENT.CLIENT_NAME, name)            .execute());

有办法做到这一点吗？

Answer 1

您需要使用thenAnswer，并从InitationOnMock获取您的参数。

final Repo clientRepo = Mockito.mock(Repo.class);
Mockito.when(clientRepo.registerNewClient(Mockito.any(String.class)))
    .thenAnswer(
        (Answer<Client>)
            invocationOnMock -> new Client(
                invocationOnMock
                    .getArgumentAt(0, String.class)
                    .toUpperCase()
            )
    );
Assertions.assertEquals(
    clientRepo.registerNewClient("fff"), new Client("FFF")
);