@Test public void testUpdate() throws Exception { Department preUpdatedDepartment = departmentDAO.get(did); Employee employee3 = new Employee("UPDATE", "UPDATE", "XXX-XXX-XXX", 40); preUpdatedDepartment.setDepartmentName("UPDATE"); preUpdatedDepartment.getEmployees().add(employee3); departmentDAO.saveOrUpdate(preUpdatedDepartment); employee3.setDepartment(department); Department updatedDepartment = departmentDAO.get(did); eid3 = (Long) employeeDAO.saveOrUpdate(employee3); Employee updatedEmployee = employeeDAO.get(eid3); assertEquals( "Add method failed: wrong department name", preUpdatedDepartment.getDepartmentName(), updatedDepartment.getDepartmentName()); assertEquals( "Add method failed: wrong employees", preUpdatedDepartment.getEmployees().size(), updatedDepartment.getEmployees().size()); assertEquals( "Add method failed: wrong department", employee3.getDepartment(), updatedEmployee.getDepartment()); }
@Before public void setUp() throws Exception { department = new Department("TEST"); did = (Long) departmentDAO.saveOrUpdate(department); employee1 = new Employee("TEST", "TEST", "xxx-xxx-xxx", 30); employee2 = new Employee("TEST", "TEST", "xxx-xxx-xxx", 35); employee1.setDepartment(department); employee2.setDepartment(department); eid1 = (Long) employeeDAO.saveOrUpdate(employee1); eid2 = (Long) employeeDAO.saveOrUpdate(employee2); }
@Test public void testAdd() throws Exception { department.setDepartmentId(did); Employee addedEmployee1 = employeeDAO.get(eid1); Employee addedEmployee2 = employeeDAO.get(eid2); Department addedDepartment = departmentDAO.get(did); assertEquals( "Add method failed: wrong department name", addedDepartment.getDepartmentName(), department.getDepartmentName()); assertEquals( "Add method failed: wrong employees", addedDepartment.getEmployees().size(), department.getEmployees().size()); assertEquals( "Add method failed: wrong department", addedEmployee1.getDepartment(), employee1.getDepartment()); assertEquals( "Add method failed: wrong department", addedEmployee2.getDepartment(), employee2.getDepartment()); }