java - Mockito.when then根据输入参数返回

Question

我有一些加密的代码。在测试中，我想分析加密的数据包，解密并返回结果。我想要的是这样的:

Mockito.when(myClient.sendMessage(someEncryptedRandomMessage)).thenReturn(encrypt(decryptAndAnalyze(someEncryptedRandomMessage)));

我怎样才能实现它？

Answer 1

有 thenAnswer 方法可以访问传递的参数。假设您的 someEncryptedRandomMessage 是 String 它看起来像这样

Mockito.when(myClient.sendMessage(someEncryptedRandomMessage))
       .thenAnswer(inv -> encrypt(decryptAndAnalyze(inv.getArgumentAt(0, String.class))));