def test_is_just_partner(self): user = CtsUserFactory(role=ROLE_PARTNER, is_superuser=False) self.assertTrue(user.is_just_partner()) user.role = ROLE_COORDINATOR user.save() self.assertFalse(user.is_just_partner()) user.role = ROLE_PARTNER user.is_superuser = True user.save() self.assertFalse(user.is_just_partner()) user = CtsUserFactory(role=ROLE_PARTNER, is_superuser=False) self.assertTrue(user.is_just_partner()) manager_group = Group.objects.get(name=ROLE_MANAGER) user.groups.add(manager_group) self.assertFalse(user.is_just_partner())
def test_save_sets_groups(self): user = CtsUserFactory(role=ROLE_PARTNER) self.assertTrue(user.groups.filter(name=ROLE_PARTNER).exists()) self.assertFalse(user.groups.filter(name=ROLE_OFFICER).exists()) self.assertFalse(user.groups.filter(name=ROLE_COORDINATOR).exists()) user.role = ROLE_OFFICER user.save() self.assertFalse(user.groups.filter(name=ROLE_PARTNER).exists()) self.assertTrue(user.groups.filter(name=ROLE_OFFICER).exists()) self.assertFalse(user.groups.filter(name=ROLE_COORDINATOR).exists()) user.role = ROLE_COORDINATOR user.save() self.assertFalse(user.groups.filter(name=ROLE_PARTNER).exists()) self.assertTrue(user.groups.filter(name=ROLE_OFFICER).exists()) self.assertTrue(user.groups.filter(name=ROLE_COORDINATOR).exists()) user.role = ROLE_PARTNER user.save() self.assertTrue(user.groups.filter(name=ROLE_PARTNER).exists()) self.assertFalse(user.groups.filter(name=ROLE_OFFICER).exists()) self.assertFalse(user.groups.filter(name=ROLE_COORDINATOR).exists())