def mock_cat_factory(self, cat, aliases=[]):
"""Factory function for Category objects, using a mock qpol object."""
mock_cat = Mock(qpol.qpol_cat_t)
mock_cat.name.return_value = cat
mock_cat.isalias.return_value = False
mock_cat.value.return_value = int(cat[1:])
mock_cat.alias_iter = lambda x: iter(aliases)
return category_factory(self.p.policy, mock_cat)
def mock_cat_factory(self, cat, aliases=[]):
"""Factory function for Category objects, using a mock qpol object."""
mock_cat = Mock(qpol.qpol_cat_t)
mock_cat.name.return_value = cat
mock_cat.isalias.return_value = False
mock_cat.value.return_value = int(cat[1:])
mock_cat.alias_iter = lambda x: iter(aliases)
return category_factory(self.p.policy, mock_cat)
def test_003_lookup_object(self):
"""Category factory policy lookup of Category object."""
cat1 = category_factory(self.p.policy, "c1")
cat2 = category_factory(self.p.policy, cat1)
self.assertIs(cat2, cat1)
def test_002_lookup_invalid(self):
"""Category factory policy invalid lookup."""
with self.assertRaises(InvalidCategory):
category_factory(self.p.policy, "INVALID")
def test_001_lookup(self):
"""Category factory policy lookup."""
cat = category_factory(self.p.policy, "c1")
self.assertEqual("c1", cat.qpol_symbol.name(self.p.policy))
def test_003_lookup_object(self):
"""Category factory policy lookup of Category object."""
cat1 = category_factory(self.p.policy, "c1")
cat2 = category_factory(self.p.policy, cat1)
self.assertIs(cat2, cat1)
def test_002_lookup_invalid(self):
"""Category factory policy invalid lookup."""
with self.assertRaises(InvalidCategory):
category_factory(self.p.policy, "INVALID")
def test_001_lookup(self):
"""Category factory policy lookup."""
cat = category_factory(self.p.policy, "c1")
self.assertEqual("c1", cat.qpol_symbol.name(self.p.policy))
