21

テストしたい次のコードがあります。

public class MessageService {
    private MessageDAO dao;

    public void acceptFromOffice(Message message) {
        message.setStatus(0);
        dao.makePersistent(message);

        message.setStatus(1);
        dao.makePersistent(message);

    }
    public void setDao (MessageDAO mD) { this.dao = mD; }
}

public class Message {
    private int status;
    public int getStatus () { return status; }
    public void setStatus (int s) { this.status = s; }

    public boolean equals (Object o) { return status == ((Message) o).status; }

    public int hashCode () { return status; }
}

メソッドacceptFromOfficeが実際にステータスを0に設定し、メッセージを永続化してから、ステータスを1に変更してから、再度永続化することを確認する必要があります。

Mockitoを使用して、次のことを試みました。

@Test
    public void testAcceptFromOffice () throws Exception {

        MessageDAO messageDAO = mock(MessageDAO.class);

        MessageService messageService = new MessageService();
        messageService.setDao(messageDAO);

        final Message message = spy(new Message());
        messageService.acceptFromOffice(message);

        verify(messageDAO).makePersistent(argThat(new BaseMatcher<Message>() {
            public boolean matches (Object item) {
                return ((Message) item).getStatus() == 0;
            }

            public void describeTo (Description description) { }
        }));

        verify(messageDAO).makePersistent(argThat(new BaseMatcher<Message>() {
            public boolean matches (Object item) {
                return ((Message) item).getStatus() == 1;
            }

            public void describeTo (Description description) { }
        }));

    }

私は実際、ここで検証により、異なるメッセージオブジェクトの状態でmakePersistentメソッドを2回呼び出すことを検証することを期待しています。しかし、それは失敗します

引数が違います！

手がかりはありますか？

4

2 に答える 2

27

コードのテストは簡単ではありませんが、不可能ではありません。私の最初のアイデアは、ArgumentCaptorを使用することでした。これは、 ArgumentMatcherと比較して、使用と理解の両方がはるかに簡単です。残念ながら、テストはまだ失敗します-理由は確かにこの回答の範囲を超えていますが、それがあなたに興味があるなら私は助けるかもしれません。それでも私はこのテストケースが示されるのに十分興味深いと思います（正しい解決策ではありません）：

@RunWith(MockitoJUnitRunner.class)
public class MessageServiceTest {

    @Mock
    private MessageDAO messageDAO = mock(MessageDAO.class);

    private MessageService messageService = new MessageService();

    @Before
    public void setup() {
        messageService.setDao(messageDAO);
    }

    @Test
    public void testAcceptFromOffice() throws Exception {
        //given
        final Message message = new Message();

        //when
        messageService.acceptFromOffice(message);

        //then
        ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);

        verify(messageDAO, times(2)).makePersistent(captor.capture());

        final List<Message> params = captor.getAllValues();
        assertThat(params).containsExactly(message, message);

        assertThat(params.get(0).getStatus()).isEqualTo(0);
        assertThat(params.get(1).getStatus()).isEqualTo(1);
    }

}

残念ながら、実用的なソリューションでは、Answerをやや複雑に使用する必要があります。簡単に言うと、Mockitoに各呼び出しを記録して検証させる代わりに、テストコードが特定のモックを実行するたびに実行される一種のコールバックメソッドを提供します。このコールバックメソッド（このMakePersistentCallback例ではオブジェクト）では、両方のパラメーターにアクセスでき、戻り値を変更できます。これは重い大砲であり、注意して使用する必要があります。

    @Test
    public void testAcceptFromOffice2() throws Exception {
        //given
        final Message message = new Message();
        doAnswer(new MakePersistentCallback()).when(messageDAO).makePersistent(message);

        //when
        messageService.acceptFromOffice(message);

        //then
        verify(messageDAO, times(2)).makePersistent(message);
    }


    private static class MakePersistentCallback implements Answer {

        private int[] expectedStatuses = {0, 1};
        private int invocationNo;

        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            final Message actual = (Message)invocation.getArguments()[0];
            assertThat(actual.getStatus()).isEqualTo(expectedStatuses[invocationNo++]);
            return null;
        }
    }

例は完全ではありませんが、CUTでほとんどすべてを変更すると、テストは成功し、さらに重要なことに失敗します。ご覧のとおり、MakePersistentCallback.answerモックが呼び出されるたびにメソッドが呼び出さmessageService.acceptFromOffice(message)れます。内部naswerでは、必要なすべての検証を実行できます。

注意：このようなテストを維持することは控えめに言っても面倒な場合があるため、注意して使用してください。

于 2011-03-09T18:38:14.220 に答える
0

事実上、ステートマシンをテストしています。いくつかのカスタム実装を使用してMessageServiceをテストするのは非常に簡単です。TestMessageが最も興味深いクラスになると思います。

DAO /メッセージがpersist呼び出しを記録できるようにするために、カスタム実装を行いました。

それはMockitoではありませんが、それは単純であり、仕事をする必要があります。

class TestMessageDAO implements MessageDAO {
  // I have no clue what the MessageDAO does except for makePersistent
  // which is the only relevant part here

  public void makePersistent(Message message) {
    if (message instanceof TestMessage) {
      TestMessage test = (TestMessage)message;
      test.persistCalled(); // will be recorded by TestMessage
    } else {
      throw RuntimeException("This test DAO does not support non-test messages");
    }
  }
}

// Message isn't final so...
class TestMessage extends Message {
  enum state {
    STARTED, STATUS0, PERSIST0, STATUS1, PERSIST1
  }

  public void persistCalled() { // For testing only
    switch (state) {
      case STATUS0:
        state = PERSIST0;
        break;
      case STATUS1:
        state = PERSIST1;
        break;
      default:
        throw new RuntimeException("Invalid transition");
    }
  }

  public void setStatus(int status) {
    switch(state) {
      case STARTED:
        if (status != 0) {
          throw new IllegalArgumentException("0 required");
        }
        state = STATUS0;
        break;
      case PERSIST0:
        if (status != 1) {
          throw new IllegalArgumentException("1 required");
        }

        state = STATUS1;
        break;
      default:
        throw new RuntimeException("Invalid transition");
    }
  }
}

public class TestMessageService {

  @Test
  public void testService() {
    MessageDAO dao = new TestMessageDAO();
    Message message = new TestMessage();
    MessageService service = new MessageService();
    service.setDao(dao);
    service.acceptFromOffice(message);
  }

}
于 2011-03-09T20:44:19.953 に答える