@Test
public void testSaveUser() throws Exception {
final User testData = new User("1");
testData.getRoles().add(new Role("user"));
// set expected behavior on dao
context.checking(
new Expectations() {
{
one(userDao).get(with(equal(1L)));
will(returnValue(testData));
}
});
final User user = userManager.getUser("1");
user.setPhoneNumber("303-555-1212");
context.checking(
new Expectations() {
{
one(userDao).saveUser(with(same(user)));
will(returnValue(user));
}
});
User returned = userManager.saveUser(user);
assertTrue(returned.getPhoneNumber().equals("303-555-1212"));
assertTrue(returned.getRoles().size() == 1);
}
@Test
public void testAddAndRemoveUser() throws Exception {
// given
User user = new User();
// call populate method in super class to populate test data
// from a properties file matching this class name
user = (User) populate(user);
Role role = new Role(Constants.USER_ROLE);
user.addRole(role);
final User user1 = user;
given(userDao.saveUser(user1)).willReturn(user1);
// when
user = userManager.saveUser(user);
// then
assertTrue(user.getUsername().equals("john"));
assertTrue(user.getRoles().size() == 1);
// given
willDoNothing().given(userDao).remove(5L);
userManager.removeUser("5");
given(userDao.get(5L)).willReturn(null);
// when
user = userManager.getUser("5");
// then
assertNull(user);
verify(userDao).remove(5L);
}
@Test
public void testAddAndRemoveUser() throws Exception {
User user = new User();
// call populate method in super class to populate test data
// from a properties file matching this class name
user = (User) populate(user);
// set expected behavior on role dao
context.checking(
new Expectations() {
{
one(roleDao).getRoleByName(with(equal("ROLE_USER")));
will(returnValue(new Role("ROLE_USER")));
}
});
Role role = roleManager.getRole(Constants.USER_ROLE);
user.addRole(role);
// set expected behavior on user dao
final User user1 = user;
context.checking(
new Expectations() {
{
one(userDao).saveUser(with(same(user1)));
will(returnValue(user1));
}
});
user = userManager.saveUser(user);
assertTrue(user.getUsername().equals("john"));
assertTrue(user.getRoles().size() == 1);
context.checking(
new Expectations() {
{
one(userDao).remove(with(equal(5L)));
}
});
userManager.removeUser("5");
context.checking(
new Expectations() {
{
one(userDao).get(with(equal(5L)));
will(returnValue(null));
}
});
user = userManager.getUser("5");
assertNull(user);
}
@Test
public void testGetUser() {
expect(dao.load(1)).andReturn(fred);
replay(dao);
assertSame(fred, manager.getUser(1));
verify(dao);
}
// ~ Methods ================================================================
@Before
public void setUp() throws Exception {
userDao = context.mock(UserDao.class);
userManager.setUserDao(userDao);
roleDao = context.mock(RoleDao.class);
roleManager = new RoleManagerImpl(roleDao);
}
@Test
public void testUserExistsException() {
// set expectations
final User user = new User("admin");
user.setEmail("*****@*****.**");
final Exception ex = new DataIntegrityViolationException("");
context.checking(
new Expectations() {
{
one(userDao).saveUser(with(same(user)));
will(throwException(ex));
}
});
// run test
try {
userManager.saveUser(user);
fail("Expected UserExistsException not thrown");
} catch (UserExistsException e) {
log.debug("expected exception: " + e.getMessage());
assertNotNull(e);
}
}
@Test
public void testSaveUser() throws Exception {
// given
final User testData = new User("1");
testData.getRoles().add(new Role("user"));
given(userDao.get(1L)).willReturn(testData);
final User user = userManager.getUser("1");
user.setPhoneNumber("303-555-1212");
given(userDao.saveUser(user)).willReturn(user);
// when
User returned = userManager.saveUser(user);
// then
assertTrue(returned.getPhoneNumber().equals("303-555-1212"));
assertTrue(returned.getRoles().size() == 1);
}
@Test
public void testGetUser() throws Exception {
// given
final User testData = new User("1");
testData.getRoles().add(new Role("user"));
given(userDao.get(1L)).willReturn(testData);
// then
User user = userManager.getUser("1");
// then
assertTrue(user != null);
assert user != null;
assertTrue(user.getRoles().size() == 1);
}
@Test
public void testUserExistsException() {
// set expectations
final User user = new User("admin");
user.setEmail("*****@*****.**");
willThrow(new DataIntegrityViolationException("")).given(userDao).saveUser(user);
// run test
try {
userManager.saveUser(user);
fail("Expected UserExistsException not thrown");
} catch (UserExistsException e) {
log.debug("expected exception: " + e.getMessage());
assertNotNull(e);
}
}
@Test
public void testGetUser() throws Exception {
final User testData = new User("1");
testData.getRoles().add(new Role("user"));
// set expected behavior on dao
context.checking(
new Expectations() {
{
one(userDao).get(with(equal(1L)));
will(returnValue(testData));
}
});
User user = userManager.getUser("1");
assertTrue(user != null);
assert user != null;
assertTrue(user.getRoles().size() == 1);
}
