@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 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);
}
@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);
}