MockitoのMathcersで引数を決め打ってリターン値を決める

概要

Mockitoでテストしていると、時々同一メソッドを複数回コールしているが、渡される引数が決まっているような状況があります。
その時に

• AだったらC
• BだったらD

のようにしたい、と

下記のメソッドを使います。

Matchers.eq

もうちょっと具体的なところへ落とし込む

とあるメソッドへ渡される引数が3つあるとし、そのうちの1つはEnum値で何が渡されるかはある程度決まっているようなケース

SampleEnum.java

public enum SampleEnum {
	CASE_A,
	CASE_B,
	CASE_C;
}

Sample.java

public class Sample {

	public String branch(String argStr, int argNum, SampleEnum sampleEnum) {
		String retStr = "";
		switch (sampleEnum) {
		case CASE_A:
			retStr = "CASE_A:" + argStr + argNum;
			break;
		case CASE_B:
			retStr = "CASE_B:" + argStr + argNum;
			break;
		case CASE_C:
			retStr = "CASE_C:" + argStr + argNum;
			break;
		default:
			break;
		}
		return retStr;
	}
}

以下がテスト。

SampleTest.java

import static org.hamcrest.CoreMatchers.*;
import static org.junit.Assert.*;
import static org.mockito.Matchers.*;
import static org.mockito.Mockito.*;

import org.junit.Test;

public class SampleTest {

	@Test
	public void test() {
		Sample testTarget = new Sample();
		Sample spyTestTarget = spy(testTarget);

		doReturn("TEST_A").when(spyTestTarget).branch(anyString(), anyInt(),
				eq(SampleEnum.CASE_A));

		String result = spyTestTarget.branch("test", 2, SampleEnum.CASE_A);
		assertThat(result, is("TEST_A"));

		result = spyTestTarget.branch("test", 2, SampleEnum.CASE_B);
		assertThat(result, is("CASE_B:test2"));
	}
}

まとめ

上記のdoReturn("TEST_A").when(spyTestTarget).branch(anyString(), anyInt(),eq(SampleEnum.CASE_A));
で、Enum値を狙い撃ってリターン値を想定通りにすることができます。
Enum値がSampleEnum.CASE_A以外が渡されると本来の処理が行われていることも確認できます。

特定のケースの時に値を変更しておきたいテストをする時とかに便利です。