def test_get_loader_cls(): res = loaders.get_loader_cls(constants.ContentType.MODULE) assert issubclass(res, loaders.PluginLoader) assert not issubclass(res, loaders.RoleLoader) res = loaders.get_loader_cls(constants.ContentType.ROLE) assert issubclass(res, loaders.RoleLoader) assert not issubclass(res, loaders.PluginLoader)
def _load_contents(self): """Find and load data for each content inside the collection.""" found_contents = ContentFinder().find_contents(self.path, self.log) for content_type, rel_path in found_contents: loader_cls = loaders.get_loader_cls(content_type) loader = loader_cls(content_type, rel_path, self.path, self.log) content_obj = loader.load() yield content_obj