JavaMockito教程：如何使用Mockito進行單元測試

一、Mockito簡介

Mockito是一個用於Java的開源測試框架，旨在提供一個簡單易用的API來進行對象的模擬和驗證。它允許開發人員在測試過程中創建虛擬對象，以便更好地對代碼進行單元測試。Mockito廣泛用於單元測試，特別是在使用依賴注入(DI)的項目中。

二、Mockito使用方法

使用Mockito進行單元測試有三個基本步驟：

1. 創建mock對象

在測試類中，使用@Mock注釋創建mock對象。例如：

public class MyTest {
  @Mock
  private MyService myService;
  
  // ...
}

這將創建一個MyService類的mock對象。

2. 初始化mock對象

在測試方法之前，使用@Before註解初始化mock對象。例如：

public class MyTest {
  @Mock
  private MyService myService;
  
  @Before
  public void setUp() {
    MockitoAnnotations.initMocks(this);
  }
  
  // ...
}

這將初始化mock對象並使其可用於測試。

3. 測試代碼

使用mock對象進行單元測試代碼，例如：

public class MyTest {
  @Mock
  private MyService myService;
  
  @Before
  public void setUp() {
    MockitoAnnotations.initMocks(this);
  }
  
  @Test
  public void testSomething() {
    when(myService.doSomething()).thenReturn("myString");
    // test code using myService
  }
}

這將測試我們的代碼是否正確使用mock對象。

三、Mockito常見使用場景

以下是Mockito常見的使用場景：

1. 模擬返回值

使用Mockito的when-thenReturn方法，可以模擬對象的返回值。例如：

public class MyTest {
  @Mock
  private MyService myService;
  
  @Before
  public void setUp() {
    MockitoAnnotations.initMocks(this);
  }
  
  @Test
  public void testSomething() {
    when(myService.doSomething()).thenReturn("myString");
    // test code using myService
  }
}

2. 驗證方法調用

使用Mockito的verify方法，可以驗證mock對象的方法是否被正確調用。例如：

public class MyTest {
  @Mock
  private MyService myService;
  
  @Before
  public void setUp() {
    MockitoAnnotations.initMocks(this);
  }
  
  @Test
  public void testSomething() {
    // test code using myService
    verify(myService).doSomething();
  }
}

3. 拋出異常

使用Mockito的when-thenThrow方法，可以模擬對象引發的異常。例如：

public class MyTest {
  @Mock
  private MyService myService;
  
  @Before
  public void setUp() {
    MockitoAnnotations.initMocks(this);
  }
  
  @Test(expected = MyException.class)
  public void testSomething() {
    when(myService.doSomething()).thenThrow(new MyException());
    // test code using myService
  }
}

四、總結

本教程介紹了Mockito的基本用法和常見的使用場景。通過使用Mockito，可以更輕鬆地進行Java單元測試並提高代碼質量。