@Test
public void testAddMembershipsWithPermission() throws Exception {
String name = "testAddMembershipsWithPermission" + r.nextInt();
UserWrapper user = UserHelper.addUser(name, null, true);
MembershipWrapper ms = MembershipHelper.newMembership(user, null, null);
ms.addToPermissionCollection(Arrays.asList(PermissionEnum.CLINIC_CREATE));
user.persist();
user.reload();
Assert.assertEquals(1, user.getMembershipCollection(false).size());
ms = user.getMembershipCollection(false).get(0);
Assert.assertEquals(1, ms.getPermissionCollection().size());
}
@Test
public void testAddMembershipsWithRole() throws Exception {
String name = "addMembershipsWithRole" + r.nextInt();
UserWrapper user = UserHelper.addUser(name, null, true);
RoleWrapper role = RoleHelper.addRole(name, true);
MembershipWrapper ms = MembershipHelper.newMembership(user, null, null);
ms.addToRoleCollection(Arrays.asList(role));
user.persist();
user.reload();
Assert.assertEquals(1, user.getMembershipCollection(false).size());
ms = user.getMembershipCollection(false).get(0);
Assert.assertEquals(1, ms.getRoleCollection(false).size());
}
@Test
public void testRemoveMembershipWithRole() throws Exception {
String name = "removeMembershipWithRole" + r.nextInt();
UserWrapper user = UserHelper.addUser(name, null, true);
RoleWrapper role = RoleHelper.addRole(name, true);
MembershipWrapper mwr = MembershipHelper.newMembership(user, null, null);
mwr.addToRoleCollection(Arrays.asList(role));
user.persist();
user.reload();
Assert.assertEquals(1, user.getMembershipCollection(false).size());
mwr = user.getMembershipCollection(false).get(0);
Integer mwrId = mwr.getId();
user.removeFromMembershipCollection(Arrays.asList(mwr));
user.persist();
user.reload();
Assert.assertEquals(0, user.getMembershipCollection(false).size());
Membership msDB = ModelUtils.getObjectWithId(appService, Membership.class, mwrId);
Assert.assertNull(msDB);
}