package com.vst.dao;
import com.vst.Constants;
import com.vst.model.Role;
import com.vst.model.User;
import com.vst.model.ext.Address;
import com.vst.model.ext.GeoCity;
import com.vst.model.ext.GeoCountry;
import com.vst.model.ext.GeoProvince;
import org.springframework.dao.DataAccessException;
import org.springframework.dao.DataIntegrityViolationException;
public class UserDaoTest extends BaseDaoTestCase {
private UserDao dao = null;
private RoleDao rdao = null;
public void setUserDao(UserDao dao) {
this.dao = dao;
}
public void setRoleDao(RoleDao rdao) {
this.rdao = rdao;
}
public void testGetUserInvalid() throws Exception {
try {
dao.getUser(new Integer(1000));
fail("'badusername' found in database, failing test...");
} catch (DataAccessException d) {
assertTrue(d != null);
}
}
public void testGetUser() throws Exception {
User user = dao.getUser(new Integer(1));
assertNotNull(user);
assertEquals(1, user.getRoles().size());
assertTrue(user.isEnabled());
}
public void testUpdateUser() throws Exception {
User user = dao.getUser(new Integer(1));
Address address = user.getAddress();
address.setStreet("new address");
dao.saveUser(user);
assertEquals(user.getAddress(), address);
assertEquals("new address", user.getAddress().getStreet());
// verify that violation occurs when adding new user with same username
user.setId(null);
endTransaction();
try {
dao.saveUser(user);
fail("saveUser didn't throw DataIntegrityViolationException");
} catch (DataIntegrityViolationException e) {
assertNotNull(e);
log.debug("expected exception: " + e.getMessage());
}
}
public void testAddUserRole() throws Exception {
User user = dao.getUser(new Integer(1));
assertEquals(1, user.getRoles().size());
Role role = rdao.getRoleByName(Constants.ADMIN_ROLE);
user.addRole(role);
dao.saveUser(user);
assertEquals(2, user.getRoles().size());
//add the same role twice - should result in no additional role
user.addRole(role);
dao.saveUser(user);
assertEquals("more than 2 roles", 2, user.getRoles().size());
user.getRoles().remove(role);
dao.saveUser(user);
assertEquals(1, user.getRoles().size());
}
public void testAddAndRemoveUser() throws Exception {
User user = new User("testuser");
user.setPassword("testpass");
user.setFirstName("Test");
user.setLastName("Last");
Address address = new Address();
address.setCity(new GeoCity("Denver"));
address.setProvince(new GeoProvince("CO"));
address.setCountry(new GeoCountry("USA"));
address.setPostalCode("80210");
user.setAddress(address);
user.setEmail("testuser@appfuse.org");
user.setWebsite("http://raibledesigns.com");
Role role = rdao.getRoleByName(Constants.USER_ROLE);
assertNotNull(role.getId());
user.addRole(role);
dao.saveUser(user);
assertNotNull(user.getId());
assertEquals("testpass", user.getPassword());
dao.removeUser(user.getId());
try {
user = dao.getUser(user.getId());
fail("getUser didn't throw DataAccessException");
} catch (DataAccessException d) {
assertNotNull(d);
}
}
}