doThrow() doAnswer() doNothing() 和 doReturn() 在mockito中的用法

Java

Mockito中的doThrow()、doAnswer()、doNothing()和doReturn()的用法

Mockito是一个用于Java的开源测试框架，用于模拟（mock）对象，以便在单元测试中进行更好的控制和验证。在Mockito中，有几个方法可以对模拟对象的行为进行定制，包括doThrow()、doAnswer()、doNothing()和doReturn()。本文将介绍这些方法的用法，并通过案例代码进行说明。

1. doThrow()

doThrow()方法用于模拟在调用模拟对象的方法时抛出异常。当我们希望测试某个方法在特定条件下抛出异常时，可以使用doThrow()方法来模拟该异常的抛出。

案例代码如下：

Java
@Test(expected = RuntimeException.class)
public void testDoThrow() {
    List<String> mockList = Mockito.mock(List.class);
    Mockito.doThrow(new RuntimeException()).when(mockList).clear();
    mockList.clear();
}

在上面的例子中，我们创建了一个模拟的List对象，并使用doThrow()方法模拟了调用clear()方法时抛出RuntimeException异常。在测试方法中，我们期望调用clear()方法时会抛出RuntimeException异常，因此使用了@Test注解的expected参数来验证异常的抛出。

2. doAnswer()

doAnswer()方法用于模拟在调用模拟对象的方法时执行自定义的操作。当我们希望测试某个方法在特定条件下执行一些自定义的操作时，可以使用doAnswer()方法来模拟这些操作。

案例代码如下：

Java
@Test
public void testDoAnswer() {
    List<String> mockList = Mockito.mock(List.class);
    Mockito.doAnswer(invocation -> {
        Object[] args = invocation.getArguments();
        System.out.println("Clearing list with args: " + Arrays.toString(args));
        return null;
    }).when(mockList).clear();
    
    mockList.clear();
}

在上面的例子中，我们创建了一个模拟的List对象，并使用doAnswer()方法模拟了调用clear()方法时执行的自定义操作。在doAnswer()方法的lambda表达式中，我们可以获取被调用方法的参数，并打印出来。在测试方法中，我们调用了clear()方法，会触发模拟的自定义操作，输出相应的信息。

3. doNothing()

doNothing()方法用于模拟在调用模拟对象的方法时不执行任何操作。当我们希望测试某个方法在特定条件下不执行任何操作时，可以使用doNothing()方法来模拟这种情况。

案例代码如下：

Java
@Test
public void testDoNothing() {
    List<String> mockList = Mockito.mock(List.class);
    Mockito.doNothing().when(mockList).clear();
    
    mockList.clear();
}

在上面的例子中，我们创建了一个模拟的List对象，并使用doNothing()方法模拟了调用clear()方法时不执行任何操作。在测试方法中，我们调用了clear()方法，但是由于模拟了doNothing()，所以不会执行任何实际操作。

4. doReturn()

doReturn()方法用于模拟在调用模拟对象的方法时返回指定的值。当我们希望测试某个方法在特定条件下返回指定的值时，可以使用doReturn()方法来模拟这种情况。

案例代码如下：

Java
@Test
public void testDoReturn() {
    List<String> mockList = Mockito.mock(List.class);
    Mockito.doReturn(10).when(mockList).size();
    
    int size = mockList.size();
    assertEquals(10, size);
}

在上面的例子中，我们创建了一个模拟的List对象，并使用doReturn()方法模拟了调用size()方法时返回10。在测试方法中，我们调用了size()方法，并使用assertEquals()方法验证返回的值是否为10。

在本文中，我们介绍了Mockito中的doThrow()、doAnswer()、doNothing()和doReturn()方法的用法，并通过案例代码进行了说明。这些方法可以帮助我们更好地对模拟对象的行为进行定制，从而进行更精确的单元测试。无论是模拟异常的抛出、自定义操作的执行、不执行任何操作还是返回指定的值，Mockito都提供了相应的方法来满足我们的需求。