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