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)
