def test_has_admin_priviledges(self): """Test the has_admin_priviledges() method works correctly.""" user_data = UserData() user_data.groups = ["wheel"] assert user_data.has_admin_priviledges() user_data = UserData() user_data.groups = ["foo"] assert not user_data.has_admin_priviledges() user_data = UserData() user_data.groups = ["foo", "wheel", "bar"] assert user_data.has_admin_priviledges() # multiple wheels user_data = UserData() user_data.groups = ["foo", "wheel", "bar", "wheel", "baz"] assert user_data.has_admin_priviledges() # group name is case sensitive user_data = UserData() user_data.groups = ["WHEEL", "Wheel"] assert not user_data.has_admin_priviledges()
def has_admin_priviledges_test(self): """Test the has_admin_priviledges() method works correctly.""" user_data = UserData() user_data.groups = ["wheel"] self.assertTrue(user_data.has_admin_priviledges()) user_data = UserData() user_data.groups = ["foo"] self.assertFalse(user_data.has_admin_priviledges()) user_data = UserData() user_data.groups = ["foo", "wheel", "bar"] self.assertTrue(user_data.has_admin_priviledges()) # multiple wheels user_data = UserData() user_data.groups = ["foo", "wheel", "bar", "wheel", "baz"] self.assertTrue(user_data.has_admin_priviledges()) # group name is case sensitive user_data = UserData() user_data.groups = ["WHEEL", "Wheel"] self.assertFalse(user_data.has_admin_priviledges())
def test_set_admin_priviledges(self): """Test setting user admin privileges works correctly.""" user_data = UserData() assert not user_data.has_admin_priviledges() assert "wheel" not in user_data.groups # turn it on user_data.set_admin_priviledges(True) assert user_data.has_admin_priviledges() assert "wheel" in user_data.groups # turn it off user_data.set_admin_priviledges(False) assert not user_data.has_admin_priviledges() assert "wheel" not in user_data.groups # existing groups - turn in on user_data = UserData() user_data.groups = ["foo", "bar"] user_data.set_admin_priviledges(True) assert user_data.has_admin_priviledges() assert "wheel" in user_data.groups assert "foo" in user_data.groups assert "bar" in user_data.groups # existing groups - turn in off user_data.set_admin_priviledges(False) assert not user_data.has_admin_priviledges() assert "wheel" not in user_data.groups assert "foo" in user_data.groups assert "bar" in user_data.groups # group wheel added externally user_data = UserData() user_data.groups = ["foo", "bar", "wheel"] assert user_data.has_admin_priviledges() assert "wheel" in user_data.groups assert "foo" in user_data.groups assert "bar" in user_data.groups # now remove the wheel group via API user_data.set_admin_priviledges(False) assert not user_data.has_admin_priviledges() assert "wheel" not in user_data.groups assert "foo" in user_data.groups assert "bar" in user_data.groups
def set_admin_priviledges_test(self): """Test setting user admin privileges works correctly.""" user_data = UserData() self.assertFalse(user_data.has_admin_priviledges()) self.assertNotIn("wheel", user_data.groups) # turn it on user_data.set_admin_priviledges(True) self.assertTrue(user_data.has_admin_priviledges()) self.assertIn("wheel", user_data.groups) # turn it off user_data.set_admin_priviledges(False) self.assertFalse(user_data.has_admin_priviledges()) self.assertNotIn("wheel", user_data.groups) # existing groups - turn in on user_data = UserData() user_data.groups = ["foo", "bar"] user_data.set_admin_priviledges(True) self.assertTrue(user_data.has_admin_priviledges()) self.assertIn("wheel", user_data.groups) self.assertIn("foo", user_data.groups) self.assertIn("bar", user_data.groups) # existing groups - turn in off user_data.set_admin_priviledges(False) self.assertFalse(user_data.has_admin_priviledges()) self.assertNotIn("wheel", user_data.groups) self.assertIn("foo", user_data.groups) self.assertIn("bar", user_data.groups) # group wheel added externally user_data = UserData() user_data.groups = ["foo", "bar", "wheel"] self.assertTrue(user_data.has_admin_priviledges()) self.assertIn("wheel", user_data.groups) self.assertIn("foo", user_data.groups) self.assertIn("bar", user_data.groups) # now remove the wheel group via API user_data.set_admin_priviledges(False) self.assertFalse(user_data.has_admin_priviledges()) self.assertNotIn("wheel", user_data.groups) self.assertIn("foo", user_data.groups) self.assertIn("bar", user_data.groups)