Mockito 使用

1. 算术测试类

 package com.smart.test.mockito;

 public interface Calculator {
     public int add(int a, int b);
 }

 package com.smart.test.mockito;

 public class CalculatorImpl implements Calculator {

     @Override
     public int add(int a, int b) {
         System.out.println("add(" + a + ", " + b + ") = " + (a + b));
         return a + b;
     }

     public int minus(int a, int b) {
         System.out.println("minus(" + a + ", " + b + ") = " + (a - b));
         return a - b;
     }
 }

2. 用户测试类

 package com.smart.test.mockito;

 public class User {

     private String name;

     private String password;

     public User() {}

     public User(String name, String password) {
         this.name = name;
         this.password = password;
     }

     public String getName() {
         return name;
     }

     public void setName(String name) {
         this.name = name;
     }

     public String getPassword() {
         return password;
     }

     public void setPassword(String password) {
         this.password = password;
     }
 }

 package com.smart.test.mockito;

 public interface UserService {

     public User findUserByUserName(String name);

     public boolean hasMatchUser(String name, String password);

     public void registerUser(User user);
 }

 package com.smart.test.mockito;

 public class UserServiceImpl implements UserService {

     @Override
     public User findUserByUserName(String name) {
         // TODO Auto-generated method stub
         return null;
     }

     @Override
     public boolean hasMatchUser(String name, String password) {
         // TODO Auto-generated method stub
         return false;
     }

     @Override
     public void registerUser(User user) {
         // TODO Auto-generated method stub

     }
 }

3. 测试驱动类

 package com.smart.test.mockito;

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

 public class MockitoSampleTest {

     CalculatorImpl mockCalculatorImpl = mock(CalculatorImpl.class);

     @Mock
     UserService mockUserService;
     @Mock
     UserServiceImpl mockUserServiceImpl;
     @Mock
     User mockUser;

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

     @Test
     public void testMockCalculator() {

         when(mockCalculatorImpl.add(3, 4)).thenReturn(6);
         int c = mockCalculatorImpl.add(3, 4);
         assertEquals(c, 7);

         doReturn(2).when(mockCalculatorImpl).minus(3, 1);
         int d = mockCalculatorImpl.minus(3, 1);
         assertEquals(d, 2);
     }

     @Test
     public void testMockUserService() {
         /*
          *  对 public User findUserByUserName(String name) 方法设定返回值
          *  模拟入参“tom”, 返回一个new User("tom", "1234")对象
          */
         when(mockUserService.findUserByUserName("tom")).thenReturn(new User("tom", "1234"));

         /*
          *  对 public boolean hasMatchUser(String name, String password) 方法设定返回值
          *  模拟入参hasMatchUser("asn", "1234") 返回 true
          */
         doReturn(true).when(mockUserService).hasMatchUser("asn", "1234");

         /*
          *  对 public void registerUser(User user) 方法设定返回值
          *  模拟入参 new User("John", "1234")能够顺利执行，返回值类型为void
          */
         doNothing().when(mockUserService).registerUser(new User("John", "1234"));

         //调用测试目标方法
         User user = mockUserService.findUserByUserName("tom");
         boolean isMatch = mockUserService.hasMatchUser("asn", "1234");
         mockUserService.registerUser(new User("John", "1234"));

         //断言结果
         System.out.println(user.getName() + ", " + user.getPassword());
         System.out.println(isMatch);
     }

     @Test
     public void testMockUser() {

         /*
          * 对。。。方法进行模拟测试
          *     public String getName() {
                 return name;
             }
          * 设定模拟对象mockUser上getName()方法的返回值为“tom”
          */
         when(mockUser.getName()).thenReturn("tom");

         assertEquals(mockUser.getName(), "tom");
     }
 }