摘要：例子使用源代碼我們先給了一個的實現然后又規定了方法的返回值。源代碼也就是說，得益于，我們能夠很方便地對依賴關系中任意層級的任意做。

Github地址

Mock測試技術能夠避免你為了測試一個方法，卻需要自行構建整個依賴關系的工作，并且能夠讓你專注于當前被測試對象的邏輯，而不是其依賴的其他對象的邏輯。

舉例來說，比如你需要測試Foo.methodA，而這個方法依賴了Bar.methodB，又傳遞依賴到了Zoo.methodC，于是它們的依賴關系就是Foo->Bar->Zoo，所以在測試代碼里你必須自行new Bar和Zoo。

有人會說："我直接用Spring的DI機制不就行了嗎？"的確，你可以用Spring的DI機制，不過解決不了測試代碼耦合度過高的問題：

因為Foo方法內部調用了Bar和Zoo的方法，所以你對其做單元測試的時候，必須完全了解Bar和Zoo方法的內部邏輯，并且謹慎的傳參和assert結果，一旦Bar和Zoo的代碼修改了，你的Foo測試代碼很可能就會運行失敗。

所以這個時候我們需要一種機制，能過讓我們在測試Foo的時候不依賴于Bar和Zoo的具體實現，即不關心其內部邏輯，只關注Foo內部的邏輯，從而將Foo的每個邏輯分支都測試到。

所以業界就產生了Mock技術，它可以讓我們做一個假的Bar（不需要Zoo，因為只有真的Bar才需要Zoo），然后控制這個假的Bar的行為（讓它返回什么就返回什么），以此來測試Foo的每個邏輯分支。

你肯定會問，這樣的測試有意義嗎？在真實環境里Foo用的是真的Bar而不是假的Bar，你用假的Bar測試成功能代表真實環境不出問題？

其實假Bar代表的是一個行為正確的Bar，用它來測試就能驗證"在Bar行為正確的情況下Foo的行為是否正確"，而真Bar的行為是否正確會由它自己的測試代碼來驗證。

Mock技術的另一個好處是能夠讓你盡量避免集成測試，比如我們可以Mock一個Repository（數據庫操作類），讓我們盡量多寫單元測試，提高測試代碼執行效率。

spring-boot-starter-test依賴了Mockito，所以我們會在本章里使用Mockito來講解。

先介紹一下接下來要被我們測試的類Foo、Bar倆兄弟。

public interface Foo {

  boolean checkCodeDuplicate(String code);

}

public interface Bar {

  Set getAllCodes();

}

@Component
public class FooImpl implements Foo {

  private Bar bar;

  @Override
  public boolean checkCodeDuplicate(String code) {
    return bar.getAllCodes().contains(code);
  }

  @Autowired
  public void setBar(Bar bar) {
    this.bar = bar;
  }

}

源代碼NoMockTest：

public class NoMockTest {

  @Test
  public void testCheckCodeDuplicate1() throws Exception {

    FooImpl foo = new FooImpl();
    foo.setBar(new Bar() {
      @Override
      public Set getAllCodes() {
        return Collections.singleton("123");
      }
    });
    assertEquals(foo.checkCodeDuplicate("123"), true);

  }

  @Test
  public void testCheckCodeDuplicate2() throws Exception {

    FooImpl foo = new FooImpl();
    foo.setBar(new FakeBar(Collections.singleton("123")));
    assertEquals(foo.checkCodeDuplicate("123"), true);

  }

  public class FakeBar implements Bar {

    private final Set codes;

    public FakeBar(Set codes) {
      this.codes = codes;
    }

    @Override
    public Set getAllCodes() {
      return codes;
    }

  }

}

這個測試代碼里用到了兩種方法來做假的Bar：

匿名內部類

做了一個FakeBar

這兩種方式都不是很優雅，看下面使用Mockito的例子。

源代碼MockitoTest：

public class MockitoTest {

  @Mock
  private Bar bar;

  @InjectMocks
  private FooImpl foo;

  @BeforeMethod(alwaysRun = true)
  public void initMock() {
    MockitoAnnotations.initMocks(this);
  }

  @Test
  public void testCheckCodeDuplicate() throws Exception {

    when(bar.getAllCodes()).thenReturn(Collections.singleton("123"));
    assertEquals(foo.checkCodeDuplicate("123"), true);

  }

}

我們先給了一個Bar的Mock實現：@Mock private Bar bar;

然后又規定了getAllCodes方法的返回值：when(bar.getAllCodes()).thenReturn(Collections.singleton("123"))。這樣就把一個假的Bar定義好了。

最后利用Mockito把Bar注入到Foo里面，@InjectMocks private FooImpl foo;、MockitoAnnotations.initMocks(this);

源代碼Spring_1_Test：

@ContextConfiguration(classes = FooImpl.class)
@TestExecutionListeners(listeners = MockitoTestExecutionListener.class)
public class Spring_1_Test extends AbstractTestNGSpringContextTests {

  @MockBean
  private Bar bar;

  @Autowired
  private Foo foo;

  @Test
  public void testCheckCodeDuplicate() throws Exception {

    when(bar.getAllCodes()).thenReturn(Collections.singleton("123"));
    assertEquals(foo.checkCodeDuplicate("123"), true);

  }

}

要注意，如果要啟用Spring和Mockito，必須添加這么一行：@TestExecutionListeners(listeners = MockitoTestExecutionListener.class)。

當Bean存在這種依賴關系當時候：LooImpl -> FooImpl -> Bar，我們應該怎么測試呢？

按照Mock測試的原則，這個時候我們應該mock一個Foo對象，把這個注入到LooImpl對象里，就像例子3里的一樣。

不過如果你不想mock Foo而是想mock Bar的時候，其實做法和前面也差不多，Spring會自動將mock Bar注入到FooImpl中，然后將FooImpl注入到LooImpl中。

源代碼Spring_2_Test:

@ContextConfiguration(classes = { FooImpl.class, LooImpl.class })
@TestExecutionListeners(listeners = MockitoTestExecutionListener.class)
public class Spring_2_Test extends AbstractTestNGSpringContextTests {

  @MockBean
  private Bar bar;

  @Autowired
  private Loo loo;

  @Test
  public void testCheckCodeDuplicate() throws Exception {

    when(bar.getAllCodes()).thenReturn(Collections.singleton("123"));
    assertEquals(loo.checkCodeDuplicate("123"), true);

  }

}

也就是說，得益于Spring Test Framework，我們能夠很方便地對依賴關系中任意層級的任意Bean做mock。

源代碼Boot_1_Test：

@SpringBoot_1_Test(classes = { FooImpl.class })
@TestExecutionListeners(listeners = MockitoTestExecutionListener.class)
public class Boot_1_Test extends AbstractTestNGSpringContextTests {

  @MockBean
  private Bar bar;

  @Autowired
  private Foo foo;

  @Test
  public void testCheckCodeDuplicate() throws Exception {

    when(bar.getAllCodes()).thenReturn(Collections.singleton("123"));
    assertEquals(foo.checkCodeDuplicate("123"), true);

  }

}

Spring Framework Testing

Spring Boot Testing

Mockito