Example #1
0
    def test_remove_user(self):
        group1 = Group(
            domain=DOMAIN,
            name='group1',
            users=[self.active_user._id, self.inactive_user._id, self.deleted_user._id]
        )
        group1.save()

        self.assertTrue(group1.remove_user(self.active_user._id))
        group1.save()

        group1 = Group.get(group1._id)
        self.assertIn(self.active_user._id, group1.removed_users)
        self.assertNotIn(self.active_user._id, group1.users)

        group1.add_user(self.active_user._id)
        group1 = Group.get(group1._id)
        self.assertNotIn(self.active_user._id, group1.removed_users)
        self.assertIn(self.active_user._id, group1.users)
Example #2
0
    def test_remove_user(self):
        group1 = Group(
            domain=DOMAIN,
            name='group1',
            users=[self.active_user._id, self.inactive_user._id, self.deleted_user._id]
        )
        group1.save()

        self.assertTrue(group1.remove_user(self.active_user._id))
        group1.save()

        group1 = Group.get(group1._id)
        self.assertIn(self.active_user._id, group1.removed_users)
        self.assertNotIn(self.active_user._id, group1.users)

        group1.add_user(self.active_user._id)
        group1 = Group.get(group1._id)
        self.assertNotIn(self.active_user._id, group1.removed_users)
        self.assertIn(self.active_user._id, group1.users)