Mockito如何捕获传递给注入的模拟对象的方法的参数？

Question

我正在尝试测试服务类，该服务类在内部使用Spring
AMQP连接对象。这个连接对象是由Spring注入的。但是，我不希望单元测试实际上与AMQP代理进行通信，因此我正在使用Mockito注入连接对象的模拟。

/** 
 * The real service class being tested.  Has an injected dependency. 
 */ 
public class UserService {

   @Autowired
   private AmqpTemplate amqpTemplate;

   public final String doSomething(final String inputString) {
      final String requestId = UUID.randomUUID().toString();
      final Message message = ...;
      amqpTemplate.send(requestId, message);
      return requestId;
   }
}

/** 
 * Unit test 
 */
public class UserServiceTest {

   /** This is the class whose real code I want to test */
   @InjectMocks
   private UserService userService;

   /** This is a dependency of the real class, that I wish to override with a mock */
   @Mock
   private AmqpTemplate amqpTemplateMock;

   @Before
   public void initMocks() {
      MockitoAnnotations.initMocks(this);
   }

   @Test
   public void testDoSomething() {
      doNothing().when(amqpTemplateMock).send(anyString(), any(Message.class));

      // Call the real service class method, which internally will make 
      // use of the mock (I've verified that this works right).
      userService.doSomething(...);

      // Okay, now I need to verify that UUID string returned by 
      // "userService.doSomething(...) matches the argument that method 
      // internally passed to "amqpTemplateMock.send(...)".  Up here 
      // at the unit test level, how can I capture the arguments passed 
      // to that inject mock for comparison?
      //
      // Since the value being compared is a UUID string created 
      // internally within "userService", I cannot just verify against 
      // a fixed expected value.  The UUID will by definition always be
      // unique.
   }
}

希望此代码示例中的注释清楚地提出问题。当Mockito将模拟依赖项注入到真实类中，并且在真实类上进行单元测试导致它html" target="_blank">调用该模拟对象时，您以后如何才能检索传递给注入的模拟对象的确切参数？

Answer 1

使用一个或多个ArgumentCaptor。

目前尚不清楚您的类型在这里，但无论如何。假设您有一个包含doSomething()将a Foo作为参数的方法的模拟，然后执行此操作：

final ArgumentCaptor<Foo> captor = ArgumentCaptor.forClass(Foo.class);

verify(mock).doSomething(captor.capture());

final Foo argument = captor.getValue();

// Test the argument

而且，看起来您的方法返回了void，并且您不希望它做任何事情。只需写下：

doNothing().when(theMock).doSomething(any());