<dependency> <groupId>org.mockito</groupId> <artifactId>mockito-all</artifactId> <version>1.8.5</version> <scope>test</scope> </dependency>
然后在程序中直接import static org.mockito.Mockito.*; 即可
List mockList = mock( List.class ); when( mockList .get(0) ).thenReturn( 1 ); assertEquals( "预期返回1", 1, mockList .get( 0 ) );
mockList模拟了List的对象,拥有List的所有方法和属性。when(...).thenReturn(...)是指当执行这个方法的时候,返回指定的值。相当于模拟配置对象的过程,为某些条件给定一个预期的返回值。
这里的List是Java.util.List接口,并不是实现类,你也可以使用实现类,我们可以使用它们作为打桩对象。这里的打桩(Stub)也可以叫存根,就是把所需的测试数据塞进对象中,关注的是输入和输出。这里的when(...).thenReturn(...)就是在定义对象方法和参数(输入),然后在thenReturn()中指定结果(输出),这个过程就是Stub打桩,一旦这个方法被Stub了,就会一直返回这个stub的值。当我们连续两次为同一个方法使用stub的时候,最后的那个stub是有效的。
一旦mock了一个对象之后,mock对象会覆盖掉整个被mock的对象,因此如果没有stub方法,就只能返回默认值。当我们mock一个接口时,很多成员方法只是一个签名,并没有实现,需要我们手动写出这些实现方法。比如说,我们模拟request请求对象,被测试的代码中使用了HttpServletRequest的什么方法,就要写出相应的实现方法:
HttpServletRequest request = mock(HttpServletRequest.class); when(request.getParameter("foo")).thenReturn("boo");
如果我们不通过when().thenReturn()返回预期值,mockito就会默认返回null,也不会报错说这个方法找不到。mock实例默认的会给所有的方法添加基本实现:返回null或者空集合,或者0等基本类型的值。
3.迭代风格
// 第一种方式 when(i.next()).thenReturn("Hello").thenReturn("World"); // 第二种方式 when(i.next()).thenReturn("Hello", "World"); // 第三种方式,都是等价的 when(i.next()).thenReturn("Hello"); when(i.next()).thenReturn("World");
4.测试无返回值的方法
doNothing().when(i).remove(); doNothing().when(obj).notify(); // 或直接 when(obj).notify();
5.抛出异常
when(i.next()).thenThrow(new RuntimeException()); doThrow(new RuntimeException()).when(i).remove();
doNothing().doThrow(new RuntimeException()).when(i).remove(); //第一次调用remove方法什么都不做,第二次调用抛出RuntimeException异常。
6.参数匹配器
@Test public void argumentMatchersTest(){ List<String> mock = mock(List.class); when(mock.get(anyInt())).thenReturn("Hello").thenReturn("World"); String result=mock.get(100)+" "+mock.get(200); verify(mock,times(2)).get(anyInt()); assertEquals("Hello World",result); }
@Test public void argumentMatchersTest(){ Map mapMock = mock(Map.class); when(mapMock.put(anyInt(), anyString())).thenReturn("world"); mapMock.put(1, "hello"); verify(mapMock).put(anyInt(), eq("hello")); }
在最后的验证时如果只输入字符串”hello”是会报错的,必须使用Matchers类内建的eq方法。如果将anyInt()换成1进行验证也需要用eq(1)。
7.验证Verify
之前的when(...).thenReturn(...)属于状态测试,有些时候,测试并不关心返回结果,而是关心方法是否被正确的参数调用过,这时候就应该使用验证方法了。从概念上讲,就是和状态测试不同的“行为测试”了。一旦通过mock对模拟对象打桩,意味着mockito会记录着这个模拟对象调用了什么方法,调用了多少次,最后由用户决定是否需要进行验证,即verify()方法。
mockedList.add("one"); mockedList.add("two"); verify(mockedList).add("one");
verify 内部跟踪了所有的方法调用和参数的调用情况,然后会返回一个结果,说明是否通过。
Map mock = Mockito.mock( Map.class ); when( mock.get( "city" ) ).thenReturn( "广州" ); // 关注参数有否传入 verify(mock).get( Matchers.eq( "city" ) ); // 关注调用的次数 verify(mock, times( 2 ));
// correct verify(mock).someMethod(anyInt(), anyString(), eq("third argument")); // will throw exception verify(mock).someMethod(anyInt(), anyString(), "third argument");
8.Spy
List spy = spy(new LinkedList()); //Impossible: real method is called so spy.get(0) throws IndexOutOfBoundsException (the list is yet empty) when(spy.get(0)).thenReturn("foo"); //You have to use doReturn() for stubbing doReturn("foo").when(spy).get(0);
当调用when(spy.get(0)).thenReturn("foo")时,会调用真实对象的get(0),由于list是空的所以会抛出IndexOutOfBoundsException异常,用doReturn可以避免这种情况的发生,因为它不会去调用get(0)方法。
9.使用ArgumentCaptor(参数捕获器) 捕获方法参数进行验证
在某些场景中,不光要对方法的返回值和调用进行验证,同时需要验证一系列交互后所传入方法的参数,这时我们可以用参数捕获器来捕获传入方法的参数进行验证,看它是否符合我们的要求。
@Test public void argumentCaptorTest() { List mock = mock(List.class); List mock2 = mock(List.class); mock.add("John"); mock2.add("Brian"); mock2.add("Jim"); ArgumentCaptor argument = ArgumentCaptor.forClass(String.class); verify(mock).add(argument.capture()); assertEquals("John", argument.getValue()); verify(mock2, times(2)).add(argument.capture()); assertEquals("Jim", argument.getValue()); assertArrayEquals(new Object[]{"Brian","Jim"},argument.getAllValues().toArray()); }