`
lijunaccp
  • 浏览: 153342 次
  • 性别: Icon_minigender_1
  • 来自: 杭州
社区版块
存档分类
最新评论

设定Mock对象的期望行为及返回值

阅读更多
package com.lijun.mockito;

import static org.mockito.Mockito.*;

import static org.junit.Assert.*;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

/*******************************
 * 对于final类,匿名类和JAVA基本类型是无法进行Mock的 
 * 对于static和final修饰的方法是无法进行设定的
 * 
 */
public class MockitoSampleTest {

	// 对接口进行模拟
	UserService mockUserService = mock(UserService.class);

	// 对类进行模拟
	UserServiceImpl mockServiceImpl = mock(UserServiceImpl.class);

	@Mock
	User mockUser;

	@Before
	public void initMocks() {
		// 初始化当前测试类所有@Mock注解模拟对象
		MockitoAnnotations.initMocks(this);
	}

	// 模拟接口UserService测试
	@Test
	public void testMockInterface() {
		// 对方法设定返回值
		when(mockUserService.findUserByUserName("tom")).thenReturn(
				new User("tom", "1234"));

		// 对方法设定返回值
		doReturn(true).when(mockUserService).hasMatchUser("tom", "1234");

		// 对void方法进行方法预期设定
		User u = new User("John", "1234");
		doNothing().when(mockUserService).registerUser(u);

		// 执行方法调用
		User user = mockUserService.findUserByUserName("tom");
		boolean isMatch = mockUserService.hasMatchUser("tom", "1234");
		mockUserService.registerUser(u);

		assertNotNull(user);
		assertEquals(user.getUserName(), "tom");
		assertEquals(isMatch, true);

	}

	@Test
	public void testMockClass() {
		// 对方法设定返回值
		when(mockServiceImpl.findUserByUserName("tom")).thenReturn(
				new User("tom", "1234"));
		doReturn(true).when(mockServiceImpl).hasMatchUser("tom", "1234");
		
		User user = mockServiceImpl.findUserByUserName("tom");
		boolean isMatch = mockServiceImpl.hasMatchUser("tom", "1234");
		
		assertNotNull(user);
		assertEquals(user.getUserName(), "tom");
		assertEquals(isMatch, true);
	}
	
	@Test
	public void testMockUser() {
		when(mockUser.getUserName()).thenReturn("tom");
		
		assertEquals(mockUser.getUserName(), "tom");
	}
}
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics