Java mock工具-mockito

60 阅读 0 评论 40 点赞

我是靠谱客的博主含糊蜡烛，最近开发中收集的这篇文章主要介绍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)间的整数。

 
import java.util.Random;
 
 
 
public class RandomUtil {
 
 
 
 private final static Random DEFAULT_RANDOM = new Random();
 
 
 
 public static int randomSelect(int[] weights) {
 
  return randomSelect(DEFAULT_RANDOM, weights);
 
 }
 
 
 
 public static int randomSelect(Random r, int[] weights) {
 
  if (weights == null || weights.length == 0) {
 
   throw new IllegalArgumentException(
 
     "weights must not be an empty array");
 
  }
 
 
 
  int target = r.nextInt(MathUtil.sum(weights));
 
  int accumulator = 0;
 
  for (int i = 0; i < weights.length; i++) {
 
   accumulator += weights[i];
 
   if (accumulator > target) {
 
    return i;
 
   }
 
  }
 
  return 0;
 
 }
 
}

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

 
import java.util.Random;
 
 
 
import org.junit.Assert;
 
import org.junit.Test;
 
 
 
public class TestRandomUtil {
 
 
 
 @Test
 
 public void testSelect() {
 
  int[] weights = { 2, 1, 3, 0, 4 };
 
  int sum = MathUtil.sum(weights);
 
  Random r = new IntArrayRandom(MathUtil.range(0, sum));
 
  int[] expected = new int[] { 0, 0, 1, 2, 2, 2, 4, 4, 4, 4 };
 
  for (int i = 0; i < sum; i++) {
 
   Assert.assertEquals(expected[i], RandomUtil.randomSelect(r, weights));
 
  }
 
 }
 
 
 
 @SuppressWarnings("serial")
 
 public static class IntArrayRandom extends Random {
 
 
 
  private int[] seq;
 
  private int index;
 
 
 
  public IntArrayRandom(int[] seq) {
 
   this.seq = seq;
 
   this.index = 0;
 
  }
 
 
 
  @Override
 
  public int nextInt(int n) {
 
   return seq[index++];
 
  }
 
 
 
 }
 
}

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

 
import static org.mockito.Mockito.mock;
 
import static org.mockito.Mockito.when;
 
 
 
import java.util.Random;
 
 
 
import org.junit.Assert;
 
import org.junit.Test;
 
 
 
public class TestRandomUtil  {
 
 
 
 @Test
 
 public void testSelect() {
 
 
 
  int[] weights = { 2, 1, 3, 0, 4 };
 
  int sum = MathUtil.sum(weights);
 
 
 
  int[] expected = new int[] { 0, 0, 1, 2, 2, 2, 4, 4, 4, 4 };
 
  for (int i = 0; i < sum; i++) {
 
   Random r = mock(Random.class);// 创建一个mock对象
 
   when(r.nextInt(sum)).thenReturn(i);// 配置mock行为：当r.nextInde被调用，且参数为sum时，返回i
 
   Assert.assertEquals(expected[i],
 
     RandomUtil.randomSelect(r, weights));
 
  }
 
 }
 
 
 
}