Y.A.M の雑記帳: Android で mockito を使う : メソッドの呼び出しをチェックする

JSON をパースして、リスナーの対応するメソッドを呼び出すユーティリティメソッドがあるとします。


public class Utils {

    public interface ResultListener {

        void onError();

        void onHoge1();

        void onHoge2();
    }

    public static void handleJson(String json, ResultListener listener) {
        if (listener == null) {
            return;
        }

        try {
            JSONObject obj = new JSONObject(json);

            String status = obj.optString("Status");

            if (status.equals("hoge1")) {
                listener.onHoge1();

            } else if (status.equals("hoge2")) {
                listener.onHoge2();
            }

        } catch (JSONException e) {
            listener.onError();
            e.printStackTrace();
        }
    }
}

このメソッドをテストするには、JSON文字列を渡して、対応するリスナーのメソッドがちゃんと呼ばれるかどうか確認すればいいわけです。

まず、テストプロジェクトにテストしたいクラスと同じパッケージを作ります。

ここではテスト対象クラス（Utils）が com.example.mockitosample なので、テストプロジェクトの MockitoSampleTest にも com.example.mockitosample を作ります。

テストプロジェクトの com.example.mockitosample を選択して右クリックし、[New] → [JUnit Test Case] を選択します。
Name にテストクラス名を入れて Finish をクリックします。
テスト対象クラス名 + Test にすることが多いです。

このユーティリティメソッドの動作パターンとして以下があります。

1. 引数で渡す文字列が JSON 文字列として正しくない場合、onError() が呼ばれ他のメソッドは呼ばれない
2. （引数で渡す）JSON 文字列に Status キーが無い場合、onError() も他のメソッドも呼ばれない
3. JSON 文字列の Status キーの値が hoge1 でも hoge2 でもない場合、onError() も他のメソッドも呼ばれない
4. JSON 文字列の Status キーの値が hoge1 の場合、onHoge1() が呼ばれ他のメソッドは呼ばれない
5. JSON 文字列の Status キーの値が hoge2 の場合、onHoge2() が呼ばれ他のメソッドは呼ばれない

ということでテストメソッドは5つです。
JUnit3 では test で始まるメソッドがテストメソッドとして認識されます。

まず、「1. 引数で渡す文字列が JSON 文字列として正しくない場合、onError() が呼ばれ他のメソッドは呼ばれない」のテストメソッドを作ってみます。


    /**
     * 引数で渡す文字列が JSON 文字列として正しくない場合、 onError() が呼ばれ他のメソッドは呼ばれないことを確認する
     */
    public void testInvalidJson() {
    }

mockito を使わない場合「グローバル変数を用意してリスナーのメソッドで変数の値を変えて、どれが呼ばれたかチェックする」みたいなことをしますが、mockito を使うともっとスマートにできます。

まず、メソッドが呼ばれたかチェックしたいクラスをモック化します。
ここでは ResultListener のメソッドが呼ばれたかチェックしたいので ResultListener をモック化します。

モック化には Mockito.mock() を使います。
他にも Mockito 下の static method をたくさん使うので

import static org.mockito.Mockito.*;

を入れておくといいでしょう。


ResultListener mockResultListener = mock(ResultListener.class);

あとはこのモックを使ってテストしたいメソッドを呼び出します。


ResultListener mockResultListener = mock(ResultListener.class);
Utils.handleJson("", mockResultListener);

ここでは JSON 文字列として不適切な空文字を渡しているので onError() が呼ばれるはずです。
それをチェックするのが verify() メソッドです。


ResultListener mockResultListener = mock(ResultListener.class);
Utils.handleJson("", mockResultListener);

verify(mockResultListener).onError();

onHoge1() と onHoge2() は呼ばれないはずです。呼ばれなかったかどうかをチェックするには、verify() メソッドの第2引数に never() を指定します。


ResultListener mockResultListener = mock(ResultListener.class);
Utils.handleJson("", mockResultListener);

verify(mockResultListener).onError();
verify(mockResultListener, never()).onHoge1();
verify(mockResultListener, never()).onHoge2();

これだと例えば onHoge3() メソッドが増えたときに修正が必要になります。
verify() メソッドの第2引数に only() を付けることで onError() しか呼ばれていないということをチェックできます。


ResultListener mockResultListener = mock(ResultListener.class);
Utils.handleJson("", mockResultListener);

verify(mockResultListener, only()).onError();

テストメソッドの全体はこうなります。


    /**
     * 引数で渡す文字列が JSON 文字列として正しくない場合、 onError() が呼ばれ他のメソッドは呼ばれないことを確認する
     */
    public void testInvalidJson() {
        ResultListener mockResultListener = mock(ResultListener.class);
        Utils.handleJson("", mockResultListener);
        
        verify(mockResultListener, only()).onError();
    }

「2. （引数で渡す）JSON 文字列に Status キーが無い場合、onError() も他のメソッドも呼ばれない」のテストを作ってみましょう。

これ以降になにも行われないことをチェックするメソッドとして verifyNoMoreInteractions() があります。
ここでは全てのメソッドが呼ばれないのでこれを使うことが出来ます。


    /**
     * Status キーが無い場合、どのメソッドも呼ばれないことを確認する
     */
    public void testNoStatusKey() {
        ResultListener mockResultListener = mock(ResultListener.class);
        Utils.handleJson("{\"State\":\"hoge1\"}", mockResultListener);
        
        verifyNoMoreInteractions(mockResultListener);
    }

「3. JSON 文字列の Status キーの値が hoge1 でも hoge2 でもない場合、onError() も他のメソッドも呼ばれない」のテストを作ってみましょう。

引数で渡す文字列が変わるだけで 2. と一緒ですね


    /**
     * Status キーの値が hoge1 でも hoge2 でも無い場合、どのメソッドも呼ばれないことを確認する
     */
    public void testInvalidStatusValue() {
        ResultListener mockResultListener = mock(ResultListener.class);
        Utils.handleJson("{\"Status\":\"hoge3\"}", mockResultListener);
        
        verifyNoMoreInteractions(mockResultListener);
    }

「4. JSON 文字列の Status キーの値が hoge1 の場合、onHoge1() が呼ばれ他のメソッドは呼ばれない」のテストを作ってみましょう。


    /**
     * Status キーの値が hoge1 の場合、onHoge1() が呼ばれ他のメソッドは呼ばれないことを確認する
     */
    public void testHoge1() {
        ResultListener mockResultListener = mock(ResultListener.class);
        Utils.handleJson("{\"Status\":\"hoge1\"}", mockResultListener);
        
        verify(mockResultListener, only()).onHoge1();
    }

「5. JSON 文字列の Status キーの値が hoge2 の場合、onHoge2() が呼ばれ他のメソッドは呼ばれない」のテストは 4. とほぼ同じなので省略します。

ResultListener をモック化する処理


ResultListener mockResultListener = mock(ResultListener.class);

は各メソッドで行っているので、mockResultListener をフィールドとして保持するようにして setup() メソッドでモック化させると、このテストクラスがどのモックに依存しているかわかりやすくなります。


public class UtilsTest extends TestCase {

    ResultListener mockResultListener;
    
    @Override
    protected void setUp() throws Exception {
        super.setUp();
        mockResultListener = mock(ResultListener.class);
    }
    
    /**
     * 引数で渡す文字列が JSON 文字列として正しくない場合、 onError() が呼ばれ他のメソッドは呼ばれないことを確認する
     */
    public void testInvalidJson() {
        Utils.handleJson("", mockResultListener);

        verify(mockResultListener, only()).onError();
    }

    ...
}

5つともグリーンになりました。

verify() の第2引数に times(int) を渡すことで何回呼ばれたかをチェックすることができます。
第2引数を指定しない場合は times(1) を指定したのと同じことになります。


verify(mockResultListener, times(2).onHoge3();

Y.A.M の雑記帳

2013年6月21日金曜日

Android で mockito を使う : メソッドの呼び出しをチェックする

1 件のコメント: