用Mockito测试SpringMVC+Hibernate

package com.websystique.springmvc.controller;import static org.mockito.Matchers.any;import static org.mockito.Matchers.anyString;import static org.mockito.Matchers.anyInt;import static org.mockito.Mockito.doNothing;import static org.mockito.Mockito.when;import static org.mockito.Mockito.verify;import java.math.BigDecimal;import java.util.ArrayList;import java.util.List;import org.joda.time.LocalDate;import org.mockito.InjectMocks;import org.mockito.Mock;import org.mockito.MockitoAnnotations;import org.mockito.Spy;import static org.mockito.Mockito.atLeastOnce;import org.springframework.context.MessageSource;import org.springframework.ui.ModelMap;import org.springframework.validation.BindingResult;import org.testng.Assert;import org.testng.annotations.BeforeClass;import org.testng.annotations.Test;import com.websystique.springmvc.model.Employee;import com.websystique.springmvc.service.EmployeeService;public class AppControllerTest {    @Mock    EmployeeService service;        @Mock    MessageSource message;        @InjectMocks    AppController appController;        @Spy    List<Employee> employees = new ArrayList<Employee>();    @Spy    ModelMap model;        @Mock    BindingResult result;        @BeforeClass    public void setUp(){        MockitoAnnotations.initMocks(this);        employees = getEmployeeList();    }        @Test    public void listEmployees(){        when(service.findAllEmployees()).thenReturn(employees);        Assert.assertEquals(appController.listEmployees(model), "allemployees");        Assert.assertEquals(model.get("employees"), employees);        verify(service, atLeastOnce()).findAllEmployees();                }        @Test    public void newEmployee(){        Assert.assertEquals(appController.newEmployee(model), "registration");        Assert.assertNotNull(model.get("employee"));        Assert.assertFalse((Boolean)model.get("edit"));        Assert.assertEquals(((Employee)model.get("employee")).getId(), 0);    }    @Test    public void saveEmployeeWithValidationError(){        when(result.hasErrors()).thenReturn(true);        doNothing().when(service).saveEmployee(any(Employee.class));        Assert.assertEquals(appController.saveEmployee(employees.get(0), result, model), "registration");    }    @Test    public void saveEmployeeWithValidationErrorNonUniqueSSN(){        when(result.hasErrors()).thenReturn(false);        when(service.isEmployeeSsnUnique(anyInt(), anyString())).thenReturn(false);        Assert.assertEquals(appController.saveEmployee(employees.get(0), result, model), "registration");    }        @Test    public void saveEmployeeWithSuccess(){        when(result.hasErrors()).thenReturn(false);        when(service.isEmployeeSsnUnique(anyInt(), anyString())).thenReturn(true);        doNothing().when(service).saveEmployee(any(Employee.class));        Assert.assertEquals(appController.saveEmployee(employees.get(0), result, model), "success");        Assert.assertEquals(model.get("success"), "Employee Axel registered successfully");    }    @Test    public void editEmployee(){        Employee emp = employees.get(0);        when(service.findEmployeeBySsn(anyString())).thenReturn(emp);        Assert.assertEquals(appController.editEmployee(anyString(), model), "registration");        Assert.assertNotNull(model.get("employee"));        Assert.assertTrue((Boolean)model.get("edit"));        Assert.assertEquals(((Employee)model.get("employee")).getId(), 1);    }    @Test    public void updateEmployeeWithValidationError(){        when(result.hasErrors()).thenReturn(true);        doNothing().when(service).updateEmployee(any(Employee.class));        Assert.assertEquals(appController.updateEmployee(employees.get(0), result, model,""), "registration");    }    @Test    public void updateEmployeeWithValidationErrorNonUniqueSSN(){        when(result.hasErrors()).thenReturn(false);        when(service.isEmployeeSsnUnique(anyInt(), anyString())).thenReturn(false);        Assert.assertEquals(appController.updateEmployee(employees.get(0), result, model,""), "registration");    }    @Test    public void updateEmployeeWithSuccess(){        when(result.hasErrors()).thenReturn(false);        when(service.isEmployeeSsnUnique(anyInt(), anyString())).thenReturn(true);        doNothing().when(service).updateEmployee(any(Employee.class));        Assert.assertEquals(appController.updateEmployee(employees.get(0), result, model, ""), "success");        Assert.assertEquals(model.get("success"), "Employee Axel updated successfully");    }            @Test    public void deleteEmployee(){        doNothing().when(service).deleteEmployeeBySsn(anyString());        Assert.assertEquals(appController.deleteEmployee("123"), "redirect:/list");    }    public List<Employee> getEmployeeList(){        Employee e1 = new Employee();        e1.setId(1);        e1.setName("Axel");        e1.setJoiningDate(new LocalDate());        e1.setSalary(new BigDecimal(10000));        e1.setSsn("XXX111");                Employee e2 = new Employee();        e2.setId(2);        e2.setName("Jeremy");        e2.setJoiningDate(new LocalDate());        e2.setSalary(new BigDecimal(20000));        e2.setSsn("XXX222");                employees.add(e1);        employees.add(e2);        return employees;    }}

package com.websystique.springmvc.service;import static org.mockito.Matchers.any;import static org.mockito.Matchers.anyString;import static org.mockito.Matchers.anyInt;import static org.mockito.Mockito.atLeastOnce;import static org.mockito.Mockito.doNothing;import static org.mockito.Mockito.verify;import java.math.BigDecimal;import java.util.ArrayList;import java.util.List;import static org.mockito.Mockito.when;import org.joda.time.LocalDate;import org.mockito.InjectMocks;import org.mockito.Mock;import org.mockito.MockitoAnnotations;import org.mockito.Spy;import org.testng.Assert;import org.testng.annotations.BeforeClass;import org.testng.annotations.Test;import com.websystique.springmvc.dao.EmployeeDao;import com.websystique.springmvc.model.Employee;public class EmployeeServiceImplTest {    @Mock    EmployeeDao dao;        @InjectMocks    EmployeeServiceImpl employeeService;        @Spy    List<Employee> employees = new ArrayList<Employee>();        @BeforeClass    public void setUp(){        MockitoAnnotations.initMocks(this);        employees = getEmployeeList();    }    @Test    public void findById(){        Employee emp = employees.get(0);        when(dao.findById(anyInt())).thenReturn(emp);        Assert.assertEquals(employeeService.findById(emp.getId()),emp);    }    @Test    public void saveEmployee(){        doNothing().when(dao).saveEmployee(any(Employee.class));        employeeService.saveEmployee(any(Employee.class));        verify(dao, atLeastOnce()).saveEmployee(any(Employee.class));    }        @Test    public void updateEmployee(){        Employee emp = employees.get(0);        when(dao.findById(anyInt())).thenReturn(emp);        employeeService.updateEmployee(emp);        verify(dao, atLeastOnce()).findById(anyInt());    }    @Test    public void deleteEmployeeBySsn(){        doNothing().when(dao).deleteEmployeeBySsn(anyString());        employeeService.deleteEmployeeBySsn(anyString());        verify(dao, atLeastOnce()).deleteEmployeeBySsn(anyString());    }        @Test    public void findAllEmployees(){        when(dao.findAllEmployees()).thenReturn(employees);        Assert.assertEquals(employeeService.findAllEmployees(), employees);    }        @Test    public void findEmployeeBySsn(){        Employee emp = employees.get(0);        when(dao.findEmployeeBySsn(anyString())).thenReturn(emp);        Assert.assertEquals(employeeService.findEmployeeBySsn(anyString()), emp);    }    @Test    public void isEmployeeSsnUnique(){        Employee emp = employees.get(0);        when(dao.findEmployeeBySsn(anyString())).thenReturn(emp);        Assert.assertEquals(employeeService.isEmployeeSsnUnique(emp.getId(), emp.getSsn()), true);    }            public List<Employee> getEmployeeList(){        Employee e1 = new Employee();        e1.setId(1);        e1.setName("Axel");        e1.setJoiningDate(new LocalDate());        e1.setSalary(new BigDecimal(10000));        e1.setSsn("XXX111");                Employee e2 = new Employee();        e2.setId(2);        e2.setName("Jeremy");        e2.setJoiningDate(new LocalDate());        e2.setSalary(new BigDecimal(20000));        e2.setSsn("XXX222");                employees.add(e1);        employees.add(e2);        return employees;    }    }

package com.websystique.springmvc.configuration;import java.util.Properties;import javax.sql.DataSource;import org.hibernate.SessionFactory;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.ComponentScan;import org.springframework.context.annotation.Configuration;import org.springframework.core.env.Environment;import org.springframework.jdbc.datasource.DriverManagerDataSource;import org.springframework.orm.hibernate4.HibernateTransactionManager;import org.springframework.orm.hibernate4.LocalSessionFactoryBean;import org.springframework.transaction.annotation.EnableTransactionManagement;/* * This class is same as real HibernateConfiguration class in sources. * Only difference is that method dataSource & hibernateProperties  * implementations are specific to Hibernate working with H2 database. */@Configuration@EnableTransactionManagement@ComponentScan({ "com.websystique.springmvc.dao" })public class HibernateTestConfiguration {	@Autowired	private Environment environment;	@Bean	public LocalSessionFactoryBean sessionFactory() {		LocalSessionFactoryBean sessionFactory = new LocalSessionFactoryBean();		sessionFactory.setDataSource(dataSource());		sessionFactory.setPackagesToScan(new String[] { "com.websystique.springmvc.model" });		sessionFactory.setHibernateProperties(hibernateProperties());		return sessionFactory;	}	@Bean(name = "dataSource")	public DataSource dataSource() {		DriverManagerDataSource dataSource = new DriverManagerDataSource();		dataSource.setDriverClassName("org.h2.Driver");		dataSource.setUrl("jdbc:h2:mem:test;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE");		dataSource.setUsername("sa");		dataSource.setPassword("");		return dataSource;	}	private Properties hibernateProperties() {		Properties properties = new Properties();		properties.put("hibernate.dialect", "org.hibernate.dialect.H2Dialect");		properties.put("hibernate.hbm2ddl.auto", "create-drop");		return properties;	}	@Bean	@Autowired	public HibernateTransactionManager transactionManager(SessionFactory s) {		HibernateTransactionManager txManager = new HibernateTransactionManager();		txManager.setSessionFactory(s);		return txManager;	}}

package com.websystique.springmvc.dao;import javax.sql.DataSource;import org.dbunit.database.DatabaseDataSourceConnection;import org.dbunit.database.IDatabaseConnection;import org.dbunit.dataset.IDataSet;import org.dbunit.operation.DatabaseOperation;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.test.context.ContextConfiguration;import org.springframework.test.context.testng.AbstractTransactionalTestNGSpringContextTests;import org.testng.annotations.BeforeMethod;import com.websystique.springmvc.configuration.HibernateTestConfiguration;@ContextConfiguration(classes = { HibernateTestConfiguration.class })public abstract class EntityDaoImplTest extends AbstractTransactionalTestNGSpringContextTests {    @Autowired    DataSource dataSource;    @BeforeMethod    public void setUp() throws Exception {        IDatabaseConnection dbConn = new DatabaseDataSourceConnection(                dataSource);        DatabaseOperation.CLEAN_INSERT.execute(dbConn, getDataSet());    }        protected abstract IDataSet getDataSet() throws Exception;}

package com.websystique.springmvc.dao;import java.math.BigDecimal;import org.dbunit.dataset.IDataSet;import org.dbunit.dataset.xml.FlatXmlDataSet;import org.joda.time.LocalDate;import org.springframework.beans.factory.annotation.Autowired;import org.testng.Assert;import org.testng.annotations.Test;import com.websystique.springmvc.model.Employee;public class EmployeeDaoImplTest extends EntityDaoImplTest{    @Autowired    EmployeeDao employeeDao;    @Override    protected IDataSet getDataSet() throws Exception{        IDataSet dataSet = new FlatXmlDataSet(this.getClass().getClassLoader().getResourceAsStream("Employee.xml"));        return dataSet;    }        /* In case you need multiple datasets (mapping different tables) and you do prefer to keep them in separate XML‘s    @Override    protected IDataSet getDataSet() throws Exception {      IDataSet[] datasets = new IDataSet[] {              new FlatXmlDataSet(this.getClass().getClassLoader().getResourceAsStream("Employee.xml")),              new FlatXmlDataSet(this.getClass().getClassLoader().getResourceAsStream("Benefits.xml")),              new FlatXmlDataSet(this.getClass().getClassLoader().getResourceAsStream("Departements.xml"))      };      return new CompositeDataSet(datasets);    }    */    @Test    public void findById(){        Assert.assertNotNull(employeeDao.findById(1));        Assert.assertNull(employeeDao.findById(3));    }        @Test    public void saveEmployee(){        employeeDao.saveEmployee(getSampleEmployee());        Assert.assertEquals(employeeDao.findAllEmployees().size(), 3);    }        @Test    public void deleteEmployeeBySsn(){        employeeDao.deleteEmployeeBySsn("11111");        Assert.assertEquals(employeeDao.findAllEmployees().size(), 1);    }        @Test    public void deleteEmployeeByInvalidSsn(){        employeeDao.deleteEmployeeBySsn("23423");        Assert.assertEquals(employeeDao.findAllEmployees().size(), 2);    }    @Test    public void findAllEmployees(){        Assert.assertEquals(employeeDao.findAllEmployees().size(), 2);    }        @Test    public void findEmployeeBySsn(){        Assert.assertNotNull(employeeDao.findEmployeeBySsn("11111"));        Assert.assertNull(employeeDao.findEmployeeBySsn("14545"));    }    public Employee getSampleEmployee(){        Employee employee = new Employee();        employee.setName("Karen");        employee.setSsn("12345");        employee.setSalary(new BigDecimal(10980));        employee.setJoiningDate(new LocalDate());        return employee;    }}