Mockito测试框架中的方法详解

这里写目录标题

  • 第一章、模拟对象
    • 1.1)①mock()方法:
    • 1.2)②spy()方法:
  • 第二章、模拟对象行为
    • 2.1)模拟方法调用
        • ①when()方法
    • 2.2)模拟返回值
        • ②thenReturn(要返回的值)
        • ③doReturn()
    • 2.3)模拟并替换原方法的行为
        • ④thenAnswer()
        • ⑤doAnswer
    • 2.4)部分模拟时是否调用真实方法
        • ⑥thenCallRealMethod()
        • ⑦doCallRealMethod()
    • 2.5)模拟抛出异常
        • ⑧thenThrow()
        • ⑨doThrow()
    • 2.6)模拟构造函数和静态方法
        • ⑩模拟构造函数MockedConstruction
        • ⑩①模拟静态方法:MockedStatic
    • 2.7)其他方法
        • 使void方法什么也不做doNothing()
        • 重置模拟对象状态reset()方法:
  • 第三章、模拟对象行为验证
    • 3.1)验证调用次数
        • ①verify()方法:
        • ②verify(object, Mockito.times(2))方法
    • 3.2)验证是否发生交互
        • ③verifyNoMoreInteractions()方法:
        • ④verifyZeroInteractions()方法:
  • 第四章、Assert结果断言
    • 4.1)验证结果
        • ①验证是否相等:Equals
        • ②验证结果真假
        • ③验证结果是否为null
        • ④验证对象引用
        • ⑤验证是否满足指定条件
    • 4.2)使结果直接失败
        • ⑥使测试用例失败:fail()

第一章、模拟对象

1.1)①mock()方法:

用于创建给定类或接口的模拟对象,创建对象的另一种方式就是直接用@Mock/InjectMocks注解来创建模拟对象,第三章有介绍。

注意:无法mock final类和final方法,静态方法和静态对象,私有方法。

import java.util.List;
import static org.mockito.Mockito.*;

public class MockitoMockMethodExample {

    public void testMultipleMocks() {
        // 使用 Mockito.mock() 方法创建多个模拟对象
        List<String> mockList1 = mock(List.class);
        List<String> mockList2 = mock(List.class);
        
        // 设置第一个模拟对象的预期行为
        when(mockList1.size()).thenReturn(5);
        
        // 设置第二个模拟对象的预期行为
        when(mockList2.size()).thenReturn(10);
        
        // 断言模拟对象的行为符合预期
        assert(mockList1.size() == 5);
        assert(mockList2.size() == 10);
    }
}

1.2)②spy()方法:

用于部分模拟对象,可以对真实对象的特定方法进行覆盖。它保留了真实对象的部分行为。Spy对象既可以模拟方法的返回值,也可以保留方法的实际行为。

示例中:spy() 方法创建了一个 ArrayList 对象的部分模拟,然后使用 when().thenReturn() 方法对 size() 方法进行了模拟。最后,调用部分模拟对象的真实方法时,模拟的方法返回了预期的值。

import static org.mockito.Mockito.*;

// 示例:创建一个部分模拟的对象
List<String> list = new ArrayList<>();
List<String> listSpy = spy(list);

// 示例:模拟部分方法
when(listSpy.size()).thenReturn(10);

// 示例:调用部分模拟对象的真实方法
listSpy.add("one");
System.out.println(listSpy.size()); // 输出 10,因为 size() 方法被模拟为返回 10

第二章、模拟对象行为

2.1)模拟方法调用

①when()方法

用于指定模拟对象的方法调用,并设置相应的操作,例如返回值、异常等。
when():此方法需要一个方法调用作为参数(通常是模拟对象的一个方法)。它会记录下这个方法调用,并允许你接下来定义这个方法调用的行为。

when(service.方法名(参数))

2.2)模拟返回值

②thenReturn(要返回的值)

thenReturn(T value),这个方法需要你想要返回的值作为参数。当 when() 中指定的方法被调用时,模拟对象将返回这个值。

当使用模拟对象(用@Mock注释),选择thenReturn模拟有返回值(非void)的方法

如果使用监视对象(用@Spy注释),when(…) thenReturn(…)在返回指定值之前会进行实际方法调用。就需要处理这个方法可能抛出的异常。所以有异常的方法一般不使用thenReturn

when(service.方法名(参数)).thenReturn(要返回的值)

示例:

public class MockitoExample {
    @Test
    void testWhenThenReturn() {
        // 创建模拟对象
        List mockedList = Mockito.mock(List.class);
        
        // 定义当调用mockedList.get(0)时返回"first"
        Mockito.when(mockedList.get(0)).thenReturn("first");

        // 使用模拟对象
        System.out.println(mockedList.get(0));  // 输出 "first"
        System.out.println(mockedList.get(1));  // 输出 null,因为我们没有定义get(1)的行为
    }
}

③doReturn()

当使用模拟对象(用@Mock注释),选择thenReturn模拟无返回值(void)的方法

如果使用监视对象(用@Spy注释),doReturn(…) when(…)不会真正调用该方法。这意味着即使被调用的真实方法抛出异常,也不会影响测试。所以doReturn() 方法通常用于对部分模拟对象(spy)进行方法模。

doReturn(10).when(listSpy).size();

示例:

import static org.mockito.Mockito.*;

// 示例:创建一个部分模拟的对象
List<String> list = new ArrayList<>();
List<String> listSpy = spy(list);

// 示例:使用doReturn方法模拟部分方法
doReturn(10).when(listSpy).size();

// 示例:调用部分模拟对象的真实方法
listSpy.add("one");
System.out.println(listSpy.size()); // 输出 10,因为 size() 方法被模拟为返回 10

2.3)模拟并替换原方法的行为

④thenAnswer()

如果在方法调用中需要固定的返回值,则应使用thenReturn()。 如果需要执行某些操作或替换原方法的行为,则应使用thenAnswer()

如下示例:在调用getCurrentTime方法时作用是获取当前的日期和时间。我想使用Instant.now().toEpochMilli();方法替换getCurrentTime方法的内部代码逻辑。

则需要先
1、实现接口org.mockito.stubbing.Answer的类的对象。
2、在方法answer(…)内部自定义行为。
3、我们从模拟对象中调用模拟方法getCurrentTime时,实际上是执行answer中的代码逻辑。

import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import java.time.Instant;

public class TimeServiceTest {
    @Test
    public void testGetCurrentTime() {
        // 创建TimeService类的模拟对象
        TimeService timeService = Mockito.mock(TimeService.class);

        // 创建一个 Answer 对象,实现 answer 方法来返回当前系统时间
        Answer<Long> answer = new Answer<Long>() {
            public Long answer(InvocationOnMock invocation) throws Throwable {
                return Instant.now().toEpochMilli();
            }
        };

        // 使用 thenAnswer() 指定模拟对象的方法返回当前系统时间
Mockito.when(timeService.getCurrentTime()).thenAnswer(answer);

        // 调用模拟对象的方法,实际执行的是answer中的代码逻辑
        long currentTime = timeService.getCurrentTime();

        // 验证方法是否被调用
        Mockito.verify(timeService).getCurrentTime();
        // 进行进一步的断言
    }
}

我们可以也使用Java 8 lambda功能来实现answer方法。

public class TimeServiceTest {
    @Test
    public void testGetCurrentTime() {
        // 创建TimeService类的模拟对象
        TimeService timeService = Mockito.mock(TimeService.class);

        // 创建一个 Answer 对象,实现 answer 方法来返回当前系统时间
        Answer<Long> answer = new Answer<Long>() {
            public Long answer(InvocationOnMock invocation) throws Throwable {
                return Instant.now().toEpochMilli();
            }
        };

        // 使用 thenAnswer() 指定模拟对象的方法返回当前系统时间
        Mockito.when(timeService.getCurrentTime()).thenAnswer(answer);

        // 调用模拟对象的方法,实际执行的是answer中的代码逻辑
        long currentTime = timeService.getCurrentTime();

        // 验证方法是否被调用
        Mockito.verify(timeService).getCurrentTime();
        // 进行进一步的断言
    }
}
⑤doAnswer

更适用于对 void 方法进行模拟。你可以使用 doAnswer 来执行额外的操作,比如触发回调、记录日志等,因为 void 方法本身没有返回值。
创建一个 Answer 对象,实现 answer 方法来指定特定的操作或返回值,

示例中:通过Mockito的模拟,我们成功地替换了原来calculate方法的实际行为

doAnswer(answer).when(service).方法(Mockito.anyInt(), Mockito.anyInt());

示例:

// 假设有一个名为Service的类,其中包含一个方法
public class Service {
    public int calculate(int a, int b) {
        // 一些复杂的计算
        return a + b;
    }
}

// 在测试中,使用Mockito来模拟Service类的行为
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

public class ServiceTest {
    @Test
    public void testCalculate() {
        // 创建Service类的模拟对象
        Service service = Mockito.mock(Service.class);
        
        // 创建一个 Answer 对象,实现 answer 方法来指定特定的操作或返回值
        Answer<Integer> answer = new Answer<Integer>() {
            public Integer answer(InvocationOnMock invocation) throws Throwable {
                int a = invocation.getArgument(0);
                int b = invocation.getArgument(1);
                // 在 answer 方法中执行特定的操作或返回特定的值
                return a * b;
            }
        };
        
//doAnswer表示当calculate方法被调用时,执行特定的操作   
Mockito.doAnswer(answer).when(service).calculate(Mockito.anyInt(), Mockito.anyInt());
        
        // 调用模拟对象的方法
        //当调用service.calculate(2, 3)时,实际执行的是2 * 3,结果为6
        int result = service.calculate(2, 3);
        
        // 验证方法是否被调用,并且返回了预期的值
        Mockito.verify(service).calculate(2, 3);
        assertEquals(6, result);
    }
}

2.4)部分模拟时是否调用真实方法

⑥thenCallRealMethod()

用于部分模拟对象(spy)时,当调用指定方法时,实际调用对象的真实方法而不是模拟方法。特别是当需要部分模拟对象并且希望某些方法执行真实逻辑时。

when(service.方法名(参数)).thenCallRealMethod()

示例:

// 假设有一个名为UserService的类
public class UserService {
    public String process(String input) {
        // 实际的方法逻辑
        return "Processed: " + input;
    }
}

// 创建UserService的部分模拟对象
UserService userService = Mockito.spy(UserService.class);

// 当调用process方法时,执行实际的方法逻辑
Mockito.when(userService.process("input")).thenCallRealMethod();

// 调用部分模拟对象的方法
String result = userService.process("input");

// 输出结果,将会是"Processed: input",因为实际的方法逻辑被执行了
System.out.println(result);
⑦doCallRealMethod()

用于部分模拟对象(spy)时,用于调用真实对象的实际方法,而不是模拟方法的行为。这在部分模拟对象(spy)中特别有用,因为它允许部分模拟对象调用其真实的方法,而不是模拟方法的行为。

doCallRealMethod().when(someObject).someMethod();

示例:

// 创建一个部分模拟对象
SomeClass someObject = mock(SomeClass.class);
doCallRealMethod().when(someObject).someMethod();

// 调用真实方法
someObject.someMethod();

2.5)模拟抛出异常

⑧thenThrow()

thenThrow 适用于有出参的方法,thenThrow方法指定了在调用该方法时抛出指定的异常,而 thenThrow() 方法更适合在特定条件下抛出异常。

when(mockedList.get(0)).thenThrow(new RuntimeException())

示例:这里是RuntimeException

when(service.方法名(参数)).thenThrow(new RuntimeException());

示例:

import static org.mockito.Mockito.*;

public class ExampleTest {
    @Test
    public void testMethod() {
        // 创建被mock的对象
        SomeClass mockObj = mock(SomeClass.class);
        
        // 设置当调用mockObj.method时抛出异常
        when(mockObj.method(anyString())).thenThrow(new RuntimeException("Something went wrong"));
        
        // 调用被mock的方法这里会抛出异常
        mockObj.method("test");
    }
}
⑨doThrow()

doThrow 适用于 没有出参的方法,在测试中模拟抛出异常的情况。doThrow() 方法更适合在调用特定方法时强制抛出异常

doThrow(new RuntimeException()).when(mockedList).get(0)

示例:

import static org.mockito.Mockito.*;

// 创建mock对象
List<String> mockedList = mock(List.class);

// 指定当调用mockedList的get()方法时抛出异常
doThrow(new RuntimeException()).when(mockedList).get(0);

// 调用被测方法
mockedList.get(0); // 这里会抛出指定的异常

2.6)模拟构造函数和静态方法

⑩模拟构造函数MockedConstruction

在单元测试中,当需要在测试中模拟某个类的实例时,使用MockedConstruction可以模拟构造函数的行为,如果没有模拟构造函数,那么在使用new关键字创建对象时,会执行实际的构造函数

作用:当需要在测试中模拟第三方库的对象时,可以使用MockedConstruction来模拟构造函数,而不实际调用第三方库的构造函数,从而避免对外部资源的依赖。

import org.junit.jupiter.api.Test;
import org.mockito.MockedConstruction;
import org.mockito.Mockito;

public class ExampleTest {

    @Test
    void testWithMockedConstruction() {
    //模拟构造函数行为
        try (MockedConstruction<MyClass> mocked = Mockito.mockConstruction(MyClass.class)) {
        //模拟完以后再创建对象,此时执行的就是模拟的构造函数,而不是实际的
            MyClass myClass = new MyClass();
            //调用get方法
            Mockito.when(myClass.getName()).thenReturn("mocked value");
            // 在这里使用myClass进行测试
        }
    }
}
⑩①模拟静态方法:MockedStatic

MockedStatic是Mockito框架中的一个类,用于模拟静态方法的行为。通过使用MockedStatic,您可以模拟静态方法的返回值,以及验证静态方法的调用次数。

import org.mockito.MockedStatic;
import org.mockito.Mockito;

public class Example {
    public void exampleMethod() {
    //YourClassWithStaticMethod是包含静态方法的类
try (MockedStatic<YourClassWithStaticMethod> mockedStatic = Mockito.mockStatic
        (YourClassWithStaticMethod.class)) {
        
mockedStatic.when(YourClassWithStaticMethod::staticMethod("param1")).thenReturn("someValue");
            // 调用静态方法
YourClassWithStaticMethod.staticMethod("param1");
            // 验证静态方法是否被调用过
mockedStatic.verify(YourClassWithStaticMethod::staticMethod("param1"));
        }
    }
}

2.7)其他方法

使void方法什么也不做doNothing()

doNothing() 方法的作用是用于设置模拟对象的 void 方法不执行任何操作。因为 mock 对象中,void 函数就是什么都不做,所以该方法更适合 spy 对象。

Mockito.doNothing().when(service).方法名(参数)

示例1:


    static class ExampleClass {
        public void hello() {
            System.out.println("吃过了");
        }
    }
/
public class MockDemo {
    @Test
    public void test() {
        ExampleClass exampleClass = spy(new ExampleClass ());
        ExampleClass .hello();  //打印吃过了
        // 通过doNothein让方法什么都不做
        doNothing().when(exampleClass ).hello();
        exampleClass.hello(); // 什么都不输出
    }
}

示例2:这是复制的国外一个博主的代码

List list = new LinkedList(); 
List spy = spy(list); 
//let's make clear() do nothing 
doNothing().when(spy).clear(); 
spy.add("one"); 
//clear() does nothing, so the list still contains "one" 
spy.clear();
重置模拟对象状态reset()方法:

用于重置模拟对象的状态或者将对象恢复到初始状态

import org.junit.Test;
import org.mockito.Mockito;

// 创建一个模拟对象
List<String> mockedList = Mockito.mock(List.class);

// 对模拟对象进行方法调用
mockedList.add("one");
System.out.println(mockedList.size()); // 输出 1

// 重置模拟对象的状态
Mockito.reset(mockedList);

// 再次对模拟对象进行方法调用
System.out.println(mockedList.size()); // 输出 0,因为模拟对象的状态已被重置

第三章、模拟对象行为验证

3.1)验证调用次数

①verify()方法:

用于验证模拟对象的指定方法是否被调用。可以进一步验证方法的调用次数和参数。

import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import java.util.List;

public class MockitoExample {
    @Test
    void testMock() {
        // 创建模拟对象
        List mockedList = Mockito.mock(List.class);

        // 使用模拟对象
        mockedList.add("one");
        Mockito.verify(mockedList).add("one");

        // 验证模拟对象的某些行为是否发生过
        Mockito.verify(mockedList, Mockito.times(1)).add("one");
    }
}

②verify(object, Mockito.times(2))方法

是一个 Mockito 框架中的验证方法,用于验证某个行为在模拟对象上被调用了几次。times() 方法需要一个整数参数,代表你期望的调用次数。然后,将返回一个 VerificationMode 实例,这个实例可以和 verify() 方法一起使用,来检查某个操作是否被执行了指定次数。

下述代码;Mockito.times(2) 指的是,我们期望 add() 方法在模拟对象 mockedList 上被调用了两次。如果 add() 方法的实际调用次数不匹配我们的期望(例如,它只被调用了一次,或者三次),那么代码会抛出一个 MockitoAssertionError 异常。

public class MockitoExample {
    @Test
    void testTimes() {
        // 创建模拟对象
        List mockedList = Mockito.mock(List.class);

        // 模拟方法调用
        mockedList.add("one");
        mockedList.add("one");

        // 验证该方法是否被调用了两次
        Mockito.verify(mockedList, Mockito.times(2)).add("one");
    }
}

下述代码:不会抛出异常因为add了两次,与预期相符

import java.util.List;
import static org.mockito.Mockito.*;

public class MockitoTimesExample {

    public void testMethodInvocation() {
        // 创建模拟对象
        List<String> mockList = mock(List.class);
        
        // 调用模拟对象的方法
        mockList.add("item1");
        mockList.add("item2");
        
        // 验证方法调用次数
        verify(mockList, times(2)).add(anyString());
    }
}

3.2)验证是否发生交互

③verifyNoMoreInteractions()方法:

verifyNoMoreInteractions() 方法用于验证mock对象在特定交互之后是否没有发生任何其他交互。它确保在测试中,mock对象在预期的交互之后没有进行多余交互。

多余交互指的是对于被mock的对象,测试代码中发生了未被预期的方法调用。这可能是因为测试代码中的某些逻辑导致了额外的方法调用。

import org.junit.Test;
import org.mockito.Mockito;

public class ProcessorTest {
  @Test
  public void processTest() {
      // 创建 MyService 的 mock 对象
      MyService myMockService = Mockito.mock(MyService.class);
      
      // 创建 MyProcessor 对象,并传入 mock 对象
      MyProcessor myProcessor = new MyProcessor(myMockService);
      
      // 调用 MyProcessor 的 process() 和 process2() 方法
      myProcessor.process();
      myProcessor.process2();

      // 验证 mock 对象的交互
      Mockito.verify(myMockService).doSomething();
      Mockito.verify(myMockService).doSomething2();

      // 确保没有更多的交互发生
      Mockito.verifyNoMoreInteractions(myMockService);
  }
}
④verifyZeroInteractions()方法:

用于验证模拟对象上是否没有发生任何交互。它确保在测试中,mock对象没有与任何其他对象进行交互。关注的是整个测试过程中是否有任何交互。

import static org.mockito.Mockito.*;

// 创建mock对象
List<String> mockedList = mock(List.class);

// 调用被测方法
mockedList.clear();

// 验证是否发生了交互
verify(mockedList).clear();
verifyNoMoreInteractions(mockedList); // 确保没有其他交互

第四章、Assert结果断言

4.1)验证结果

①验证是否相等:Equals

assertEquals(expected, actual):验证两个对象是否相等。
assertNotEquals(unexpected, actual):验证两个对象是否不相等。

示例:验证 Calculator.add(3, 5) 的结果是否等于 8。如果不相等,测试用例将会失败,并输出指定的失败信息。

import static org.junit.jupiter.api.Assertions.assertEquals;

@Test
void testAddition() {
    int result = Calculator.add(3, 5);
    assertEquals(8, result, "The addition result should be 8");
}

assertArrayEquals(expectedArray, resultArray):验证两个数组是否相等。
assertArrayEquals(expecteds, actuals, delta):验证两个浮点数数组是否相等,可以指定误差范围。

示例:验证 expectedArray 和 resultArray 是否相等。如果两个数组不相等,测试用例将会失败,并输出指定的失败信息。

import static org.junit.jupiter.api.Assertions.assertArrayEquals;

@Test
void testArrayEquality() {
    int[] expectedArray = {1, 2, 3};
    int[] resultArray = {1, 2, 3};
    assertArrayEquals(expectedArray, resultArray, "The arrays should be equal");
}
②验证结果真假

assertTrue(): 期待结果为true
assertFalse(): 期待结果为false

示例:验证 number 是否大于 0。如果条件不满足,测试用例将会失败,并输出指定的失败信息。

import static org.junit.jupiter.api.Assertions.assertTrue;

@Test
void testIsPositive() {
    int number = 10;
    assertTrue(number > 0, "The number should be positive");
}
③验证结果是否为null

assertNull(object):期待结果为空。
assertNotNull(object): 期待结果为非空

示例:验证 obj 是否为空。如果对象不为空,测试用例将会失败,并输出指定的失败信息。

import static org.junit.jupiter.api.Assertions.assertNull;

@Test
void testObjectNull() {
    Object obj = null;
    assertNull(obj, "对象应为空");
}
④验证对象引用

assertSame(object1, object2):验证两个对象引用是否指向同一个对象。
assertNotSame(object1, object2):验证两个对象引用是否不指向同一个对象。

示例:验证 obj1 和 obj2 是否引用同一个对象。如果两个对象引用不同的对象,测试用例将会失败,并输出指定的失败信息。

import static org.junit.jupiter.api.Assertions.assertSame;

@Test
void testObjectReference() {
    Object obj1 = new Object();
    Object obj2 = obj1;
    assertSame(obj1, obj2, "两个对象应该引用同一个对象");
}
⑤验证是否满足指定条件

assertThat(actual, matcher):使用Matcher对象验证实际值是否满足指定条件。
assertThat 方法接受两个参数:实际值和Matcher对象。Matcher对象定义了对实际值的特定条件,例如包含特定子串、大于某个值等。通过使用Matcher对象,可以实现更加灵活和具体的断言验证。
示例1:

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.*;

// 示例:验证字符串是否包含指定子串
String actual = "Hello World";
Matcher<String> containsString = containsString("Hello");
assertThat(actual, containsString);

示例2:


// 示例:验证数字是否大于指定值
int actualNumber = 10;
Matcher<Integer> greaterThan = greaterThan(5);
assertThat(actualNumber, greaterThan);

4.2)使结果直接失败

⑥使测试用例失败:fail()

直接使测试用例失败,可用于标记测试用例未通过的情况。

import static org.junit.jupiter.api.Assertions.fail;

@Test
void testSomething() {
    // 执行一些测试逻辑
    // 如果满足特定条件,则强制测试失败
    if (conditionIsMet) {
        fail("Test 测试 失败 because xxxx");
    }
}

你可能感兴趣的:(Java基础知识汇总,spring框架汇总,Mock)