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