Wie kann Mockito Argumente erfassen, die an die Methoden eines eingefügten Mock-Objekts übergeben werden?

Ich versuche, eine Dienstklasse zu testen, die intern ein Spring-AMQP-Verbindungsobjekt verwendet. Dieses Verbindungsobjekt wird von Spring injiziert. Ich möchte jedoch nicht, dass mein Komponententest tatsächlich mit dem AMQP-Broker kommuniziert, also verwende ich Mockito, um einen Schein des Verbindungsobjekts einzufügen.

/** 
 * 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.
   }
}

Die Kommentare in diesem Codebeispiel stellen die Frage hoffentlich klar dar. Wenn Mockito eine Scheinabhängigkeit in eine reale Klasse einfügt und Komponententests für die reale Klasse dazu führen, dass sie Aufrufe an die Scheinstruktur tätigt, wie können Sie dann später die genauen Argumente abrufen, die an die eingefügte Scheinklasse übergeben wurden?

Verwenden Sie eine oder mehrere ArgumentCaptorS.

Es ist unklar, was Ihre Typen hier sind, aber trotzdem. Nehmen wir an, Sie haben einen Mock, der eine Methode hat doSomething() ein Nehmen Foo als Argument, dann machst du das:

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

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

final Foo argument = captor.getValue();

// Test the argument

Außerdem sieht es so aus, als ob Ihre Methode void zurückgibt und Sie nicht möchten, dass sie etwas tut. Schreib einfach das:

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

Sie können haken doAnswer() zum Stummel der send() Methode an amqpTemplateMock und erfassen Sie dann die Aufrufargumente von AmqpTemplate.send().

Machen Sie die erste Zeile Ihrer testDoSomething() sei dies

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

Alles zusammengenommen wird der Test

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

Dies ergibt eine Ausgabe

UUID=8e276a73-12fa-4a7e-a7cc-488d1ce0291f

Ich habe dies gefunden, indem ich diesen Beitrag gelesen habe, Wie man Mock-to-Void-Methoden mit Mockito erstellt