@Test public void should_find_users_when_a_department_is_provided() { UserSelectedCriteria userSelectedCriteria = new UserSelectedCriteria(); UserDepartment userDepartment = UserDepartmentObjectMother.createUserDepartment(1); userSelectedCriteria.setDepartments(Lists.newArrayList(userDepartment)); User matchingUser = UserObjectMother.createUser(); matchingUser.setUserDepartment(userDepartment); User nonMatchingUser = UserObjectMother.createUser(); UserDepartment wrongDepartment = UserDepartmentObjectMother.createUserDepartment(2); wrongDepartment.setCode("cc"); nonMatchingUser.setUserDepartment(wrongDepartment); when(userDao.findUsers(true)).thenReturn(Lists.newArrayList(matchingUser, nonMatchingUser)); UsersAndProjects usersAndProjects = reportCriteriaService.criteriaToUsersAndProjects(userSelectedCriteria); assertEquals(1, usersAndProjects.getUsers().size()); assertThat(usersAndProjects.getUsers(), contains(matchingUser)); assertTrue("Projects should be empty", usersAndProjects.getProjects().isEmpty()); }
@Test public void shouldPersist() { UserDepartment org = UserDepartmentObjectMother.createUserDepartment(); User user = UserObjectMother.createUser(); user.setUserId(5); ProjectAssignment assignment = new ProjectAssignment(); assignment.setUser(user); assignment.setAssignmentId(1); assignment.setProject(new Project(1)); assignment.setAssignmentType(new ProjectAssignmentType(EhourConstants.ASSIGNMENT_DATE)); Set<ProjectAssignment> assignments = Sets.newHashSet(assignment); user.setUserDepartment(org); user.setProjectAssignments(assignments); userDao.persist(user); assertNotNull(user.getUserId()); }