def mock_attr_factory(self, name, types=[]):
"""Factory function for TypeAttribute objects, using a mock qpol object."""
mock_type = Mock(qpol.qpol_type_t)
mock_type.name.return_value = name
mock_type.type_iter = lambda x: iter(types)
mock_type.attr_iter.side_effect = AssertionError("Attr iter used")
mock_type.alias_iter.side_effect = AssertionError("Alias iter used")
mock_type.ispermissive.side_effect = AssertionError("Permissive used")
mock_type.isattr.return_value = True
mock_type.isalias.side_effect = AssertionError("Alias used")
return attribute_factory(self.p.policy, mock_type)
def mock_attr_factory(self, name, types=[]):
"""Factory function for TypeAttribute objects, using a mock qpol object."""
mock_type = Mock(qpol.qpol_type_t)
mock_type.name.return_value = name
mock_type.type_iter = lambda x: iter(types)
mock_type.attr_iter.side_effect = AssertionError("Attr iter used")
mock_type.alias_iter.side_effect = AssertionError("Alias iter used")
mock_type.ispermissive.side_effect = AssertionError("Permissive used")
mock_type.isattr.return_value = True
mock_type.isalias.side_effect = AssertionError("Alias used")
return attribute_factory(self.p.policy, mock_type)
def test_008_lookup_object(self):
"""TypeAttribute factory policy lookup of TypeAttribute object."""
attr1 = attribute_factory(self.p.policy, "attr1")
attr2 = attribute_factory(self.p.policy, attr1)
self.assertIs(attr2, attr1)
def test_002_lookup_invalid(self):
"""TypeAttribute factory policy invalid lookup."""
with self.assertRaises(InvalidType):
attribute_factory(self.p.policy, "INVALID")
def test_001_lookup(self):
"""TypeAttribute factory policy lookup."""
attr = attribute_factory(self.p.policy, "attr1")
self.assertEqual("attr1", attr.qpol_symbol.name(self.p.policy))
def test_008_lookup_object(self):
"""TypeAttribute factory policy lookup of TypeAttribute object."""
attr1 = attribute_factory(self.p.policy, "attr1")
attr2 = attribute_factory(self.p.policy, attr1)
self.assertIs(attr2, attr1)
def test_002_lookup_invalid(self):
"""TypeAttribute factory policy invalid lookup."""
with self.assertRaises(InvalidType):
attribute_factory(self.p.policy, "INVALID")
def test_001_lookup(self):
"""TypeAttribute factory policy lookup."""
attr = attribute_factory(self.p.policy, "attr1")
self.assertEqual("attr1", attr.qpol_symbol.name(self.p.policy))
