forked from Rust-related/RustPython
af884cb284
* Mark 3.12 * Update importlib from Python 3.12.0 * Update test_importlib from Python3.12 * Mark failings tests from importlib * Update test.support from Python3.12 * Fix unsupported parser feature * mark failing test * Update functools from Python 3.12 * manual type annotation * slice behavior changed in 3.12 * empty unittest.main returns non-zero * test_decimal from CPython 3.12 * Mark failing tests * Update test_unicode from CPython 3.12 * Update test_functools from Python 3.12 * Update enum from Python 3.12 * enum * Doc format changed * Update test_module from CPython --------- Co-authored-by: CPython developers <>
47 lines
1.1 KiB
Python
Vendored
47 lines
1.1 KiB
Python
Vendored
import unittest
|
|
import contextlib
|
|
import pathlib
|
|
|
|
from test.support import os_helper
|
|
|
|
from importlib import resources
|
|
from importlib.resources.abc import TraversableResources, ResourceReader
|
|
from . import util
|
|
|
|
|
|
class SimpleLoader:
|
|
"""
|
|
A simple loader that only implements a resource reader.
|
|
"""
|
|
|
|
def __init__(self, reader: ResourceReader):
|
|
self.reader = reader
|
|
|
|
def get_resource_reader(self, package):
|
|
return self.reader
|
|
|
|
|
|
class MagicResources(TraversableResources):
|
|
"""
|
|
Magically returns the resources at path.
|
|
"""
|
|
|
|
def __init__(self, path: pathlib.Path):
|
|
self.path = path
|
|
|
|
def files(self):
|
|
return self.path
|
|
|
|
|
|
class CustomTraversableResourcesTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.fixtures = contextlib.ExitStack()
|
|
self.addCleanup(self.fixtures.close)
|
|
|
|
def test_custom_loader(self):
|
|
temp_dir = self.fixtures.enter_context(os_helper.temp_dir())
|
|
loader = SimpleLoader(MagicResources(temp_dir))
|
|
pkg = util.create_package_from_loader(loader)
|
|
files = resources.files(pkg)
|
|
assert files is temp_dir
|