python (3.12.0)
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