Java mock工具-mockito

转载自：http://blog.iamzsx.me/show.html?id=118001

Mockito是目前java中使用比较流行的mock工具。http://code.google.com/p/mockito/

所谓的mock，就是指，如果我们写的代码依赖于某些对象，而这些对象又很难手动创建，那么就使用一个虚拟的对象来测试。

下面举个例子。

如下所示，我写了一个类，randomSelect函数的主要功能是，根据weights数组指定的比例（概率），随机返回一个[0,weights.length)间的整数。

01	import java.util.Random;

02

03	public class RandomUtil {

04

05	private final static Random DEFAULT_RANDOM = new Random();

06

07	public static int randomSelect(int[] weights) {

08	return randomSelect(DEFAULT_RANDOM, weights);

09

}

10

11	public static int randomSelect(Random r, int[] weights) {

12	if (weights == null || weights.length == 0) {

13	throw new IllegalArgumentException(

14	"weights must not be an empty array");

15

}

16

17	int target = r.nextInt(MathUtil.sum(weights));

18	int accumulator = 0;

19	for (int i = 0; i < weights.length; i++) {

20	accumulator += weights[i];

21	if (accumulator > target) {

22

return i;

23

}

24

}

25

return 0;

26

}

27

}

由于Random是一个系统的类，用来产生随机数，这样子我们无法控制Random的行为。如果不使用mockito，那么我们的测试代码可能是下面这样子。我们会写一个子类来继承Random，这样子我们就可以通过覆盖父类的方法来控制Random的行为了。

01	import java.util.Random;

02

03	import org.junit.Assert;

04	import org.junit.Test;

05

06	public class TestRandomUtil {

07

08

@Test

09	public void testSelect() {

10	int[] weights = { 2, 1, 3, 0, 4 };

11	int sum = MathUtil.sum(weights);

12	Random r = new IntArrayRandom(MathUtil.range(0, sum));

13	int[] expected = new int[] { 0, 0, 1, 2, 2, 2, 4, 4, 4, 4 };

14	for (int i = 0; i < sum; i++) {

15	Assert.assertEquals(expected[i], RandomUtil.randomSelect(r, weights));

16

}

17

}

18

19	@SuppressWarnings("serial")

20	public static class IntArrayRandom extends Random {

21

22	private int[] seq;

23	private int index;

24

25	public IntArrayRandom(int[] seq) {

26	this.seq = seq;

27	this.index = 0;

28

}

29

30

@Override

31	public int nextInt(int n) {

32	return seq[index++];

33

}

34

35

}

36

}

那么，使用mockito有什么好处呢，mockito可以帮我们省去IntArrayRandom这个类。见下面的代码：

01	import static org.mockito.Mockito.mock;

02	import static org.mockito.Mockito.when;

03

04	import java.util.Random;

05

06	import org.junit.Assert;

07	import org.junit.Test;

08

09	public class TestRandomUtil  {

10

11

@Test

12	public void testSelect() {

13

14	int[] weights = { 2, 1, 3, 0, 4 };

15	int sum = MathUtil.sum(weights);

16

17	int[] expected = new int[] { 0, 0, 1, 2, 2, 2, 4, 4, 4, 4 };

18	for (int i = 0; i < sum; i++) {

19	Random r = mock(Random.class);// 创建一个mock对象

20	when(r.nextInt(sum)).thenReturn(i);// 配置mock行为：当r.nextInde被调用，且参数为sum时，返回i

21	Assert.assertEquals(expected[i],

22	RandomUtil.randomSelect(r, weights));

23

}

24

}

25

26

}

可以看到，mockito帮我省掉了一大堆代码。这里只是很简单的例子，在实际开发中，可以大大提高编写单元测试的效率。