def test_130_exclude_class(self):
"""PermMap exclude class."""
permmap = PermissionMap("tests/perm_map")
permmap.exclude_class("file")
self.validate_permmap_entry(permmap._permmap, 'file', 'execute', 'r',
10, False)
self.validate_permmap_entry(permmap._permmap, 'file', 'entrypoint',
'r', 10, False)
def test_132_include_class(self):
"""PermMap exclude class."""
permmap = PermissionMap("tests/perm_map")
permmap.exclude_class("file")
self.validate_permmap_entry(permmap.permmap, 'file', 'execute', 'r', 10, False)
self.validate_permmap_entry(permmap.permmap, 'file', 'entrypoint', 'r', 10, False)
permmap.include_class("file")
self.validate_permmap_entry(permmap.permmap, 'file', 'execute', 'r', 10, True)
self.validate_permmap_entry(permmap.permmap, 'file', 'entrypoint', 'r', 10, True)
def test_148_weight_excluded_class(self):
"""PermMap get weight of a rule with excluded class."""
rule = Mock()
rule.ruletype = TERuletype.allow
rule.tclass = "infoflow"
rule.perms = set(["low_r", "med_r", "hi_r", "low_w", "med_w", "hi_w"])
permmap = PermissionMap("tests/perm_map")
permmap.exclude_class("infoflow")
r, w = permmap.rule_weight(rule)
self.assertEqual(r, 0)
self.assertEqual(w, 0)
def test_148_weight_excluded_class(self):
"""PermMap get weight of a rule with excluded class."""
rule = Mock()
rule.ruletype = "allow"
rule.tclass = "infoflow"
rule.perms = set(["low_r", "med_r", "hi_r", "low_w", "med_w", "hi_w"])
permmap = PermissionMap("tests/perm_map")
permmap.exclude_class("infoflow")
r, w = permmap.rule_weight(rule)
self.assertEqual(r, 0)
self.assertEqual(w, 0)
def test_131_exclude_class_unmapped_class(self):
"""PermMap exclude class unmapped class."""
permmap = PermissionMap("tests/perm_map")
with self.assertRaises(UnmappedClass):
permmap.exclude_class("UNMAPPED")
def test_131_exclude_class_unmapped_class(self):
"""PermMap exclude class unmapped class."""
permmap = PermissionMap("tests/perm_map")
with self.assertRaises(UnmappedClass):
permmap.exclude_class("UNMAPPED")
