web-dev-qa-db-ja.com

Mockitoは、注入されたモックオブジェクトのメソッドに渡される引数をどのようにキャプチャできますか？

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がモックの依存関係を実際のクラスに注入し、実際のクラスの単体テストによってモックが呼び出されると、注入されたモックに渡された正確な引数を後で取得する方法を教えてください。

javaspringunit-testingmockito
34
Steve Perkins

1つ以上のArgumentCaptorsを使用します。

どのタイプがここにあるかは不明ですが、とにかくです。 doSomething()が引数として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());
70
fge

doAnswer()をamqpTemplateMockのsend()メソッドのスタブにフックし、AmqpTemplate.send()の呼び出し引数をキャプチャできます。

testDoSomething()の最初の行をこれにします

    Mockito.doAnswer(new Answer<Void>() {
          @Override
          public Void answer(final InvocationOnMock invocation) {
            final Object[] args = invocation.getArguments();
            System.out.println("UUID=" + args[0]);  // do your assertions here
            return null;
          }
    }).when(amqpTemplateMock).send(Matchers.anyString(), Matchers.anyObject());

すべてをまとめると、テストは

import org.junit.Before;
import org.junit.Test;
import org.mockito.InjectMocks;
import org.mockito.Matchers;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

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() throws Exception {
    Mockito.doAnswer(new Answer<Void>() {
      @Override
      public Void answer(final InvocationOnMock invocation) {
        final Object[] args = invocation.getArguments();
        System.out.println("UUID=" + args[0]);  // do your assertions here
        return null;
      }
    }).when(amqpTemplateMock).send(Matchers.anyString(), Matchers.anyObject());
    userService.doSomething(Long.toString(System.currentTimeMillis()));
  }
}

これは出力を与えます

UUID = 8e276a73-12fa-4a7e-a​​7cc-488d1ce0291f

私はこの投稿を読んでこれを見つけました mockitoでモックを無効にする方法

8
Kirby