1  import unittest
       2  import contextlib
       3  import pathlib
       4  
       5  from test.support import os_helper
       6  
       7  from importlib import resources
       8  from importlib.resources.abc import TraversableResources, ResourceReader
       9  from . import util
      10  
      11  
      12  class ESC[4;38;5;81mSimpleLoader:
      13      """
      14      A simple loader that only implements a resource reader.
      15      """
      16  
      17      def __init__(self, reader: ResourceReader):
      18          self.reader = reader
      19  
      20      def get_resource_reader(self, package):
      21          return self.reader
      22  
      23  
      24  class ESC[4;38;5;81mMagicResources(ESC[4;38;5;149mTraversableResources):
      25      """
      26      Magically returns the resources at path.
      27      """
      28  
      29      def __init__(self, path: pathlib.Path):
      30          self.path = path
      31  
      32      def files(self):
      33          return self.path
      34  
      35  
      36  class ESC[4;38;5;81mCustomTraversableResourcesTests(ESC[4;38;5;149munittestESC[4;38;5;149m.ESC[4;38;5;149mTestCase):
      37      def setUp(self):
      38          self.fixtures = contextlib.ExitStack()
      39          self.addCleanup(self.fixtures.close)
      40  
      41      def test_custom_loader(self):
      42          temp_dir = self.fixtures.enter_context(os_helper.temp_dir())
      43          loader = SimpleLoader(MagicResources(temp_dir))
      44          pkg = util.create_package_from_loader(loader)
      45          files = resources.files(pkg)
      46          assert files is temp_dir