Mockito.when thenReturn 基于输入参数

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))));