@Test
public void testRemovePrivilege() throws Exception {

  Set<PrivilegeEntity> privileges = new HashSet<PrivilegeEntity>();
  PrivilegeEntity privilegeEntity1 = new PrivilegeEntity();
  privilegeEntity1.setId(1);
  privileges.add(privilegeEntity1);
  PrivilegeEntity privilegeEntity2 = new PrivilegeEntity();
  privilegeEntity2.setId(2);
  privileges.add(privilegeEntity2);
  PrincipalEntity entity = new PrincipalEntity();
  entity.setPrivileges(privileges);
  entity.removePrivilege(privilegeEntity2);
  privileges = entity.getPrivileges();
  Assert.assertEquals(1, privileges.size());
  privileges.contains(privilegeEntity1);
}