Mockito—返回期望值

import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.Iterator;
import java.util.List;
import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

/**
 * Mockito 示例.<br>
 * @author gongqiang <br>
 * @version 1.0.0 2021年5月6日<br>
 * @see 
 * @since JDK 1.5.0
 */
public class MockitoExpected {

	/**
	 * 模拟期望的结果（多次）.
	 */
    @Test
    public void whenThenReturn(){
        //mock一个Iterator类
        Iterator iterator = Mockito.mock(Iterator.class);
        //预设当iterator调用next()时第一次返回hello，第n次都返回world
        Mockito.when(iterator.next()).thenReturn("hello").thenReturn("world");
        //使用mock的对象
        String result = iterator.next() + " " + iterator.next() + " " + iterator.next();
        //验证结果
        Assert.assertEquals("hello world world",result);
    }

    /**
	 * 模拟期望的结果（一次）.
	 */
	@Test
	public void doReturnWhen() {
		//mock一个Iterator类
		final Iterator<String> iterator = Mockito.mock(Iterator.class);
		Mockito.doReturn("Hello World.").when(iterator).next();
		//验证结果
		Assert.assertEquals("Hello World.", iterator.next());
	}
	
	static class CustomerAnswer implements Answer<String> {
		@Override
		public String answer(InvocationOnMock invocation) throws Throwable {
			Object[] args = invocation.getArguments();
			return "Hello:" + args[0];
		}
		
	}
	
	 /**
	 * 模拟期望的结果（动态构建）.
	 */
	@Test
	public void whenThenAnswer() {
		List mockList = Mockito.mock(List.class);
		Mockito.when(mockList.get(org.mockito.Matchers.anyInt())).thenAnswer(new CustomerAnswer());
		Assert.assertEquals("Hello:0", mockList.get(0));
		Assert.assertEquals("Hello:999", mockList.get(999));
	}
	
    /**
	 * 模拟期望的异常.
	 */
    @Test(expected = IOException.class)
    public void whenThenThrow() throws IOException {
        OutputStream outputStream = Mockito.mock(OutputStream.class);
        OutputStreamWriter writer = new OutputStreamWriter(outputStream);
        //预设当流关闭时抛出异常
        Mockito.doThrow(new IOException()).when(outputStream).close();
        outputStream.close();
    }

	static class Real {
		public int call(int i) {
			return i * i;
		}
		
		public int add(int a, int b) {
			return a + b;
		}
		
		public void append(StringBuilder sb) {
			sb.append("Hello.");
		}
	}
	
	/**
	 * void方法模拟.
	 */
	@Test
	public void whenDoAnswer() {
		Real spyReal = Mockito.spy(Real.class);
		Answer<Object> answer = invocationOnMock -> {
			Object[] args = invocationOnMock.getArguments();
			StringBuilder sb = (StringBuilder)args[0];
			sb.append("ABC");
			return "OK";
		};
		StringBuilder sb = new StringBuilder();
		Mockito.doAnswer(answer).when(spyReal).append(sb);
		
		// 执行
		spyReal.append(sb);
		Assert.assertEquals("ABC", sb.toString());
	}
	
	/**
	 * 不做任何处理.
	 */
	@Test
	public void whenDoNothing() {
		Real spyReal = Mockito.spy(Real.class);
		StringBuilder sb = new StringBuilder();
		Mockito.doNothing().when(spyReal).append(sb);
		// 执行
		spyReal.append(sb);
		Assert.assertEquals("", sb.toString());
	}
}