@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.getAddress().setPhoneNumber("303-555-1212");
context.checking(
new Expectations() {
{
one(userDao).saveUser(with(same(user)));
will(returnValue(user));
}
});
User returned = userManager.saveUser(user);
assertTrue(returned.getAddress().getPhoneNumber().equals("303-555-1212"));
assertTrue(returned.getRoles().size() == 1);
}
@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);
}
// ~ Methods ================================================================
@Before
public void setUp() throws Exception {
userDao = context.mock(UserDao.class);
userManager.setUserDao(userDao);
roleDao = context.mock(RoleDao.class);
roleManager.setRoleDao(roleDao);
}
@Test
public void testUserExistsException() {
// set expectations
final User user = new User("admin");
user.getAddress().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 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);
}