JUnit+Mockito单元测试之打桩when().thenReturn();

た入场券 2024-04-19 12:48 225阅读 0赞

## 什么是Mock 测试 ##

Mock 测试就是在测试过程中，对于某些`不容易构造`（如 HttpServletRequest 必须在Servlet 容器中才能构造出来）或者`不容易获取`的对象（如 JDBC 中的ResultSet 对象,JPA的CRUDRepository,需要执行数据库操作的），用一个虚拟的对象（Mock 对象）来创建（覆盖方法返回）以便测试的测试方法。

*  `JUnit` 是一个单元测试框架。
 *  `Mockito` 是用于数据模拟对象的框架。

## when().thenReturn(); ##

when( mockRepository.getMock("x") ).thenReturn( "1024" );
    String mock= mockRepository.getMock("x");
    assertEquals( "预期x=1024","1024", mock);

when(xxxx).thenReturn(yyyy); 是指定当执行了这个方法的时候，返回 thenReturn 的值，相当于是对模拟对象的配置过程，为某些条件给定一个预期的返回值。

HttpServletRequest request = mock(HttpServletRequest.class);
    when(request.getParameter("csdn")).thenReturn("zhengkai");
    assertEquals( "预期csdn=zhengkai",request.getParameter("csdn"), "zhengkai");

## Stub 打桩 ##

Mockito 中 `when().thenReturn();` 这种语法来定义对象方法和参数（输入），然后在 thenReturn 中指定结果（输出）。此过程称为 `Stub 打桩` 。一旦这个方法被 stub 了，就会一直返回这个 stub 的值。

`!!!`Stub 打桩 需要注意的是:

*  对于 static 和 final 方法， Mockito 无法对其 when(…).thenReturn(…) 操作。
 *  当我们连续两次为同一个方法使用 stub 的时候，他只会只用最新的一次。

## 迭代打桩 ##

打桩支持迭代风格的返回值，第一次调用 i.next() 将返回 ”Hello”，第二次的调用会返回 ”World”。

// 第一种方式 ，都是等价的
    when(i.next()).thenReturn("Hello").thenReturn("World");
    // 第二种方式，都是等价的
    when(i.next()).thenReturn("Hello", "World");
    // 第三种方式，都是等价的
    when(i.next()).thenReturn("Hello"); when(i.next()).thenReturn("World");

## void如何打桩 ##

没有返回值的 void 方法呢？不需要执行，只需要模拟跳过，写法上会有所不同，没返回值了调用 thenReturn(xxx) 肯定不行，取而代之的用 `doNothing().when().notify();`

doNothing().when(obj).notify();
    // 或直接
    when(obj).notify();

## 抛出异常 ##

when(i.next()).thenThrow(new RuntimeException());
    doThrow(new RuntimeException()).when(i).remove(); // void 方法的
    // 迭代风格 
    doNothing().doThrow(new RuntimeException()).when(i).remove(); 
    // 第一次调用 remove 方法什么都不做，第二次调用抛出 RuntimeException 异常。

## Any() ##

`anyString()` 匹配任何 String 参数，`anyInt()` 匹配任何 int 参数，`anySet()` 匹配任何 Set，`any()` 则意味着参数为任意值 `any(User.class)` 匹配任何 User类。

when(mockedList.get(anyInt())).thenReturn("element");   
    System.out.println(mockedList.get(999));// 此时打印是 element
    System.out.println(mockedList.get(777));// 此时打印是 element