def _get_policy_by_name(self): _topic = self.opts.topic.split('.')[-1] # mimic policy loading to discover all policiy classes without # needing to manually define each here import sos.policies.distros _helper = ImporterHelper(sos.policies.distros) for mod in _helper.get_modules(): for policy in import_policy(mod): _p = policy.__name__.lower().replace('policy', '') if _p == _topic: return policy
def test_cant_import(self): self.assertTrue(import_policy('notreal') is None)
def test_can_import(self): self.assertTrue(import_policy('redhat') is not None)