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

来源:互联网 发布:ubuntu安装类型为空 编辑:程序博客网 时间:2024/05/21 06:01
  1. 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");
    }
    }
原创粉丝点击