Skip to content

Commit 2bd9048

Browse files
committed
Add tests capturing expectation of custom loaders. Ref python#268.
1 parent f9d039d commit 2bd9048

File tree

1 file changed

+45
-0
lines changed

1 file changed

+45
-0
lines changed
Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
import unittest
2+
import contextlib
3+
import pathlib
4+
5+
import importlib_resources as resources
6+
from ..abc import TraversableResources, ResourceReader
7+
from . import util
8+
from ._compat import os_helper
9+
10+
11+
class SimpleLoader:
12+
"""
13+
A simple loader that only implements a resource reader.
14+
"""
15+
16+
def __init__(self, reader: ResourceReader):
17+
self.reader = reader
18+
19+
def get_resource_reader(self, package):
20+
return self.reader
21+
22+
23+
class MagicResources(TraversableResources):
24+
"""
25+
Magically returns the resources at path.
26+
"""
27+
28+
def __init__(self, path: pathlib.Path):
29+
self.path = path
30+
31+
def files(self):
32+
return self.path
33+
34+
35+
class CustomTraversableResourcesTests(unittest.TestCase):
36+
def setUp(self):
37+
self.fixtures = contextlib.ExitStack()
38+
self.addCleanup(self.fixtures.close)
39+
40+
def test_custom_loader(self):
41+
temp_dir = self.fixtures.enter_context(os_helper.temp_dir())
42+
loader = SimpleLoader(MagicResources(temp_dir))
43+
pkg = util.create_package_from_loader(loader)
44+
files = resources.files(pkg)
45+
assert files is temp_dir

0 commit comments

Comments
 (0)