def test_get_specific_entry_point(self): """Test get a specific entry point.""" load_casefiles = loaders.get_entry_point_loader(self.settings) casefiles = load_casefiles(casefile_name='bang') casefile_names = sorted([x.name() for x in casefiles]) expected_casefile_names = sorted(['bang']) depends_on = 'deputy_lib' try: __import__(depends_on) self.assertEqual(casefile_names, expected_casefile_names) except ImportError: warn('Test depends on deputy_lib')
def test_get_all_entry_points(self): """Test get all entry points.""" load_casefiles = loaders.get_entry_point_loader(self.settings) casefiles = load_casefiles() casefile_names = sorted([x.name() for x in casefiles]) expected_casefile_names = sorted(['bang', 'bash', 'pop']) depends_on = 'deputy_lib' try: __import__(depends_on) self.assertEqual(casefile_names, expected_casefile_names) except ImportError: warn('Skipping test, deputy_lib is not available.')