codekingpro/portable-devtools

1import contextlib2import pathlib3import unittest4 5import importlib_resources as resources6 7from .. import abc8from ..abc import ResourceReader, TraversableResources9from . import util10from .compat.py39 import os_helper11 12 13class SimpleLoader:14    """15    A simple loader that only implements a resource reader.16    """17 18    def __init__(self, reader: ResourceReader):19        self.reader = reader20 21    def get_resource_reader(self, package):22        return self.reader23 24 25class MagicResources(TraversableResources):26    """27    Magically returns the resources at path.28    """29 30    def __init__(self, path: pathlib.Path):31        self.path = path32 33    def files(self):34        return self.path35 36 37class CustomTraversableResourcesTests(unittest.TestCase):38    def setUp(self):39        self.fixtures = contextlib.ExitStack()40        self.addCleanup(self.fixtures.close)41 42    def test_custom_loader(self):43        temp_dir = pathlib.Path(self.fixtures.enter_context(os_helper.temp_dir()))44        loader = SimpleLoader(MagicResources(temp_dir))45        pkg = util.create_package_from_loader(loader)46        files = resources.files(pkg)47        assert isinstance(files, abc.Traversable)48        assert list(files.iterdir()) == []49